[CI] skip sparse tensor core related tests as only sm90 is supported

Author: botbw

examples/sparse_tensorcore/test_example_sparse_tensorcore.py

@@ -1,11 +1,12 @@
 # Copyright (c) Tile-AI Corporation.
 # Licensed under the MIT License.
 import tilelang.testing
-
+import tilelang
 import tilelang_example_sparse_tensorcore
 
 
 @tilelang.testing.requires_cuda
+@tilelang.testing.requires_cuda_compute_version(9, 0)
 def test_tilelang_example_sparse_tensorcore():
     tilelang_example_sparse_tensorcore.main()
 

testing/python/tilelibrary/test_tilelang_tilelibrary_gemm_sp.py

@@ -210,6 +210,8 @@ def _matmul(A, B):
     print("pass")
 
 
+@tilelang.testing.requires_cuda
+@tilelang.testing.requires_cuda_compute_version(9, 0)
 def test_gemm_sp():
     run_gemm_sp(512, 1024, 768, "float16", "float16", "float32", 64, 64, 32, 2, 128)
     run_gemm_sp(512, 1024, 768, "float16", "float16", "float32", 64, 64, 32, 0, 256)

testing/python/utils/test_compress_utils.py

@@ -1,4 +1,5 @@
 import torch
+import tilelang
 from tilelang.utils.sparse import compress_sm90
 
 
@@ -32,6 +33,8 @@ def _test_compress_sm90(M, K, block_k, dtype):
     A_sparse, E = compress_sm90(A, block_k)
 
 
+@tilelang.testing.requires_cuda
+@tilelang.testing.requires_cuda_compute_version(9, 0)
 def test_compress_sm90():
     _test_compress_sm90(1024, 1024, 128, torch.float16)
     _test_compress_sm90(1024, 1024, 64, torch.float16)