Skip to content

Commit 7ea0aba

Browse files
committed
add test
Signed-off-by: Qidong Su <soodoshll@gmail.com>
1 parent 1356e89 commit 7ea0aba

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/examples/test_examples.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
from typing import Optional
2424

2525
import pytest
26-
from tilus.target import Target, get_current_target, nvgpu_sm80, nvgpu_sm100a
26+
from tilus.target import Target, get_current_target, nvgpu_sm80, nvgpu_sm90, nvgpu_sm100a
2727

2828
# Get the project root directory
2929
PROJECT_ROOT = Path(__file__).parent.parent.parent
@@ -54,6 +54,9 @@
5454
("blackwell_matmul", "matmul_v3.py", nvgpu_sm100a),
5555
("blackwell_matmul", "matmul_v4.py", nvgpu_sm100a),
5656
("blackwell_matmul", "matmul_v5.py", nvgpu_sm100a),
57+
# hopper matmul example (SM 9.0)
58+
("hopper_matmul", "matmul_tma.py", nvgpu_sm90),
59+
("hopper_matmul", "matmul_wgmma.py", nvgpu_sm90),
5760
# quantization examples (SM 8.0+)
5861
("quantization", "matmul_a16wx.py", nvgpu_sm80),
5962
# flash attention decode examples (SM 8.0+)

0 commit comments

Comments
 (0)