commit 434d664095b2417453c9e86943c9a94ae1d4bd47
author Matt Arsenault <Matthew.Arsenault@amd.com> Mon Jul 15 19:37:34 2019 +0000
committer Matt Arsenault <Matthew.Arsenault@amd.com> Mon Jul 15 19:37:34 2019 +0000
tree a8bc7f992003cd539df3b85258d18c188d88488b
parent ba4373ea7d9c816232d44061230f5dca9582b4b7

GlobalISel: Implement narrowScalar for vector extract/insert indexes

llvm-svn: 366113

llvm/lib/CodeGen/GlobalISel/LegalizerHelper.cpp

diff --git a/llvm/lib/CodeGen/GlobalISel/LegalizerHelper.cpp b/llvm/lib/CodeGen/GlobalISel/LegalizerHelper.cpp
index d13b1a5..bf3cca4 100644
--- a/llvm/lib/CodeGen/GlobalISel/LegalizerHelper.cpp
+++ b/llvm/lib/CodeGen/GlobalISel/LegalizerHelper.cpp
@@ -736,6 +736,17 @@
     MI.eraseFromParent();
     return Legalized;
   }
+  case TargetOpcode::G_EXTRACT_VECTOR_ELT:
+  case TargetOpcode::G_INSERT_VECTOR_ELT: {
+    if (TypeIdx != 2)
+      return UnableToLegalize;
+
+    int OpIdx = MI.getOpcode() == TargetOpcode::G_EXTRACT_VECTOR_ELT ? 2 : 3;
+    Observer.changingInstr(MI);
+    narrowScalarSrc(MI, NarrowTy, OpIdx);
+    Observer.changedInstr(MI);
+    return Legalized;
+  }
   }
 }