Skip to content

Commit 27b4491

Browse files
committed
[Update] Modify CUDA version requirements in test_gemm_sp_sm80 and mark cutlass subproject as dirty
1 parent 597b66f commit 27b4491

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

testing/python/tilelibrary/test_tilelang_tilelibrary_gemm_sp.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -365,7 +365,8 @@ def test_gemm_sp_sm90():
365365

366366

367367
@tilelang.testing.requires_cuda
368-
@tilelang.testing.requires_cuda_compute_version(8, 0)
368+
@tilelang.testing.requires_cuda_compute_version_ge(8, 0)
369+
@tilelang.testing.requires_cuda_compute_version_le(8, 9)
369370
def test_gemm_sp_sm80():
370371
run_gemm_sp_sm80(512, 1024, 768, "float16", "float32", "float32", 32, 32, 32, 0, 32)
371372
run_gemm_sp_sm80(512, 1024, 768, "float16", "float32", "float32", 64, 64, 64, 0, 32)

0 commit comments

Comments
 (0)