[llvm] r181186 - Fix slightly too aggressive conact_vector optimization.

[Michael Kuperstein]

Author: mkuper
Date: Mon May  6 03:06:13 2013
New Revision: 181186

URL: http://llvm.org/viewvc/llvm-project?rev=181186&view=rev
Log:
Fix slightly too aggressive conact_vector optimization.
(Would sometimes optimize away conacts used to extend a vector with undef values)

Added:
    llvm/trunk/test/CodeGen/X86/2013-05-06-ConactVectorCrash.ll
Modified:
    llvm/trunk/lib/CodeGen/SelectionDAG/DAGCombiner.cpp

Modified: llvm/trunk/lib/CodeGen/SelectionDAG/DAGCombiner.cpp
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/lib/CodeGen/SelectionDAG/DAGCombiner.cpp?rev=181186&r1=181185&r2=181186&view=diff
==============================================================================
--- llvm/trunk/lib/CodeGen/SelectionDAG/DAGCombiner.cpp (original)
+++ llvm/trunk/lib/CodeGen/SelectionDAG/DAGCombiner.cpp Mon May  6 03:06:13 2013
@@ -9145,6 +9145,12 @@ SDValue DAGCombiner::visitCONCAT_VECTORS
         return SDValue();
     } else {
       SingleSource = Op.getOperand(0);
+
+      // Check the source type is the same as the type of the result.
+      // If not, this concat may extend the vector, so we can not
+      // optimize it away.
+      if (SingleSource.getValueType() != N->getValueType(0))
+        return SDValue();
     }
 
     unsigned IdentityIndex = i * PartNumElem;

Added: llvm/trunk/test/CodeGen/X86/2013-05-06-ConactVectorCrash.ll
URL: http://llvm.org/viewvc/llvm-project/llvm/trunk/test/CodeGen/X86/2013-05-06-ConactVectorCrash.ll?rev=181186&view=auto
==============================================================================
--- llvm/trunk/test/CodeGen/X86/2013-05-06-ConactVectorCrash.ll (added)
+++ llvm/trunk/test/CodeGen/X86/2013-05-06-ConactVectorCrash.ll Mon May  6 03:06:13 2013
@@ -0,0 +1,14 @@
+; RUN: llc < %s -march=x86
+
+; Make sure this doesn't crash
+
+target datalayout = "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-f32:32:32-f64:64:64-v64:64:64-v128:128:128-a0:0:64-s0:64:64-f80:128:128-n8:16:32:64-S128"
+target triple = "x86_64-pc-win32"
+
+define void @foo() {
+  %1 = shufflevector <3 x i8> undef, <3 x i8> undef, <2 x i32> <i32 0, i32 1>
+  %2 = shufflevector <2 x i8> %1, <2 x i8> undef, <4 x i32> <i32 0, i32 1, i32 undef, i32 undef>
+  %3 = shufflevector <4 x i8> undef, <4 x i8> %2, <4 x i32> <i32 0, i32 1, i32 4, i32 5>
+  store <4 x i8> %3, <4 x i8>* undef
+  ret void
+}