[Test] Disable mma layout for amd hardware

Disable mma layout testing by looking at is_hip instead of wave size. This fixes tests on Navi GPUs with wave size == 32.
ROCm · Nov 9, 2023 · ea5c894 · ea5c894
1 parent d4eda83
commit ea5c894
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/test/unit/language/test_core_amd.py b/python/test/unit/language/test_core_amd.py
@@ -2830,7 +2830,7 @@ def _get_warp_size():
     return _triton.get_warp_size()
 
 
-if _get_warp_size() == 64:
+if is_hip():
     layouts = [
         # MmaLayout(version=1, warps_per_cta=[1, 4]),
         # MmaLayout(version=(2, 0), warps_per_cta=[1, 4]),