Lines Matching full:d1
88 d1 = AffineExpr.get_dim(1)
93 print(d1.position)
94 # CHECK: d1
95 print(d1)
102 assert d1 == d11
103 assert d1 != d2
132 d1 = AffineDimExpr.get(1)
134 d12 = AffineExpr.get_add(d1, d2)
135 # CHECK: d1 + d2
138 d12op = d1 + d2
139 # CHECK: d1 + d2
142 d1cst_op = d1 + 2
143 # CHECK: d1 + 2
146 d1cst_op2 = 2 + d1
147 # CHECK: d1 + 2
151 assert d12.lhs == d1
159 d1 = AffineDimExpr.get(1)
161 expr = AffineExpr.get_mul(d1, c2)
162 # CHECK: d1 * 2
165 # CHECK: d1 * 2
166 op = d1 * c2
169 # CHECK: d1 * 2
170 op_cst = d1 * 2
173 # CHECK: d1 * 2
174 op_cst2 = 2 * d1
179 assert expr.lhs == d1
187 d1 = AffineDimExpr.get(1)
189 expr = AffineExpr.get_mod(d1, c2)
190 # CHECK: d1 mod 2
193 # CHECK: d1 mod 2
194 op = d1 % c2
197 # CHECK: d1 mod 2
198 op_cst = d1 % 2
201 # CHECK: 2 mod d1
202 print(2 % d1)
206 assert expr.lhs == d1
209 expr2 = AffineExpr.get_mod(c2, d1)
210 expr3 = AffineExpr.get_mod(2, d1)
211 expr4 = AffineExpr.get_mod(d1, 2)
213 # CHECK: 2 mod d1
215 # CHECK: 2 mod d1
217 # CHECK: d1 mod 2
228 d1 = AffineDimExpr.get(1)
230 expr = AffineExpr.get_floor_div(d1, c2)
231 # CHECK: d1 floordiv 2
234 assert expr.lhs == d1
237 expr2 = AffineExpr.get_floor_div(c2, d1)
238 expr3 = AffineExpr.get_floor_div(2, d1)
239 expr4 = AffineExpr.get_floor_div(d1, 2)
241 # CHECK: 2 floordiv d1
243 # CHECK: 2 floordiv d1
245 # CHECK: d1 floordiv 2
256 d1 = AffineDimExpr.get(1)
258 expr = AffineExpr.get_ceil_div(d1, c2)
259 # CHECK: d1 ceildiv 2
262 assert expr.lhs == d1
265 expr2 = AffineExpr.get_ceil_div(c2, d1)
266 expr3 = AffineExpr.get_ceil_div(2, d1)
267 expr4 = AffineExpr.get_ceil_div(d1, 2)
269 # CHECK: 2 ceildiv d1
271 # CHECK: 2 ceildiv d1
273 # CHECK: d1 ceildiv 2
284 d1 = AffineDimExpr.get(1)
286 expr = d1 - d2
287 # CHECK: d1 - d2
290 assert expr.lhs == d1
297 # CHECK: d1 - 42
298 print(d1 - 42)
299 # CHECK: -d1 + 42
300 print(42 - d1)
303 assert d1 - 42 == d1 - c42
304 assert 42 - d1 == c42 - d1
311 d1 = AffineDimExpr.get(1)
313 add = AffineAddExpr.get(d1, c2)
314 mul = AffineMulExpr.get(d1, c2)
315 mod = AffineModExpr.get(d1, c2)
316 floor_div = AffineFloorDivExpr.get(d1, c2)
317 ceil_div = AffineCeilDivExpr.get(d1, c2)
320 print(isinstance(d1, AffineBinaryExpr))
335 AffineBinaryExpr(d1)
351 d1 = AffineDimExpr.get(1)
353 add = AffineAddExpr.get(d1, c2)
354 mul = AffineMulExpr.get(d1, c2)
357 print(AffineDimExpr.isinstance(d1))
359 print(AffineConstantExpr.isinstance(d1))
381 # (d0, d1, d2)[s0, s1] -> (d0 + s1, d1 + s0, d0 + d1 + d2)
390 # CHECK: d0 + s1 + d0 + d1 + d2