Lines Matching full:d1

33         d1 = AffineDimExpr.get(1)
36         # CHECK: (d0, d1)[s0, s1, s2] -> ()
40         # CHECK: (d0, d1)[s0, s1, s2] -> (d1, 2)
41         map1 = AffineMap.get(2, 3, [d1, c2])
48         # CHECK: (d0, d1) -> (d0, d1)
49         map3 = AffineMap.get(2, 0, [d0, d1])
52         # CHECK: (d0, d1) -> (d1)
53         map4 = AffineMap.get(2, 0, [d1])
56         # CHECK: (d0, d1, d2) -> (d2, d0, d1)
60         assert map1 == AffineMap.get(2, 3, [d1, c2])
109         # CHECK: (d0, d1, d2, d3, d4) -> (d1, d2, d3)
113         # CHECK: (d0, d1, d2, d3, d4) -> (d0, d1)
117         # CHECK: (d0, d1, d2, d3, d4) -> (d3, d4)
127         d1 = AffineDimExpr.get(1)
130         map2 = AffineMap.get(3, 0, [d2, d0, d1])
131         map3 = AffineMap.get(3, 1, [d2, d0, d1])
151         d1 = AffineDimExpr.get(1)
153         map3 = AffineMap.get(3, 1, [d2, d0, d1])
168             # CHECK: d1
171             # CHECK: d1
175         assert list(map3.results) == [d2, d0, d1]
182         d0, d1, d2 = (AffineDimExpr.get(0), AffineDimExpr.get(1), AffineDimExpr.get(2))
189             AffineMap.get(3, 3, [d2, d0, d1]),
190             AffineMap.get(3, 3, [d2, d0 + s2, d1]),
191             AffineMap.get(3, 3, [d1, d2, d0]),
196         #      CHECK: AffineMap((d0, d1, d2)[s0, s1, s2] -> (d2, d0, d1))
197         # CHECK-SAME: AffineMap((d0, d1, d2)[s0, s1, s2] -> (d2, d0 + s2, d1))
198         # CHECK-SAME: AffineMap((d0, d1, d2)[s0, s1, s2] -> (d1, d2, d0))
201         #      CHECK: AffineMap((d0, d1, d2)[s0] -> (d2, d0, d1))
202         # CHECK-SAME: AffineMap((d0, d1, d2)[s0] -> (d2, d0 + s0, d1))
203         # CHECK-SAME: AffineMap((d0, d1, d2)[s0] -> (d1, d2, d0))
211         d0, d1, d2 = (AffineDimExpr.get(0), AffineDimExpr.get(1), AffineDimExpr.get(2))
217         map1 = AffineMap.get(3, 3, [d2, d0 + s1 + s2, d1 + s0])
223         # CHECK: (d0, d1, d2)[s0, s1, s2] -> (d2, d0 + s1 + s2, d1 + 42)
226         # CHECK: (d0, d1, d2)[s0, s1, s2] -> (d2, d0 + s2 + 42, d1 + s0)
229         # CHECK: (d0, d1, d2)[s0, s1] -> (d2, d0 + s1 + 42, d1 + s0)
237         d0, d1 = AffineDimExpr.get(0), AffineDimExpr.get(1)
238         m1 = AffineMap.get(2, 0, [d0, d1])
239         m2 = AffineMap.get(2, 0, [d1, d0])
240         assert hash(m1) == hash(AffineMap.get(2, 0, [d0, d1]))