summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp8
-rw-r--r--llvm/test/CodeGen/WebAssembly/legalize.ll24
2 files changed, 31 insertions, 1 deletions
diff --git a/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp b/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp
index a8b93ca8a2f..37b82d6385a 100644
--- a/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp
+++ b/llvm/lib/Target/WebAssembly/WebAssemblyISelLowering.cpp
@@ -202,7 +202,13 @@ bool WebAssemblyTargetLowering::isOffsetFoldingLegal(
MVT WebAssemblyTargetLowering::getScalarShiftAmountTy(const DataLayout & /*DL*/,
EVT VT) const {
- return VT.getSimpleVT();
+ unsigned BitWidth = NextPowerOf2(VT.getSizeInBits() - 1);
+ if (BitWidth > 1 && BitWidth < 8)
+ BitWidth = 8;
+ MVT Result = MVT::getIntegerVT(BitWidth);
+ assert(Result != MVT::INVALID_SIMPLE_VALUE_TYPE &&
+ "Unable to represent scalar shift amount type");
+ return Result;
}
const char *
diff --git a/llvm/test/CodeGen/WebAssembly/legalize.ll b/llvm/test/CodeGen/WebAssembly/legalize.ll
new file mode 100644
index 00000000000..4bfec212cf7
--- /dev/null
+++ b/llvm/test/CodeGen/WebAssembly/legalize.ll
@@ -0,0 +1,24 @@
+; RUN: llc < %s -asm-verbose=false | FileCheck %s
+
+; Test various types and operators that need to be legalized.
+
+target datalayout = "e-p:32:32-i64:64-n32:64-S128"
+target triple = "wasm32-unknown-unknown"
+
+; CHECK-LABEL: shl_i3:
+; CHECK: i32.const $push0=, 7
+; CHECK: i32.and $push1=, $1, $pop0
+; CHECK: i32.shl $push2=, $0, $pop1
+define i3 @shl_i3(i3 %a, i3 %b, i3* %p) {
+ %t = shl i3 %a, %b
+ ret i3 %t
+}
+
+; CHECK-LABEL: shl_i53:
+; CHECK: i64.const $push0=, 9007199254740991
+; CHECK: i64.and $push1=, $1, $pop0
+; CHECK: i64.shl $push2=, $0, $pop1
+define i53 @shl_i53(i53 %a, i53 %b, i53* %p) {
+ %t = shl i53 %a, %b
+ ret i53 %t
+}
OpenPOWER on IntegriCloud