From 0dfb87f25fb255cf8c648449227844efdb66a2b9 Mon Sep 17 00:00:00 2001
From: ywang103-amd <ywang103@amd.com>
Date: Tue, 7 Jan 2025 23:23:55 -0500
Subject: [PATCH] made change on the logic to add roofline file to expected csv
 list to fix crash on mi100 (#528)

---
 tests/test_profile_general.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/tests/test_profile_general.py b/tests/test_profile_general.py
index 05a583dae7..9e4632d72f 100644
--- a/tests/test_profile_general.py
+++ b/tests/test_profile_general.py
@@ -712,7 +712,7 @@ def test_block_SQC():
         "sysinfo.csv",
         "timestamps.csv",
     ]
-    if soc != ("MI50" or "MI60" or "MI100"):
+    if soc not in ["MI50", "MI60", "MI100"]:
         expected_csvs.append("roofline.csv")
 
     assert sorted(list(file_dict.keys())) == sorted(expected_csvs)
@@ -746,7 +746,7 @@ def test_block_TA():
         "sysinfo.csv",
         "timestamps.csv",
     ]
-    if soc != ("MI50" or "MI60" or "MI100"):
+    if soc not in ["MI50", "MI60", "MI100"]:
         expected_csvs.append("roofline.csv")
 
     assert sorted(list(file_dict.keys())) == sorted(expected_csvs)
@@ -820,7 +820,7 @@ def test_block_TCP():
         "sysinfo.csv",
         "timestamps.csv",
     ]
-    if soc != ("MI50" or "MI60" or "MI100"):
+    if soc not in ["MI50", "MI60", "MI100"]:
         expected_csvs.append("roofline.csv")
 
     assert sorted(list(file_dict.keys())) == sorted(expected_csvs)
@@ -906,7 +906,7 @@ def test_block_SPI():
         "sysinfo.csv",
         "timestamps.csv",
     ]
-    if soc != ("MI50" or "MI60" or "MI100"):
+    if soc not in ["MI50", "MI60", "MI100"]:
         expected_csvs.append("roofline.csv")
 
     assert sorted(list(file_dict.keys())) == sorted(expected_csvs)
@@ -938,7 +938,7 @@ def test_block_CPC():
         "sysinfo.csv",
         "timestamps.csv",
     ]
-    if soc != ("MI50" or "MI60" or "MI100"):
+    if soc not in ["MI50", "MI60", "MI100"]:
         expected_csvs.append("roofline.csv")
 
     assert sorted(list(file_dict.keys())) == sorted(expected_csvs)
@@ -964,7 +964,7 @@ def test_block_CPF():
         "sysinfo.csv",
         "timestamps.csv",
     ]
-    if soc != ("MI50" or "MI60" or "MI100"):
+    if soc not in ["MI50", "MI60", "MI100"]:
         expected_csvs.append("roofline.csv")
 
     assert sorted(list(file_dict.keys())) == sorted(expected_csvs)