[llvm] r336227 - [NVPTX] Expand v2f16 INSERT_VECTOR_ELT

[Benjamin Kramer via llvm-commits]

Author: d0k
Date: Tue Jul  3 13:40:04 2018
New Revision: 336227

URL: http://llvm.org/viewvc/llvm-project?rev=336227&view=rev
Log:
[NVPTX] Expand v2f16 INSERT_VECTOR_ELT

Vectorization can create them.

Modified:
    llvm/trunk/lib/Target/NVPTX/NVPTXISelLowering.cpp
    llvm/trunk/test/CodeGen/NVPTX/f16x2-instructions.ll

Modified: llvm/trunk/lib/Target/NVPTX/NVPTXISelLowering.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/Target/NVPTX/NVPTXISelLowering.cpp?rev=336227&r1=336226&r2=336227&view=diff
==============================================================================
--- llvm/trunk/lib/Target/NVPTX/NVPTXISelLowering.cpp (original)
+++ llvm/trunk/lib/Target/NVPTX/NVPTXISelLowering.cpp Tue Jul  3 13:40:04 2018
@@ -375,6 +375,7 @@ NVPTXTargetLowering::NVPTXTargetLowering
   setOperationAction(ISD::FP_TO_SINT, MVT::f16, Legal);
   setOperationAction(ISD::BUILD_VECTOR, MVT::v2f16, Custom);
   setOperationAction(ISD::EXTRACT_VECTOR_ELT, MVT::v2f16, Custom);
+  setOperationAction(ISD::INSERT_VECTOR_ELT, MVT::v2f16, Expand);
   setOperationAction(ISD::VECTOR_SHUFFLE, MVT::v2f16, Expand);
 
   setFP16OperationAction(ISD::SETCC, MVT::f16, Legal, Promote);

Modified: llvm/trunk/test/CodeGen/NVPTX/f16x2-instructions.ll
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/CodeGen/NVPTX/f16x2-instructions.ll?rev=336227&r1=336226&r2=336227&view=diff
==============================================================================
--- llvm/trunk/test/CodeGen/NVPTX/f16x2-instructions.ll (original)
+++ llvm/trunk/test/CodeGen/NVPTX/f16x2-instructions.ll Tue Jul  3 13:40:04 2018
@@ -1431,5 +1431,13 @@ define <2 x half> @test_shufflevector(<2
   ret <2 x half> %s
 }
 
+; CHECK-LABEL: test_insertelement(
+; CHECK: mov.b32 {%h2, %tmp_hi}, %hh1;
+; CHECK: mov.b32 %hh2, {%h2, %h1};
+define <2 x half> @test_insertelement(<2 x half> %a, half %x) #0 {
+  %i = insertelement <2 x half> %a, half %x, i64 1
+  ret <2 x half> %i
+}
+
 attributes #0 = { nounwind }
 attributes #1 = { "unsafe-fp-math" = "true" }