[PATCH] D105385: [AMDGPU] Add SelectionDAG support for insert_subvector on v4f64

[Carl Ritson via Phabricator via llvm-commits]

This revision was automatically updated to reflect the committed changes.
Closed by commit rGfbaa35e16947: [AMDGPU] Add SelectionDAG support for insert_subvector on v4f64 (authored by critson).

Repository:
  rG LLVM Github Monorepo

CHANGES SINCE LAST ACTION
  https://reviews.llvm.org/D105385/new/

https://reviews.llvm.org/D105385

Files:
  llvm/lib/Target/AMDGPU/SIISelLowering.cpp
  llvm/test/CodeGen/AMDGPU/bug-v4f64-subvector.ll


Index: llvm/test/CodeGen/AMDGPU/bug-v4f64-subvector.ll
===================================================================
--- /dev/null
+++ llvm/test/CodeGen/AMDGPU/bug-v4f64-subvector.ll
@@ -0,0 +1,68 @@
+; RUN: llc < %s -mtriple=amdgcn-amd-amdhsa -mcpu=gfx900 -start-before=amdgpu-isel -stop-after=amdgpu-isel -verify-machineinstrs | FileCheck %s --check-prefixes=CHECK
+
+; This caused failure in infinite cycle in Selection DAG (combine) due to missing insert_subvector.
+;
+; CHECK-LABEL: name: test1
+; CHECK: GLOBAL_LOAD_DWORDX4
+; CHECK: GLOBAL_LOAD_DWORDX4
+; CHECK: GLOBAL_STORE_DWORDX4
+define protected amdgpu_kernel void @test1() local_unnamed_addr !kernel_arg_addr_space !0 !kernel_arg_access_qual !1 !kernel_arg_type !2 !kernel_arg_base_type !2 !kernel_arg_type_qual !3 !kernel_arg_name !4 {
+entry:
+  %tmp = load <3 x i64>, <3 x i64> addrspace(4)* undef, align 16, !invariant.load !5
+  %srcA.load2 = extractelement <3 x i64> %tmp, i32 0
+  %tmp1 = inttoptr i64 %srcA.load2 to double addrspace(1)*
+  %tmp2 = getelementptr inbounds double, double addrspace(1)* %tmp1, i64 undef
+  %tmp3 = bitcast double addrspace(1)* %tmp2 to <3 x double> addrspace(1)*
+  %tmp4 = load <3 x double>, <3 x double> addrspace(1)* %tmp3, align 8, !tbaa !6
+  %tmp5 = extractelement <3 x double> %tmp4, i32 1
+  %tmp6 = insertelement <3 x double> undef, double %tmp5, i32 1
+  %tmp7 = insertelement <3 x double> %tmp6, double undef, i32 2
+  %tmp8 = load <3 x double>, <3 x double> addrspace(1)* undef, align 8, !tbaa !6
+  %tmp9 = extractelement <3 x double> %tmp8, i32 2
+  %tmp10 = insertelement <3 x double> undef, double %tmp9, i32 2
+  %tmp11 = fcmp olt <3 x double> %tmp10, %tmp7
+  %tmp12 = select <3 x i1> %tmp11, <3 x double> zeroinitializer, <3 x double> <double 1.000000e+00, double 1.000000e+00, double 1.000000e+00>
+  %tmp13 = extractelement <3 x double> %tmp12, i64 1
+  %tmp14 = insertelement <2 x double> undef, double %tmp13, i32 1
+  store <2 x double> %tmp14, <2 x double> addrspace(1)* undef, align 8, !tbaa !6
+  ret void
+}
+
+; This caused failure in Selection DAG due to lack of insert_subvector implementation.
+;
+; CHECK-LABEL: name: test2
+; CHECK: GLOBAL_LOAD_DWORDX2
+; CHECK: GLOBAL_LOAD_DWORDX2
+; CHECK: GLOBAL_STORE_DWORDX2
+define protected amdgpu_kernel void @test2() local_unnamed_addr !kernel_arg_addr_space !0 !kernel_arg_access_qual !1 !kernel_arg_type !2 !kernel_arg_base_type !2 !kernel_arg_type_qual !3 !kernel_arg_name !4 {
+entry:
+  %tmp = load <3 x i64>, <3 x i64> addrspace(4)* undef, align 16, !invariant.load !5
+  %srcA.load2 = extractelement <3 x i64> %tmp, i32 0
+  %tmp1 = inttoptr i64 %srcA.load2 to double addrspace(1)*
+  %tmp2 = getelementptr inbounds double, double addrspace(1)* %tmp1, i64 undef
+  %tmp3 = bitcast double addrspace(1)* %tmp2 to <3 x double> addrspace(1)*
+  %tmp4 = load <3 x double>, <3 x double> addrspace(1)* %tmp3, align 8, !tbaa !6
+  %tmp5 = extractelement <3 x double> %tmp4, i32 1
+  %tmp6 = insertelement <3 x double> undef, double %tmp5, i32 1
+  %tmp7 = insertelement <3 x double> %tmp6, double undef, i32 2
+  %tmp8 = load <3 x double>, <3 x double> addrspace(1)* undef, align 8, !tbaa !6
+  %tmp9 = extractelement <3 x double> %tmp8, i32 1
+  %tmp10 = insertelement <3 x double> undef, double %tmp9, i32 1
+  %tmp11 = insertelement <3 x double> %tmp10, double undef, i32 2
+  %tmp12 = fcmp olt <3 x double> %tmp11, %tmp7
+  %tmp13 = select <3 x i1> %tmp12, <3 x double> zeroinitializer, <3 x double> <double 1.000000e+00, double 1.000000e+00, double 1.000000e+00>
+  %tmp14 = extractelement <3 x double> %tmp13, i64 2
+  store double %tmp14, double addrspace(1)* undef, align 8, !tbaa !6
+  ret void
+}
+
+!0 = !{i32 1, i32 1, i32 1}
+!1 = !{!"none", !"none", !"none"}
+!2 = !{!"double*", !"double*", !"double*"}
+!3 = !{!"", !"", !""}
+!4 = !{!"srcA", !"srcB", !"dst"}
+!5 = !{}
+!6 = !{!7, !7, i64 0}
+!7 = !{!"double", !8, i64 0}
+!8 = !{!"omnipotent char", !9, i64 0}
+!9 = !{!"Simple C/C++ TBAA"}
Index: llvm/lib/Target/AMDGPU/SIISelLowering.cpp
===================================================================
--- llvm/lib/Target/AMDGPU/SIISelLowering.cpp
+++ llvm/lib/Target/AMDGPU/SIISelLowering.cpp
@@ -279,10 +279,10 @@
       case ISD::BITCAST:
       case ISD::EXTRACT_VECTOR_ELT:
       case ISD::INSERT_VECTOR_ELT:
-      case ISD::INSERT_SUBVECTOR:
       case ISD::EXTRACT_SUBVECTOR:
       case ISD::SCALAR_TO_VECTOR:
         break;
+      case ISD::INSERT_SUBVECTOR:
       case ISD::CONCAT_VECTORS:
         setOperationAction(Op, VT, Custom);
         break;


-------------- next part --------------
A non-text attachment was scrubbed...
Name: D105385.361876.patch
Type: text/x-patch
Size: 4572 bytes
Desc: not available
URL: <http://lists.llvm.org/pipermail/llvm-commits/attachments/20210727/dce31d01/attachment.bin>