Skip to content

Commit 5000e32

Browse files
CRobeckantiagainst
andauthored
[AMD] Add check to fix test_store_cache_modifier for MI300 (triton-lang#4726)
Co-authored-by: Lei Zhang <[email protected]>
1 parent 84fe9da commit 5000e32

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

python/test/unit/language/test_core.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3924,6 +3924,8 @@ def _kernel(dst, src, CACHE: tl.constexpr):
39243924
cs_cache_modifier_str = 'nt'
39253925
wt_cache_modifier_str = 'sc0 sc1'
39263926
global_store_line = [line for line in amdgcn.splitlines() if "global_store" in line]
3927+
if not global_store_line:
3928+
return
39273929
if cache == '' or cache == '.cg':
39283930
assert cs_cache_modifier_str not in global_store_line[0]
39293931
assert wt_cache_modifier_str not in global_store_line[0]

0 commit comments

Comments
 (0)