Lines Matching full:index
10 // CHECK-SAME: %[[a:.*]]: index
15 // CHECK-ARITH-SAME: %[[a:.*]]: index
16 // CHECK-ARITH: %[[c5:.*]] = arith.constant 5 : index
18 // CHECK-ARITH: %[[c5:.*]] = arith.constant 5 : index
21 func.func @arith_addi(%a: index) -> index {
22 %0 = arith.constant 5 : index
23 %1 = arith.addi %0, %a : index
24 %2 = "test.reify_bound"(%1) : (index) -> (index)
25 return %2 : index
32 // CHECK-SAME: %[[a:.*]]: index
35 func.func @arith_subi(%a: index) -> index {
36 %0 = arith.constant 5 : index
37 %1 = arith.subi %0, %a : index
38 %2 = "test.reify_bound"(%1) : (index) -> (index)
39 return %2 : index
46 // CHECK-SAME: %[[a:.*]]: index
49 func.func @arith_muli(%a: index) -> index {
50 %0 = arith.constant 5 : index
51 %1 = arith.muli %0, %a : index
52 %2 = "test.reify_bound"(%1) : (index) -> (index)
53 return %2 : index
58 func.func @arith_muli_non_pure(%a: index, %b: index) -> index {
59 %0 = arith.muli %a, %b : index
62 %1 = "test.reify_bound"(%0) : (index) -> (index)
63 return %1 : index
69 // CHECK: %[[c5:.*]] = arith.constant 5 : index
70 // CHECK: %[[c5:.*]] = arith.constant 5 : index
72 func.func @arith_const() -> index {
73 %c5 = arith.constant 5 : index
74 %0 = "test.reify_bound"(%c5) : (index) -> (index)
75 return %0 : index
81 func.func @arith_select(%c: i1) -> (index, index) {
82 // CHECK: arith.constant 5 : index
83 %c5 = arith.constant 5 : index
84 // CHECK: arith.constant 9 : index
85 %c9 = arith.constant 9 : index
86 %r = arith.select %c, %c5, %c9 : index
87 // CHECK: %[[c5:.*]] = arith.constant 5 : index
88 // CHECK: %[[c10:.*]] = arith.constant 10 : index
89 %0 = "test.reify_bound"(%r) {type = "LB"} : (index) -> (index)
90 %1 = "test.reify_bound"(%r) {type = "UB"} : (index) -> (index)
92 return %0, %1 : index, index
99 func.func @arith_select_elementwise(%a: tensor<?xf32>, %b: tensor<?xf32>, %c: tensor<?xi1>) -> index {
101 // CHECK: %[[c0:.*]] = arith.constant 0 : index
104 : (tensor<?xf32>) -> (index)
106 return %0 : index