| H A D | scf.py | 27 def simple_loop(lb, ub, step): argument 28 loop = scf.ForOp(lb, ub, step, [lb, lb]) 35 # CHECK: scf.for %{{.*}} = %[[ARG0]] to %[[ARG1]] step %[[ARG2]] 46 def induction_var(lb, ub, step): argument 47 loop = scf.ForOp(lb, ub, step, [lb]) 54 # CHECK: scf.for %[[IV:.*]] = %[[ARG0]] to %[[ARG1]] step %[[ARG2]] 66 # CHECK: scf.for %[[VAL_4:.*]] = %[[VAL_0]] to %[[VAL_1]] step %[[VAL_2]] { 73 def range_loop_1(lb, ub, step, memref_v): argument 74 for i in range(lb, ub, step): 83 # CHECK: scf.for %[[VAL_6:.*]] = %[[VAL_0]] to %[[VAL_4]] step 90 range_loop_2(lb, ub, step, memref_v) global() argument 106 range_loop_3(lb, ub, step, memref_v) global() argument 122 range_loop_4(lb, ub, step, memref_v) global() argument 139 range_loop_5(lb, ub, step, memref_v) global() argument 156 range_loop_6(lb, ub, step, memref_v) global() argument 173 range_loop_7(lb, ub, step, memref_v) global() argument 193 loop_yield_1(lb, ub, step, memref_v) global() argument 217 loop_yield_2(lb, ub, step, memref_v) global() argument [all...] |