Skip to content

Commit b8e195f

Browse files
committed
[CI] fix arch
1 parent 122bff0 commit b8e195f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

examples/sparse_tensorcore/tilelang_example_sparse_tensorcore.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,12 +41,12 @@ def main(
4141
T.annotate_layout({
4242
E:
4343
make_metadata_layout(
44-
E, mma_dtype="float16", arch="sm90", backend="cutlass", block_k=block_K),
44+
E, mma_dtype="float16", arch="9.0", backend="cutlass", block_k=block_K),
4545
E_shared:
4646
make_metadata_layout(
4747
E_shared,
4848
mma_dtype="float16",
49-
arch="sm90",
49+
arch="9.0",
5050
backend="cutlass",
5151
block_k=block_K),
5252
})

0 commit comments

Comments
 (0)