Skip to content

Commit 55702d9

Browse files
leonling-llchengjunlu
leonling-ll
and
chengjunlu
authored
Enable 3d dot (#2518)
This pull request enable 3d dot. With this change, we can pass 161 `test_core.py::test_dot3d` unit test cases. --------- Co-authored-by: Lu, Chengjun <[email protected]>
1 parent 04a7b65 commit 55702d9

File tree

13 files changed

+557
-809
lines changed

13 files changed

+557
-809
lines changed

‎python/test/unit/language/test_core.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3557,7 +3557,7 @@ def test_dot3d(B, num_warps, M, N, K, BLOCK_M, BLOCK_N, in_dtype_str, out_dtype_
35573557
if out_dtype_str == "float16":
35583558
pytest.skip(f"{out_dtype_str} has low precision in WMMA dot")
35593559
else:
3560-
input_precision = "tf32" if is_cuda() and in_dtype_str == 'float32' else "ieee"
3560+
input_precision = "tf32" if (is_cuda() or is_xpu()) and in_dtype_str == 'float32' else "ieee"
35613561

35623562
if B == 8 and M == 64 and in_dtype_str == "float32" and out_dtype_str == "float32":
35633563
if not is_interpreter() and torch.cuda.is_available(

‎scripts/skiplist/a770/language.txt‎

Lines changed: 0 additions & 161 deletions
Original file line numberDiff line numberDiff line change
@@ -1,166 +1,5 @@
11
# https://github.com/intel/intel-xpu-backend-for-triton/issues/1434
22
test/unit/language/test_core.py::test_precise_math[1-tl.math.sqrt_rn(x)-tl.math.sqrt(x.to(tl.float64)).to(tl.float32)]
3-
test/unit/language/test_core.py::test_dot3d[1-1-32-32-32-32-32-float16-float16]
4-
test/unit/language/test_core.py::test_dot3d[1-1-32-32-32-32-32-float16-float32]
5-
test/unit/language/test_core.py::test_dot3d[1-1-32-32-32-32-32-float32-float32]
6-
test/unit/language/test_core.py::test_dot3d[1-1-32-32-32-32-32-int8-int8]
7-
test/unit/language/test_core.py::test_dot3d[1-1-64-64-64-32-32-float16-float16]
8-
test/unit/language/test_core.py::test_dot3d[1-1-64-64-64-32-32-float16-float32]
9-
test/unit/language/test_core.py::test_dot3d[1-1-64-64-64-32-32-float32-float32]
10-
test/unit/language/test_core.py::test_dot3d[1-1-64-64-64-32-32-int8-int8]
11-
test/unit/language/test_core.py::test_dot3d[1-16-32-32-32-32-32-float16-float16]
12-
test/unit/language/test_core.py::test_dot3d[1-16-32-32-32-32-32-float16-float32]
13-
test/unit/language/test_core.py::test_dot3d[1-16-32-32-32-32-32-float32-float32]
14-
test/unit/language/test_core.py::test_dot3d[1-16-32-32-32-32-32-int8-int8]
15-
test/unit/language/test_core.py::test_dot3d[1-16-64-64-64-32-32-float16-float16]
16-
test/unit/language/test_core.py::test_dot3d[1-16-64-64-64-32-32-float16-float32]
17-
test/unit/language/test_core.py::test_dot3d[1-16-64-64-64-32-32-float32-float32]
18-
test/unit/language/test_core.py::test_dot3d[1-16-64-64-64-32-32-int8-int8]
19-
test/unit/language/test_core.py::test_dot3d[1-2-32-32-32-32-32-float16-float16]
20-
test/unit/language/test_core.py::test_dot3d[1-2-32-32-32-32-32-float16-float32]
21-
test/unit/language/test_core.py::test_dot3d[1-2-32-32-32-32-32-float32-float32]
22-
test/unit/language/test_core.py::test_dot3d[1-2-32-32-32-32-32-int8-int8]
23-
test/unit/language/test_core.py::test_dot3d[1-2-64-64-64-32-32-float16-float16]
24-
test/unit/language/test_core.py::test_dot3d[1-2-64-64-64-32-32-float16-float32]
25-
test/unit/language/test_core.py::test_dot3d[1-2-64-64-64-32-32-float32-float32]
26-
test/unit/language/test_core.py::test_dot3d[1-2-64-64-64-32-32-int8-int8]
27-
test/unit/language/test_core.py::test_dot3d[1-4-32-32-32-32-32-float16-float16]
28-
test/unit/language/test_core.py::test_dot3d[1-4-32-32-32-32-32-float16-float32]
29-
test/unit/language/test_core.py::test_dot3d[1-4-32-32-32-32-32-float32-float32]
30-
test/unit/language/test_core.py::test_dot3d[1-4-32-32-32-32-32-int8-int8]
31-
test/unit/language/test_core.py::test_dot3d[1-4-64-64-64-32-32-float16-float16]
32-
test/unit/language/test_core.py::test_dot3d[1-4-64-64-64-32-32-float16-float32]
33-
test/unit/language/test_core.py::test_dot3d[1-4-64-64-64-32-32-float32-float32]
34-
test/unit/language/test_core.py::test_dot3d[1-4-64-64-64-32-32-int8-int8]
35-
test/unit/language/test_core.py::test_dot3d[1-8-32-32-32-32-32-float16-float16]
36-
test/unit/language/test_core.py::test_dot3d[1-8-32-32-32-32-32-float16-float32]
37-
test/unit/language/test_core.py::test_dot3d[1-8-32-32-32-32-32-float32-float32]
38-
test/unit/language/test_core.py::test_dot3d[1-8-32-32-32-32-32-int8-int8]
39-
test/unit/language/test_core.py::test_dot3d[1-8-64-64-64-32-32-float16-float16]
40-
test/unit/language/test_core.py::test_dot3d[1-8-64-64-64-32-32-float16-float32]
41-
test/unit/language/test_core.py::test_dot3d[1-8-64-64-64-32-32-float32-float32]
42-
test/unit/language/test_core.py::test_dot3d[1-8-64-64-64-32-32-int8-int8]
43-
test/unit/language/test_core.py::test_dot3d[2-1-32-32-32-32-32-float16-float16]
44-
test/unit/language/test_core.py::test_dot3d[2-1-32-32-32-32-32-float16-float32]
45-
test/unit/language/test_core.py::test_dot3d[2-1-32-32-32-32-32-float32-float32]
46-
test/unit/language/test_core.py::test_dot3d[2-1-32-32-32-32-32-int8-int8]
47-
test/unit/language/test_core.py::test_dot3d[2-1-64-64-64-32-32-float16-float16]
48-
test/unit/language/test_core.py::test_dot3d[2-1-64-64-64-32-32-float16-float32]
49-
test/unit/language/test_core.py::test_dot3d[2-1-64-64-64-32-32-float32-float32]
50-
test/unit/language/test_core.py::test_dot3d[2-1-64-64-64-32-32-int8-int8]
51-
test/unit/language/test_core.py::test_dot3d[2-16-32-32-32-32-32-float16-float16]
52-
test/unit/language/test_core.py::test_dot3d[2-16-32-32-32-32-32-float16-float32]
53-
test/unit/language/test_core.py::test_dot3d[2-16-32-32-32-32-32-float32-float32]
54-
test/unit/language/test_core.py::test_dot3d[2-16-32-32-32-32-32-int8-int8]
55-
test/unit/language/test_core.py::test_dot3d[2-16-64-64-64-32-32-float16-float16]
56-
test/unit/language/test_core.py::test_dot3d[2-16-64-64-64-32-32-float16-float32]
57-
test/unit/language/test_core.py::test_dot3d[2-16-64-64-64-32-32-float32-float32]
58-
test/unit/language/test_core.py::test_dot3d[2-16-64-64-64-32-32-int8-int8]
59-
test/unit/language/test_core.py::test_dot3d[2-2-32-32-32-32-32-float16-float16]
60-
test/unit/language/test_core.py::test_dot3d[2-2-32-32-32-32-32-float16-float32]
61-
test/unit/language/test_core.py::test_dot3d[2-2-32-32-32-32-32-float32-float32]
62-
test/unit/language/test_core.py::test_dot3d[2-2-32-32-32-32-32-int8-int8]
63-
test/unit/language/test_core.py::test_dot3d[2-2-64-64-64-32-32-float16-float16]
64-
test/unit/language/test_core.py::test_dot3d[2-2-64-64-64-32-32-float16-float32]
65-
test/unit/language/test_core.py::test_dot3d[2-2-64-64-64-32-32-float32-float32]
66-
test/unit/language/test_core.py::test_dot3d[2-2-64-64-64-32-32-int8-int8]
67-
test/unit/language/test_core.py::test_dot3d[2-4-32-32-32-32-32-float16-float16]
68-
test/unit/language/test_core.py::test_dot3d[2-4-32-32-32-32-32-float16-float32]
69-
test/unit/language/test_core.py::test_dot3d[2-4-32-32-32-32-32-float32-float32]
70-
test/unit/language/test_core.py::test_dot3d[2-4-32-32-32-32-32-int8-int8]
71-
test/unit/language/test_core.py::test_dot3d[2-4-64-64-64-32-32-float16-float16]
72-
test/unit/language/test_core.py::test_dot3d[2-4-64-64-64-32-32-float16-float32]
73-
test/unit/language/test_core.py::test_dot3d[2-4-64-64-64-32-32-float32-float32]
74-
test/unit/language/test_core.py::test_dot3d[2-4-64-64-64-32-32-int8-int8]
75-
test/unit/language/test_core.py::test_dot3d[2-8-32-32-32-32-32-float16-float16]
76-
test/unit/language/test_core.py::test_dot3d[2-8-32-32-32-32-32-float16-float32]
77-
test/unit/language/test_core.py::test_dot3d[2-8-32-32-32-32-32-float32-float32]
78-
test/unit/language/test_core.py::test_dot3d[2-8-32-32-32-32-32-int8-int8]
79-
test/unit/language/test_core.py::test_dot3d[2-8-64-64-64-32-32-float16-float16]
80-
test/unit/language/test_core.py::test_dot3d[2-8-64-64-64-32-32-float16-float32]
81-
test/unit/language/test_core.py::test_dot3d[2-8-64-64-64-32-32-float32-float32]
82-
test/unit/language/test_core.py::test_dot3d[2-8-64-64-64-32-32-int8-int8]
83-
test/unit/language/test_core.py::test_dot3d[4-1-32-32-32-32-32-float16-float16]
84-
test/unit/language/test_core.py::test_dot3d[4-1-32-32-32-32-32-float16-float32]
85-
test/unit/language/test_core.py::test_dot3d[4-1-32-32-32-32-32-float32-float32]
86-
test/unit/language/test_core.py::test_dot3d[4-1-32-32-32-32-32-int8-int8]
87-
test/unit/language/test_core.py::test_dot3d[4-1-64-64-64-32-32-float16-float16]
88-
test/unit/language/test_core.py::test_dot3d[4-1-64-64-64-32-32-float16-float32]
89-
test/unit/language/test_core.py::test_dot3d[4-1-64-64-64-32-32-float32-float32]
90-
test/unit/language/test_core.py::test_dot3d[4-1-64-64-64-32-32-int8-int8]
91-
test/unit/language/test_core.py::test_dot3d[4-16-32-32-32-32-32-float16-float16]
92-
test/unit/language/test_core.py::test_dot3d[4-16-32-32-32-32-32-float16-float32]
93-
test/unit/language/test_core.py::test_dot3d[4-16-32-32-32-32-32-float32-float32]
94-
test/unit/language/test_core.py::test_dot3d[4-16-32-32-32-32-32-int8-int8]
95-
test/unit/language/test_core.py::test_dot3d[4-16-64-64-64-32-32-float16-float16]
96-
test/unit/language/test_core.py::test_dot3d[4-16-64-64-64-32-32-float16-float32]
97-
test/unit/language/test_core.py::test_dot3d[4-16-64-64-64-32-32-float32-float32]
98-
test/unit/language/test_core.py::test_dot3d[4-16-64-64-64-32-32-int8-int8]
99-
test/unit/language/test_core.py::test_dot3d[4-2-32-32-32-32-32-float16-float16]
100-
test/unit/language/test_core.py::test_dot3d[4-2-32-32-32-32-32-float16-float32]
101-
test/unit/language/test_core.py::test_dot3d[4-2-32-32-32-32-32-float32-float32]
102-
test/unit/language/test_core.py::test_dot3d[4-2-32-32-32-32-32-int8-int8]
103-
test/unit/language/test_core.py::test_dot3d[4-2-64-64-64-32-32-float16-float16]
104-
test/unit/language/test_core.py::test_dot3d[4-2-64-64-64-32-32-float16-float32]
105-
test/unit/language/test_core.py::test_dot3d[4-2-64-64-64-32-32-float32-float32]
106-
test/unit/language/test_core.py::test_dot3d[4-2-64-64-64-32-32-int8-int8]
107-
test/unit/language/test_core.py::test_dot3d[4-4-128-128-64-64-64-float16-float16]
108-
test/unit/language/test_core.py::test_dot3d[4-4-32-32-32-32-32-float16-float16]
109-
test/unit/language/test_core.py::test_dot3d[4-4-32-32-32-32-32-float16-float32]
110-
test/unit/language/test_core.py::test_dot3d[4-4-32-32-32-32-32-float32-float32]
111-
test/unit/language/test_core.py::test_dot3d[4-4-32-32-32-32-32-int8-int8]
112-
test/unit/language/test_core.py::test_dot3d[4-4-64-64-64-32-32-float16-float16]
113-
test/unit/language/test_core.py::test_dot3d[4-4-64-64-64-32-32-float16-float32]
114-
test/unit/language/test_core.py::test_dot3d[4-4-64-64-64-32-32-float32-float32]
115-
test/unit/language/test_core.py::test_dot3d[4-4-64-64-64-32-32-int8-int8]
116-
test/unit/language/test_core.py::test_dot3d[4-8-32-32-32-32-32-float16-float16]
117-
test/unit/language/test_core.py::test_dot3d[4-8-32-32-32-32-32-float16-float32]
118-
test/unit/language/test_core.py::test_dot3d[4-8-32-32-32-32-32-float32-float32]
119-
test/unit/language/test_core.py::test_dot3d[4-8-32-32-32-32-32-int8-int8]
120-
test/unit/language/test_core.py::test_dot3d[4-8-64-64-64-32-32-float16-float16]
121-
test/unit/language/test_core.py::test_dot3d[4-8-64-64-64-32-32-float16-float32]
122-
test/unit/language/test_core.py::test_dot3d[4-8-64-64-64-32-32-float32-float32]
123-
test/unit/language/test_core.py::test_dot3d[4-8-64-64-64-32-32-int8-int8]
124-
test/unit/language/test_core.py::test_dot3d[8-1-32-32-32-32-32-float16-float16]
125-
test/unit/language/test_core.py::test_dot3d[8-1-32-32-32-32-32-float16-float32]
126-
test/unit/language/test_core.py::test_dot3d[8-1-32-32-32-32-32-float32-float32]
127-
test/unit/language/test_core.py::test_dot3d[8-1-32-32-32-32-32-int8-int8]
128-
test/unit/language/test_core.py::test_dot3d[8-1-64-64-64-32-32-float16-float16]
129-
test/unit/language/test_core.py::test_dot3d[8-1-64-64-64-32-32-float16-float32]
130-
test/unit/language/test_core.py::test_dot3d[8-1-64-64-64-32-32-float32-float32]
131-
test/unit/language/test_core.py::test_dot3d[8-1-64-64-64-32-32-int8-int8]
132-
test/unit/language/test_core.py::test_dot3d[8-16-32-32-32-32-32-float16-float16]
133-
test/unit/language/test_core.py::test_dot3d[8-16-32-32-32-32-32-float16-float32]
134-
test/unit/language/test_core.py::test_dot3d[8-16-32-32-32-32-32-float32-float32]
135-
test/unit/language/test_core.py::test_dot3d[8-16-32-32-32-32-32-int8-int8]
136-
test/unit/language/test_core.py::test_dot3d[8-16-64-64-64-32-32-float16-float16]
137-
test/unit/language/test_core.py::test_dot3d[8-16-64-64-64-32-32-float16-float32]
138-
test/unit/language/test_core.py::test_dot3d[8-16-64-64-64-32-32-float32-float32]
139-
test/unit/language/test_core.py::test_dot3d[8-16-64-64-64-32-32-int8-int8]
140-
test/unit/language/test_core.py::test_dot3d[8-2-32-32-32-32-32-float16-float16]
141-
test/unit/language/test_core.py::test_dot3d[8-2-32-32-32-32-32-float16-float32]
142-
test/unit/language/test_core.py::test_dot3d[8-2-32-32-32-32-32-float32-float32]
143-
test/unit/language/test_core.py::test_dot3d[8-2-32-32-32-32-32-int8-int8]
144-
test/unit/language/test_core.py::test_dot3d[8-2-64-64-64-32-32-float16-float16]
145-
test/unit/language/test_core.py::test_dot3d[8-2-64-64-64-32-32-float16-float32]
146-
test/unit/language/test_core.py::test_dot3d[8-2-64-64-64-32-32-float32-float32]
147-
test/unit/language/test_core.py::test_dot3d[8-2-64-64-64-32-32-int8-int8]
148-
test/unit/language/test_core.py::test_dot3d[8-4-32-32-32-32-32-float16-float16]
149-
test/unit/language/test_core.py::test_dot3d[8-4-32-32-32-32-32-float16-float32]
150-
test/unit/language/test_core.py::test_dot3d[8-4-32-32-32-32-32-float32-float32]
151-
test/unit/language/test_core.py::test_dot3d[8-4-32-32-32-32-32-int8-int8]
152-
test/unit/language/test_core.py::test_dot3d[8-4-64-64-64-32-32-float16-float16]
153-
test/unit/language/test_core.py::test_dot3d[8-4-64-64-64-32-32-float16-float32]
154-
test/unit/language/test_core.py::test_dot3d[8-4-64-64-64-32-32-float32-float32]
155-
test/unit/language/test_core.py::test_dot3d[8-4-64-64-64-32-32-int8-int8]
156-
test/unit/language/test_core.py::test_dot3d[8-8-32-32-32-32-32-float16-float16]
157-
test/unit/language/test_core.py::test_dot3d[8-8-32-32-32-32-32-float16-float32]
158-
test/unit/language/test_core.py::test_dot3d[8-8-32-32-32-32-32-float32-float32]
159-
test/unit/language/test_core.py::test_dot3d[8-8-32-32-32-32-32-int8-int8]
160-
test/unit/language/test_core.py::test_dot3d[8-8-64-64-64-32-32-float16-float16]
161-
test/unit/language/test_core.py::test_dot3d[8-8-64-64-64-32-32-float16-float32]
162-
test/unit/language/test_core.py::test_dot3d[8-8-64-64-64-32-32-float32-float32]
163-
test/unit/language/test_core.py::test_dot3d[8-8-64-64-64-32-32-int8-int8]
1643
# https://github.com/intel/intel-xpu-backend-for-triton/issues/983
1654
test/unit/language/test_core.py::test_noinline[shared]
1665
test/unit/language/test_core.py::test_dot[1-128-128-64-2-True-True-none-tf32-int8-int8-1_0]

0 commit comments

Comments
 (0)