119 lines · plain
1// RUN: mlir-opt %s | mlir-opt | FileCheck %s2 3// CHECK-LABEL: @binary_ops4// CHECK-SAME: %[[A:.*]]: index, %[[B:.*]]: index5func.func @binary_ops(%a: index, %b: index) {6 // CHECK-NEXT: index.add %[[A]], %[[B]]7 %0 = index.add %a, %b8 // CHECK-NEXT: index.sub %[[A]], %[[B]]9 %1 = index.sub %a, %b10 // CHECK-NEXT: index.mul %[[A]], %[[B]]11 %2 = index.mul %a, %b12 // CHECK-NEXT: index.divs %[[A]], %[[B]]13 %3 = index.divs %a, %b14 // CHECK-NEXT: index.divu %[[A]], %[[B]]15 %4 = index.divu %a, %b16 // CHECK-NEXT: index.ceildivs %[[A]], %[[B]]17 %5 = index.ceildivs %a, %b18 // CHECK-NEXT: index.ceildivu %[[A]], %[[B]]19 %6 = index.ceildivu %a, %b20 // CHECK-NEXT: index.floordivs %[[A]], %[[B]]21 %7 = index.floordivs %a, %b22 // CHECK-NEXT: index.rems %[[A]], %[[B]]23 %8 = index.rems %a, %b24 // CHECK-NEXT: index.remu %[[A]], %[[B]]25 %9 = index.remu %a, %b26 // CHECK-NEXT: index.maxs %[[A]], %[[B]]27 %10 = index.maxs %a, %b28 // CHECK-NEXT: index.maxu %[[A]], %[[B]]29 %11 = index.maxu %a, %b30 // CHECK-NEXT: index.mins %[[A]], %[[B]]31 %12 = index.mins %a, %b32 // CHECK-NEXT: index.minu %[[A]], %[[B]]33 %13 = index.minu %a, %b34 // CHECK-NEXT: index.shl %[[A]], %[[B]]35 %14 = index.shl %a, %b36 // CHECK-NEXT: index.shrs %[[A]], %[[B]]37 %15 = index.shrs %a, %b38 // CHECK-NEXT: index.shru %[[A]], %[[B]]39 %16 = index.shru %a, %b40 // CHECK-NEXT: index.and %[[A]], %[[B]]41 %17 = index.and %a, %b42 // CHECK-NEXT: index.or %[[A]], %[[B]]43 %18 = index.or %a, %b44 // CHECK-NEXT: index.xor %[[A]], %[[B]]45 %19 = index.xor %a, %b46 return47}48 49// CHECK-LABEL: @cmp_op50// CHECK-SAME: %[[A:.*]]: index, %[[B:.*]]: index51func.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 return73}74 75// CHECK-LABEL: @sizeof_op76func.func @sizeof_op() {77 // CHECK: index.sizeof78 %0 = index.sizeof79 return80}81 82// CHECK-LABEL: @constant_op83func.func @constant_op() {84 // CHECK-NEXT: %idx0 = index.constant 085 %0 = index.constant 086 // CHECK-NEXT: %idx1 = index.constant 187 %1 = index.constant 188 // CHECK-NEXT: %idx42 = index.constant 4289 %2 = index.constant 4290 return91}92 93// CHECK-LABEL: @bool_constant_op94func.func @bool_constant_op() {95 // CHECK-NEXT: %true = index.bool.constant true96 %0 = index.bool.constant true97 // CHECK-NEXT: %false = index.bool.constant false98 %1 = index.bool.constant false99 return100}101 102// CHECK-LABEL: @cast_op103// CHECK-SAME: %[[A:.*]]: index, %[[B:.*]]: i32, %[[C:.*]]: i64104func.func @cast_op(%a: index, %b: i32, %c: i64) {105 // CHECK-NEXT: index.casts %[[A]] : index to i64106 %0 = index.casts %a : index to i64107 // CHECK-NEXT: index.casts %[[B]] : i32 to index108 %1 = index.casts %b : i32 to index109 // CHECK-NEXT: index.casts %[[C]] : i64 to index110 %2 = index.casts %c : i64 to index111 // CHECK-NEXT: index.castu %[[A]] : index to i64112 %3 = index.castu %a : index to i64113 // CHECK-NEXT: index.castu %[[B]] : i32 to index114 %4 = index.castu %b : i32 to index115 // CHECK-NEXT: index.castu %[[C]] : i64 to index116 %5 = index.castu %c : i64 to index117 return118}119