Skip to content

Commit ebd6146

Browse files
committed
Add test file for error with slice when using descript_scheduler
1 parent f1ec0c1 commit ebd6146

1 file changed

Lines changed: 45 additions & 0 deletions

File tree

Lines changed: 45 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,45 @@
1+
# RUN: python %s 2>&1 | filecheck %s
2+
3+
import xtc.graphs.xtc.op as O
4+
from xtc.backends.mlir import Backend
5+
from xtc.schedules.descript import descript_scheduler
6+
7+
I, J, K, dtype = 50, 64, 64, "float32"
8+
a = O.tensor((I, K), dtype, name="A")
9+
b = O.tensor((K, J), dtype, name="B")
10+
11+
with O.graph(name="matmul") as gb:
12+
O.matmul(a, b, name="C")
13+
14+
graph = gb.graph
15+
print(graph)
16+
17+
impl = Backend(graph)
18+
19+
sch = impl.get_scheduler()
20+
descript_scheduler(
21+
scheduler=sch,
22+
node_name="C",
23+
abstract_axis=["i", "j", "k"],
24+
spec={
25+
'k': {},
26+
'j': {},
27+
'i[0:18]':
28+
{'i#18': {},
29+
'k#32': {},
30+
'k#16': {},
31+
'i#9': {'unroll': 9},
32+
'j#32': {'unroll': 32},
33+
'j#16': {'vectorize': None}},
34+
'i[18:]': {
35+
'i#32': {},
36+
'k#32': {},
37+
'k#16': {},
38+
'i#16': {'unroll': 16},
39+
'j#32': {'unroll': 32},
40+
'j#16': {'vectorize': None}
41+
}
42+
}
43+
)
44+
45+
# XFAIL:*

0 commit comments

Comments
 (0)