Skip to content

Commit 5576cc0

Browse files
authored
dialects: (stencil) Deprecate get constructors (#5741)
1 parent 1dcbc36 commit 5576cc0

17 files changed

Lines changed: 444 additions & 270 deletions

tests/dialects/test_csl_stencil.py

Lines changed: 15 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,32 +1,39 @@
11
from xdsl.builder import Builder
2-
from xdsl.dialects.builtin import IntegerAttr, IntegerType, MemRefType, TensorType, f32
2+
from xdsl.dialects.builtin import (
3+
IntAttr,
4+
IntegerAttr,
5+
IntegerType,
6+
MemRefType,
7+
TensorType,
8+
f32,
9+
)
310
from xdsl.dialects.csl.csl_stencil import AccessOp, ApplyOp
411
from xdsl.dialects.stencil import IndexAttr, TempType
512
from xdsl.ir import Region, SSAValue
613
from xdsl.utils.test_value import create_ssa_value
714

815

916
def test_access_patterns():
10-
temp_t = TempType(5, f32)
17+
temp_t = TempType(IntAttr(5), f32)
1118
temp = create_ssa_value(temp_t)
1219
mref = create_ssa_value(mref_t := MemRefType(tens_t := TensorType(f32, (5,)), (4,)))
1320

1421
@Builder.implicit_region((mref_t, temp_t))
1522
def region0(args: tuple[SSAValue, ...]):
1623
t0, t1 = args
1724
for x in (-1, 1):
18-
AccessOp(t0, IndexAttr.get(x, 0), tens_t)
25+
AccessOp(t0, IndexAttr.from_indices(x, 0), tens_t)
1926
for y in (-1, 1):
20-
AccessOp(t0, IndexAttr.get(0, y), tens_t)
27+
AccessOp(t0, IndexAttr.from_indices(0, y), tens_t)
2128

22-
AccessOp(t1, IndexAttr.get(1, 1), tens_t)
23-
AccessOp(t1, IndexAttr.get(-1, -1), tens_t)
29+
AccessOp(t1, IndexAttr.from_indices(1, 1), tens_t)
30+
AccessOp(t1, IndexAttr.from_indices(-1, -1), tens_t)
2431

2532
@Builder.implicit_region((temp_t, temp_t))
2633
def region1(args: tuple[SSAValue, ...]):
2734
t0, t1 = args
28-
AccessOp(t0, IndexAttr.get(0, 0), tens_t)
29-
AccessOp(t1, IndexAttr.get(0, 0), tens_t)
35+
AccessOp(t0, IndexAttr.from_indices(0, 0), tens_t)
36+
AccessOp(t1, IndexAttr.from_indices(0, 0), tens_t)
3037

3138
apply = ApplyOp(
3239
operands=[temp, mref, [], [], []],

0 commit comments

Comments
 (0)