github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-05 03:01:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

[Test] Disable mma layout for amd hardware (#384)

Browse Source 
Disable mma layout testing by looking at is_hip instead of wave size.
This fixes tests on Navi GPUs with wave size == 32.
This commit is contained in:
Alexander Efimov
2023-11-13 17:52:35 +01:00
committed by Jason Furmanek
parent 181bdbd410
commit 096def0c9b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/test/unit/language/test_core_amd.py
View File

@@ -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]),