[PATCH] D41126: [SelectionDAG] Fixed f16-from-vector promotion problem

[Tim Renouf via Phabricator via llvm-commits]

tpr updated this revision to Diff 127248.
tpr added a comment.

Another test tidy-up.


https://reviews.llvm.org/D41126

Files:
  lib/CodeGen/SelectionDAG/LegalizeVectorTypes.cpp
  test/CodeGen/AMDGPU/unpack-half.ll


Index: test/CodeGen/AMDGPU/unpack-half.ll
===================================================================
--- /dev/null
+++ test/CodeGen/AMDGPU/unpack-half.ll
@@ -0,0 +1,26 @@
+; RUN: llc -march=amdgcn -mcpu=gfx600 -verify-machineinstrs < %s | FileCheck %s
+; RUN: llc -march=amdgcn -mcpu=gfx700 -verify-machineinstrs < %s | FileCheck %s
+
+; On gfx6 and gfx7, this test shows a bug in SelectionDAG where scalarizing the
+; extension of a vector of f16 generates an illegal node that errors later.
+
+; CHECK-LABEL: {{^}}main:
+; CHECK: v_cvt_f32_f16
+
+define amdgpu_gs void @main(i32 inreg %arg) local_unnamed_addr #0 {
+.entry:
+  %tmp = load volatile float, float addrspace(1)* undef
+  %tmp1 = bitcast float %tmp to i32
+  %im0.i = lshr i32 %tmp1, 16
+  %tmp2 = insertelement <2 x i32> undef, i32 %im0.i, i32 1
+  %tmp3 = trunc <2 x i32> %tmp2 to <2 x i16>
+  %tmp4 = bitcast <2 x i16> %tmp3 to <2 x half>
+  %tmp5 = fpext <2 x half> %tmp4 to <2 x float>
+  %bc = bitcast <2 x float> %tmp5 to <2 x i32>
+  %tmp6 = extractelement <2 x i32> %bc, i32 1
+  store volatile i32 %tmp6, i32 addrspace(1)* undef
+  ret void
+}
+
+attributes #0 = { nounwind }
+
Index: lib/CodeGen/SelectionDAG/LegalizeVectorTypes.cpp
===================================================================
--- lib/CodeGen/SelectionDAG/LegalizeVectorTypes.cpp
+++ lib/CodeGen/SelectionDAG/LegalizeVectorTypes.cpp
@@ -169,9 +169,13 @@
 }
 
 SDValue DAGTypeLegalizer::ScalarizeVecRes_BITCAST(SDNode *N) {
+  SDValue Op = N->getOperand(0);
+  if (Op.getValueType().isVector()
+      && Op.getValueType().getVectorNumElements() == 1)
+    Op = GetScalarizedVector(Op);
   EVT NewVT = N->getValueType(0).getVectorElementType();
   return DAG.getNode(ISD::BITCAST, SDLoc(N),
-                     NewVT, N->getOperand(0));
+                     NewVT, Op);
 }
 
 SDValue DAGTypeLegalizer::ScalarizeVecRes_BUILD_VECTOR(SDNode *N) {


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D41126.127248.patch
Type: text/x-patch
Size: 1903 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20171216/942a67ea/attachment.bin>