Skip to content

Commit c669a23

Browse files
authored
Add 03i-matrix-multiplication (#4205)
Runs Matrix Multiplication tutorial with TRITON_INTEL_RAISE_BLOCK_POINTER=1. The unique name for this tutorial allows adding it to skip list if necessary, and reporting it in the pass rate report as an independent tutorial (the total number of tutorials increases to 11). --------- Signed-off-by: Pavel Chekin <[email protected]>
1 parent 669d3dd commit c669a23

File tree

3 files changed

+15
-2
lines changed

3 files changed

+15
-2
lines changed
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
"""
2+
Matrix Multiplication
3+
=====================
4+
Runs Matrix Multiplication tutorial with TRITON_INTEL_RAISE_BLOCK_POINTER=1.
5+
"""
6+
7+
import os.path
8+
import runpy
9+
10+
if __name__ == '__main__':
11+
os.environ['TRITON_INTEL_RAISE_BLOCK_POINTER'] = '1'
12+
dirname = os.path.dirname(__file__)
13+
runpy.run_path(f'{dirname}/03-matrix-multiplication.py')

scripts/skiplist/lts/tutorials.txt

+1
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
03-matrix-multiplication
2+
03i-matrix-multiplication
23
06-fused-attention
34
08-grouped-gemm
45
09-persistent-matmul

scripts/test-triton.sh

+1-2
Original file line numberDiff line numberDiff line change
@@ -373,8 +373,7 @@ run_tutorial_tests() {
373373
echo "Running with TRITON_INTEL_RAISE_BLOCK_POINTER "
374374
echo "***************************************************"
375375

376-
TRITON_TEST_REPORTS=false TRITON_INTEL_RAISE_BLOCK_POINTER=1 \
377-
run_tutorial_test "03-matrix-multiplication"
376+
run_tutorial_test "03i-matrix-multiplication"
378377
}
379378

380379
run_microbench_tests() {

0 commit comments

Comments
 (0)