1// RUN: mlir-opt %s | mlir-opt | FileCheck %s 2 3// CHECK-LABEL: @binary_ops 4// CHECK-SAME: %[[A:.*]]: index, %[[B:.*]]: index 5func.func @binary_ops(%a: index, %b: index) { 6 // CHECK-NEXT: index.add %[[A]], %[[B]] 7 %0 = index.add %a, %b 8 // CHECK-NEXT: index.sub %[[A]], %[[B]] 9 %1 = index.sub %a, %b 10 // CHECK-NEXT: index.mul %[[A]], %[[B]] 11 %2 = index.mul %a, %b 12 // CHECK-NEXT: index.divs %[[A]], %[[B]] 13 %3 = index.divs %a, %b 14 // CHECK-NEXT: index.divu %[[A]], %[[B]] 15 %4 = index.divu %a, %b 16 // CHECK-NEXT: index.ceildivs %[[A]], %[[B]] 17 %5 = index.ceildivs %a, %b 18 // CHECK-NEXT: index.ceildivu %[[A]], %[[B]] 19 %6 = index.ceildivu %a, %b 20 // CHECK-NEXT: index.floordivs %[[A]], %[[B]] 21 %7 = index.floordivs %a, %b 22 // CHECK-NEXT: index.rems %[[A]], %[[B]] 23 %8 = index.rems %a, %b 24 // CHECK-NEXT: index.remu %[[A]], %[[B]] 25 %9 = index.remu %a, %b 26 // CHECK-NEXT: index.maxs %[[A]], %[[B]] 27 %10 = index.maxs %a, %b 28 // CHECK-NEXT: index.maxu %[[A]], %[[B]] 29 %11 = index.maxu %a, %b 30 // CHECK-NEXT: index.mins %[[A]], %[[B]] 31 %12 = index.mins %a, %b 32 // CHECK-NEXT: index.minu %[[A]], %[[B]] 33 %13 = index.minu %a, %b 34 // CHECK-NEXT: index.shl %[[A]], %[[B]] 35 %14 = index.shl %a, %b 36 // CHECK-NEXT: index.shrs %[[A]], %[[B]] 37 %15 = index.shrs %a, %b 38 // CHECK-NEXT: index.shru %[[A]], %[[B]] 39 %16 = index.shru %a, %b 40 // CHECK-NEXT: index.and %[[A]], %[[B]] 41 %17 = index.and %a, %b 42 // CHECK-NEXT: index.or %[[A]], %[[B]] 43 %18 = index.or %a, %b 44 // CHECK-NEXT: index.xor %[[A]], %[[B]] 45 %19 = index.xor %a, %b 46 return 47} 48 49// CHECK-LABEL: @cmp_op 50// CHECK-SAME: %[[A:.*]]: index, %[[B:.*]]: index 51func.func @cmp_op(%a: index, %b: index) { 52 // CHECK-NEXT: index.cmp eq(%[[A]], %[[B]]) 53 %0 = index.cmp eq(%a, %b) 54 // CHECK-NEXT: index.cmp ne(%[[A]], %[[B]]) 55 %1 = index.cmp ne(%a, %b) 56 // CHECK-NEXT: index.cmp slt(%[[A]], %[[B]]) 57 %2 = index.cmp slt(%a, %b) 58 // CHECK-NEXT: index.cmp sle(%[[A]], %[[B]]) 59 %3 = index.cmp sle(%a, %b) 60 // CHECK-NEXT: index.cmp sgt(%[[A]], %[[B]]) 61 %4 = index.cmp sgt(%a, %b) 62 // CHECK-NEXT: index.cmp sge(%[[A]], %[[B]]) 63 %5 = index.cmp sge(%a, %b) 64 // CHECK-NEXT: index.cmp ult(%[[A]], %[[B]]) 65 %6 = index.cmp ult(%a, %b) 66 // CHECK-NEXT: index.cmp ule(%[[A]], %[[B]]) 67 %7 = index.cmp ule(%a, %b) 68 // CHECK-NEXT: index.cmp ugt(%[[A]], %[[B]]) 69 %8 = index.cmp ugt(%a, %b) 70 // CHECK-NEXT: index.cmp uge(%[[A]], %[[B]]) 71 %9 = index.cmp uge(%a, %b) 72 return 73} 74 75// CHECK-LABEL: @sizeof_op 76func.func @sizeof_op() { 77 // CHECK: index.sizeof 78 %0 = index.sizeof 79 return 80} 81 82// CHECK-LABEL: @constant_op 83func.func @constant_op() { 84 // CHECK-NEXT: %idx0 = index.constant 0 85 %0 = index.constant 0 86 // CHECK-NEXT: %idx1 = index.constant 1 87 %1 = index.constant 1 88 // CHECK-NEXT: %idx42 = index.constant 42 89 %2 = index.constant 42 90 return 91} 92 93// CHECK-LABEL: @bool_constant_op 94func.func @bool_constant_op() { 95 // CHECK-NEXT: %true = index.bool.constant true 96 %0 = index.bool.constant true 97 // CHECK-NEXT: %false = index.bool.constant false 98 %1 = index.bool.constant false 99 return 100} 101 102// CHECK-LABEL: @cast_op 103// CHECK-SAME: %[[A:.*]]: index, %[[B:.*]]: i32, %[[C:.*]]: i64 104func.func @cast_op(%a: index, %b: i32, %c: i64) { 105 // CHECK-NEXT: index.casts %[[A]] : index to i64 106 %0 = index.casts %a : index to i64 107 // CHECK-NEXT: index.casts %[[B]] : i32 to index 108 %1 = index.casts %b : i32 to index 109 // CHECK-NEXT: index.casts %[[C]] : i64 to index 110 %2 = index.casts %c : i64 to index 111 // CHECK-NEXT: index.castu %[[A]] : index to i64 112 %3 = index.castu %a : index to i64 113 // CHECK-NEXT: index.castu %[[B]] : i32 to index 114 %4 = index.castu %b : i32 to index 115 // CHECK-NEXT: index.castu %[[C]] : i64 to index 116 %5 = index.castu %c : i64 to index 117 return 118} 119