add requires.nvgpu_sm90 for one test

yaoyaoding · yaoyaoding · commit 9abac20069b6 · 2025-09-12T03:06:05.000-04:00
Signed-off-by: Yaoyao Ding &lt;dingyaoyao.cs@gmail.com&gt;
diff --git a/tests/lang/test_shared_memory_addressing.py b/tests/lang/test_shared_memory_addressing.py
@@ -21,6 +21,7 @@
     cvta_shared_to_generic,
 )
 from tilus.extensions.hidet.ir.primitives.cuda.mapa import mapa_shared
+from tilus.testing import requires
 
 
 class SharedMemoryAddressingExample(tilus.Script):
@@ -96,6 +97,7 @@ def __call__(self):
         )
 
 
+@requires.nvgpu_sm90
 def test_shared_memory_address_conversion():
     kernel = SharedMemoryAddressingExample()
     kernel()

Original file line number	Diff line number	Diff line change
`@@ -21,6 +21,7 @@`
`21`	`21`	`cvta_shared_to_generic,`
`22`	`22`	`)`
`23`	`23`	`from tilus.extensions.hidet.ir.primitives.cuda.mapa import mapa_shared`
	`24`	`+from tilus.testing import requires`
`24`	`25`
`25`	`26`
`26`	`27`	`class SharedMemoryAddressingExample(tilus.Script):`
`@@ -96,6 +97,7 @@ def __call__(self):`
`96`	`97`	`)`
`97`	`98`
`98`	`99`
	`100`	`+@requires.nvgpu_sm90`
`99`	`101`	`def test_shared_memory_address_conversion():`
`100`	`102`	`kernel = SharedMemoryAddressingExample()`
`101`	`103`	`kernel()`