198 lines · plain
1// RUN: mlir-opt %s -convert-index-to-llvm | FileCheck %s2// RUN: mlir-opt %s -convert-index-to-llvm=index-bitwidth=32 | FileCheck %s --check-prefix=INDEX323// RUN: mlir-opt %s -convert-index-to-llvm=index-bitwidth=64 | FileCheck %s --check-prefix=INDEX644 5// Same below, but using the `ConvertToLLVMPatternInterface` entry point6// and the generic `convert-to-llvm` pass.7// RUN: mlir-opt --convert-to-llvm="filter-dialects=index" --split-input-file %s | FileCheck %s8// RUN: mlir-opt --convert-to-llvm="filter-dialects=index allow-pattern-rollback=0" --split-input-file %s | FileCheck %s9 10// CHECK-LABEL: @trivial_ops11func.func @trivial_ops(%a: index, %b: index) {12 // CHECK: llvm.add13 %0 = index.add %a, %b14 // CHECK: llvm.sub15 %1 = index.sub %a, %b16 // CHECK: llvm.mul17 %2 = index.mul %a, %b18 // CHECK: llvm.sdiv19 %3 = index.divs %a, %b20 // CHECK: llvm.udiv21 %4 = index.divu %a, %b22 // CHECK: llvm.srem23 %5 = index.rems %a, %b24 // CHECK: llvm.urem25 %6 = index.remu %a, %b26 // CHECK: llvm.intr.smax27 %7 = index.maxs %a, %b28 // CHECK: llvm.intr.umax29 %8 = index.maxu %a, %b30 // CHECK: llvm.intr.smin31 %9 = index.mins %a, %b32 // CHECK: llvm.intr.umin33 %10 = index.minu %a, %b34 // CHECK: llvm.shl35 %11 = index.shl %a, %b36 // CHECK: llvm.ashr37 %12 = index.shrs %a, %b38 // CHECK: llvm.lshr39 %13 = index.shru %a, %b40 // CHECK: llvm.add41 %14 = index.add %a, %b42 // CHECK: llvm.or43 %15 = index.or %a, %b44 // CHECK: llvm.xor45 %16 = index.xor %a, %b46 // CHECK: llvm.mlir.constant(true47 %17 = index.bool.constant true48 return49}50 51// CHECK-LABEL: @ceildivs52// CHECK-SAME: %[[NI:.*]]: index, %[[MI:.*]]: index53func.func @ceildivs(%n: index, %m: index) -> index {54 // CHECK-DAG: %[[N:.*]] = builtin.unrealized_conversion_cast %[[NI]]55 // CHECK-DAG: %[[M:.*]] = builtin.unrealized_conversion_cast %[[MI]]56 // CHECK: %[[ZERO:.*]] = llvm.mlir.constant(0 :57 // CHECK: %[[POS_ONE:.*]] = llvm.mlir.constant(1 :58 // CHECK: %[[NEG_ONE:.*]] = llvm.mlir.constant(-1 :59 60 // CHECK: %[[M_POS:.*]] = llvm.icmp "sgt" %[[M]], %[[ZERO]]61 // CHECK: %[[X:.*]] = llvm.select %[[M_POS]], %[[NEG_ONE]], %[[POS_ONE]]62 63 // CHECK: %[[N_PLUS_X:.*]] = llvm.add %[[N]], %[[X]]64 // CHECK: %[[N_PLUS_X_DIV_M:.*]] = llvm.sdiv %[[N_PLUS_X]], %[[M]]65 // CHECK: %[[POS_RES:.*]] = llvm.add %[[N_PLUS_X_DIV_M]], %[[POS_ONE]]66 67 // CHECK: %[[NEG_N:.*]] = llvm.sub %[[ZERO]], %[[N]]68 // CHECK: %[[NEG_N_DIV_M:.*]] = llvm.sdiv %[[NEG_N]], %[[M]]69 // CHECK: %[[NEG_RES:.*]] = llvm.sub %[[ZERO]], %[[NEG_N_DIV_M]]70 71 // CHECK: %[[N_POS:.*]] = llvm.icmp "sgt" %[[N]], %[[ZERO]]72 // CHECK: %[[SAME_SIGN:.*]] = llvm.icmp "eq" %[[N_POS]], %[[M_POS]]73 // CHECK: %[[N_NON_ZERO:.*]] = llvm.icmp "ne" %[[N]], %[[ZERO]]74 // CHECK: %[[CMP:.*]] = llvm.and %[[SAME_SIGN]], %[[N_NON_ZERO]]75 // CHECK: %[[RESULT:.*]] = llvm.select %[[CMP]], %[[POS_RES]], %[[NEG_RES]]76 %result = index.ceildivs %n, %m77 78 // CHECK: %[[RESULTI:.*]] = builtin.unrealized_conversion_cast %[[RESULT]]79 // CHECK: return %[[RESULTI]]80 return %result : index81}82 83// CHECK-LABEL: @ceildivu84// CHECK-SAME: %[[NI:.*]]: index, %[[MI:.*]]: index85func.func @ceildivu(%n: index, %m: index) -> index {86 // CHECK-DAG: %[[N:.*]] = builtin.unrealized_conversion_cast %[[NI]]87 // CHECK-DAG: %[[M:.*]] = builtin.unrealized_conversion_cast %[[MI]]88 // CHECK: %[[ZERO:.*]] = llvm.mlir.constant(0 :89 // CHECK: %[[ONE:.*]] = llvm.mlir.constant(1 :90 91 // CHECK: %[[MINUS_ONE:.*]] = llvm.sub %[[N]], %[[ONE]]92 // CHECK: %[[QUOTIENT:.*]] = llvm.udiv %[[MINUS_ONE]], %[[M]]93 // CHECK: %[[PLUS_ONE:.*]] = llvm.add %[[QUOTIENT]], %[[ONE]]94 95 // CHECK: %[[CMP:.*]] = llvm.icmp "eq" %[[N]], %[[ZERO]]96 // CHECK: %[[RESULT:.*]] = llvm.select %[[CMP]], %[[ZERO]], %[[PLUS_ONE]]97 %result = index.ceildivu %n, %m98 99 // CHECK: %[[RESULTI:.*]] = builtin.unrealized_conversion_cast %[[RESULT]]100 // CHECK: return %[[RESULTI]]101 return %result : index102}103 104// CHECK-LABEL: @floordivs105// CHECK-SAME: %[[NI:.*]]: index, %[[MI:.*]]: index106func.func @floordivs(%n: index, %m: index) -> index {107 // CHECK-DAG: %[[N:.*]] = builtin.unrealized_conversion_cast %[[NI]]108 // CHECK-DAG: %[[M:.*]] = builtin.unrealized_conversion_cast %[[MI]]109 // CHECK-DAG: %[[ZERO:.*]] = llvm.mlir.constant(0 :110 // CHECK-DAG: %[[POS_ONE:.*]] = llvm.mlir.constant(1 :111 // CHECK-DAG: %[[NEG_ONE:.*]] = llvm.mlir.constant(-1 :112 113 // CHECK: %[[M_NEG:.*]] = llvm.icmp "slt" %[[M]], %[[ZERO]]114 // CHECK: %[[X:.*]] = llvm.select %[[M_NEG]], %[[POS_ONE]], %[[NEG_ONE]]115 116 // CHECK: %[[X_MINUS_N:.*]] = llvm.sub %[[X]], %[[N]]117 // CHECK: %[[X_MINUS_N_DIV_M:.*]] = llvm.sdiv %[[X_MINUS_N]], %[[M]]118 // CHECK: %[[NEG_RES:.*]] = llvm.sub %[[NEG_ONE]], %[[X_MINUS_N_DIV_M]]119 120 // CHECK: %[[POS_RES:.*]] = llvm.sdiv %[[N]], %[[M]]121 122 // CHECK: %[[N_NEG:.*]] = llvm.icmp "slt" %[[N]], %[[ZERO]]123 // CHECK: %[[DIFF_SIGN:.*]] = llvm.icmp "ne" %[[N_NEG]], %[[M_NEG]]124 // CHECK: %[[N_NON_ZERO:.*]] = llvm.icmp "ne" %[[N]], %[[ZERO]]125 // CHECK: %[[CMP:.*]] = llvm.and %[[DIFF_SIGN]], %[[N_NON_ZERO]]126 // CHECK: %[[RESULT:.*]] = llvm.select %[[CMP]], %[[NEG_RES]], %[[POS_RES]]127 %result = index.floordivs %n, %m128 129 // CHECK: %[[RESULTI:.*]] = builtin.unrealized_conversion_cast %[[RESULT]]130 // CHECK: return %[[RESULTI]]131 return %result : index132}133 134// INDEX32-LABEL: @index_cast_from135// INDEX64-LABEL: @index_cast_from136// INDEX32-SAME: %[[AI:.*]]: index137// INDEX64-SAME: %[[AI:.*]]: index138func.func @index_cast_from(%a: index) -> (i64, i32, i64, i32) {139 // INDEX32: %[[A:.*]] = builtin.unrealized_conversion_cast %[[AI]] : index to i32140 // INDEX64: %[[A:.*]] = builtin.unrealized_conversion_cast %[[AI]] : index to i64141 142 // INDEX32: %[[V0:.*]] = llvm.sext %[[A]] : i32 to i64143 %0 = index.casts %a : index to i64144 // INDEX64: %[[V1:.*]] = llvm.trunc %[[A]] : i64 to i32145 %1 = index.casts %a : index to i32146 // INDEX32: %[[V2:.*]] = llvm.zext %[[A]] : i32 to i64147 %2 = index.castu %a : index to i64148 // INDEX64: %[[V3:.*]] = llvm.trunc %[[A]] : i64 to i32149 %3 = index.castu %a : index to i32150 151 // INDEX32: return %[[V0]], %[[A]], %[[V2]], %[[A]]152 // INDEX64: return %[[A]], %[[V1]], %[[A]], %[[V3]]153 return %0, %1, %2, %3 : i64, i32, i64, i32154}155 156// INDEX32-LABEL: @index_cast_to157// INDEX64-LABEL: @index_cast_to158// INDEX32-SAME: %[[A:.*]]: i32, %[[B:.*]]: i64159// INDEX64-SAME: %[[A:.*]]: i32, %[[B:.*]]: i64160func.func @index_cast_to(%a: i32, %b: i64) -> (index, index, index, index) {161 // INDEX64: %[[V0:.*]] = llvm.sext %[[A]] : i32 to i64162 %0 = index.casts %a : i32 to index163 // INDEX32: %[[V1:.*]] = llvm.trunc %[[B]] : i64 to i32164 %1 = index.casts %b : i64 to index165 // INDEX64: %[[V2:.*]] = llvm.zext %[[A]] : i32 to i64166 %2 = index.castu %a : i32 to index167 // INDEX32: %[[V3:.*]] = llvm.trunc %[[B]] : i64 to i32168 %3 = index.castu %b : i64 to index169 return %0, %1, %2, %3 : index, index, index, index170}171 172// INDEX32-LABEL: @index_sizeof173// INDEX64-LABEL: @index_sizeof174func.func @index_sizeof() {175 // INDEX32-NEXT: llvm.mlir.constant(32 : i32)176 // INDEX64-NEXT: llvm.mlir.constant(64 : i64)177 %0 = index.sizeof178 return179}180 181// INDEX32-LABEL: @index_constant182// INDEX64-LABEL: @index_constant183func.func @index_constant() {184 // INDEX32: llvm.mlir.constant(-2100000000 : i32) : i32185 // INDEX64: llvm.mlir.constant(-2100000000 : i64) : i64186 %0 = index.constant -2100000000187 // INDEX32: llvm.mlir.constant(2100000000 : i32) : i32188 // INDEX64: llvm.mlir.constant(2100000000 : i64) : i64189 %1 = index.constant 2100000000190 // INDEX32: llvm.mlir.constant(1294967296 : i32) : i32191 // INDEX64: llvm.mlir.constant(-3000000000 : i64) : i64192 %2 = index.constant -3000000000193 // INDEX32: llvm.mlir.constant(-1294967296 : i32) : i32194 // INDEX64: llvm.mlir.constant(3000000000 : i64) : i64195 %3 = index.constant 3000000000196 return197}198