From ea5c8946c6a3e75f592386ef3207ac3c1d927afc Mon Sep 17 00:00:00 2001 From: Alexander Efimov Date: Wed, 1 Nov 2023 18:34:37 +0000 Subject: [PATCH] [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. --- python/test/unit/language/test_core_amd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/test/unit/language/test_core_amd.py b/python/test/unit/language/test_core_amd.py index 6264ec62fef4..18bd831f3164 100644 --- 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]),