commit 96e47014010440adbc700e66c68e566236dfc2dd
author Matt Arsenault <Matthew.Arsenault@amd.com> Fri Jan 18 21:42:55 2019 +0000
committer Matt Arsenault <Matthew.Arsenault@amd.com> Fri Jan 18 21:42:55 2019 +0000
tree 716ab2643502c437df16d022fe1e7cbb29b6ac3b
parent 86ac5326879a9e1ab8b92e28c02c0ba5aed880e8

AMDGPU/GlobalISel: Legalize more types for select

llvm-svn: 351599

llvm/lib/Target/AMDGPU/AMDGPULegalizerInfo.cpp

diff --git a/llvm/lib/Target/AMDGPU/AMDGPULegalizerInfo.cpp b/llvm/lib/Target/AMDGPU/AMDGPULegalizerInfo.cpp
index b73386e..d160c7c 100644
--- a/llvm/lib/Target/AMDGPU/AMDGPULegalizerInfo.cpp
+++ b/llvm/lib/Target/AMDGPU/AMDGPULegalizerInfo.cpp
@@ -245,8 +245,10 @@
     Atomics.legalFor({{S32, FlatPtr}, {S64, FlatPtr}});
   }
 
-  setAction({G_SELECT, S32}, Legal);
-  setAction({G_SELECT, 1, S1}, Legal);
+  // TODO: Pointer types, any 32-bit or 64-bit vector
+  getActionDefinitionsBuilder(G_SELECT)
+    .legalFor({{S32, S1}, {S64, S1}, {V2S32, S1}, {V2S16, S1}})
+    .clampScalar(0, S32, S64);
 
   setAction({G_SHL, S32}, Legal);