From 3ba5f1d33769138e8466b545f8c06ce94b13a5a4 Mon Sep 17 00:00:00 2001
From: Karl W Schulz <karl.schulz@amd.com>
Date: Thu, 7 Dec 2023 15:57:45 -0600
Subject: [PATCH] Update return code error checks for roofline-only analysis on
 MI100

Signed-off-by: Karl W Schulz <karl.schulz@amd.com>
---
 tests/test_profile_general.py | 20 ++++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git a/tests/test_profile_general.py b/tests/test_profile_general.py
index 903c8a7523..1eee84cef0 100644
--- a/tests/test_profile_general.py
+++ b/tests/test_profile_general.py
@@ -1974,7 +1974,7 @@ def test_sort_dispatches():
 
     if soc == "mi100":
         # assert that it did not run
-        assert e.value.code == 2
+        assert e.value.code >= 1
         # Do not continue testing
         return
 
@@ -2026,7 +2026,7 @@ def test_sort_kernels():
             omniperf.main()
     if soc == "mi100":
         # assert that it did not run
-        assert e.value.code == 2
+        assert e.value.code >= 1
         # Do not continue testing
         return
 
@@ -2079,7 +2079,7 @@ def test_mem_levels_HBM():
 
     if soc == "mi100":
         # assert that it did not run
-        assert e.value.code == 2
+        assert e.value.code >= 1
         # Do not continue testing
         return
 
@@ -2132,7 +2132,7 @@ def test_mem_levels_L2():
 
     if soc == "mi100":
         # assert that it did not run
-        assert e.value.code == 2
+        assert e.value.code >= 1
         # Do not continue testing
         return
 
@@ -2184,7 +2184,7 @@ def test_mem_levels_vL1D():
             omniperf.main()
     if soc == "mi100":
         # assert that it did not run
-        assert e.value.code == 2
+        assert e.value.code >= 1
         # Do not continue testing
         return
 
@@ -2236,7 +2236,7 @@ def test_mem_levels_LDS():
             omniperf.main()
     if soc == "mi100":
         # assert that it did not run
-        assert e.value.code == 2
+        assert e.value.code >= 1
         # Do not continue testing
         return
 
@@ -2289,7 +2289,7 @@ def test_mem_levels_HBM_LDS():
             omniperf.main()
     if soc == "mi100":
         # assert that it did not run
-        assert e.value.code == 2
+        assert e.value.code >= 1
         # Do not continue testing
         return
 
@@ -2342,7 +2342,7 @@ def test_mem_levels_vL1D_LDS():
             omniperf.main()
     if soc == "mi100":
         # assert that it did not run
-        assert e.value.code == 2
+        assert e.value.code >= 1
         # Do not continue testing
         return
 
@@ -2396,7 +2396,7 @@ def test_mem_levels_L2_vL1D_LDS():
             omniperf.main()
     if soc == "mi100":
         # assert that it did not run
-        assert e.value.code == 2
+        assert e.value.code >= 1
         # Do not continue testing
         return
     # assert successful run
@@ -2447,7 +2447,7 @@ def test_kernel_names():
 
     if soc == "mi100":
         # assert that it did not run
-        assert e.value.code == 2
+        assert e.value.code >= 1
         # Do not continue testing
         return
     # assert successful run