1 files changed, 5 insertions, 2 deletions
diff --git a/llvm/lib/Target/AArch64/AArch64MachineLegalizer.cpp b/llvm/lib/Target/AArch64/AArch64MachineLegalizer.cpp
index b91d6040bb6..d5bf1dbf386 100644
--- a/llvm/lib/Target/AArch64/AArch64MachineLegalizer.cpp
+++ b/llvm/lib/Target/AArch64/AArch64MachineLegalizer.cpp
@@ -51,10 +51,13 @@ AArch64MachineLegalizer::AArch64MachineLegalizer() {
     for (auto Ty : {s32, s64})
       setAction(BinOp, Ty, Legal);
 
-  for (auto MemOp : {G_LOAD, G_STORE})
-    for (auto Ty : {s32, s64})
+  for (auto MemOp : {G_LOAD, G_STORE}) {
+    for (auto Ty : {s8, s16, s32, s64})
       setAction(MemOp, Ty, Legal);
 
+    setAction(MemOp, s1, WidenScalar);
+  }
+
   for (auto Ty : {s32, s64}) {
     setAction(TargetOpcode::G_CONSTANT, Ty, Legal);
     setAction(TargetOpcode::G_FCONSTANT, Ty, Legal);