; RUN: llc < %s | FileCheck --check-prefix AS %s ; RUN: opt -S -interleaved-load-combine < %s | FileCheck %s ; RUN: opt -S -passes=interleaved-load-combine < %s | FileCheck %s ; ModuleID = 'aarch64_interleaved-ld-combine.bc' target datalayout = "e-m:e-i64:64-i128:128-n32:64-S128" target triple = "arm64--linux-gnu" ; This should be lowered into LD4 define void @aarch64_ilc_const(ptr %ptr) { entry: ;;; Check LLVM transformation ; CHECK-LABEL: @aarch64_ilc_const( ; CHECK-DAG: [[GEP:%.+]] = getelementptr inbounds <4 x float>, ptr %ptr, i64 2 ; CHECK-DAG: [[LOAD:%.+]] = load <16 x float>, ptr [[GEP]], align 16 ; CHECK-DAG: %{{.* }}= shufflevector <16 x float> [[LOAD]], <16 x float> poison, <4 x i32> ; CHECK-DAG: %{{.* }}= shufflevector <16 x float> [[LOAD]], <16 x float> poison, <4 x i32> ; CHECK-DAG: %{{.* }}= shufflevector <16 x float> [[LOAD]], <16 x float> poison, <4 x i32> ; CHECK-DAG: %{{.* }}= shufflevector <16 x float> [[LOAD]], <16 x float> poison, <4 x i32> ; CHECK: ret void ;;; Check if it gets lowerd ; AS-LABEL: aarch64_ilc_const ; AS: ld4 ; AS: ret %gep1 = getelementptr inbounds <4 x float>, ptr %ptr, i64 2 %gep2 = getelementptr inbounds <4 x float>, ptr %ptr, i64 3 %gep3 = getelementptr inbounds <4 x float>, ptr %ptr, i64 4 %gep4 = getelementptr inbounds <4 x float>, ptr %ptr, i64 5 %ld1 = load <4 x float>, ptr %gep1, align 16 %ld2 = load <4 x float>, ptr %gep2, align 16 %ld3 = load <4 x float>, ptr %gep3, align 16 %ld4 = load <4 x float>, ptr %gep4, align 16 %sv1 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> %sv2 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> %sv3 = shufflevector <4 x float> %ld3, <4 x float> %ld4, <4 x i32> %sv4 = shufflevector <4 x float> %ld3, <4 x float> %ld4, <4 x i32> %m0_3 = shufflevector <4 x float> %sv1, <4 x float> %sv3, <4 x i32> %m4_7 = shufflevector <4 x float> %sv1, <4 x float> %sv3, <4 x i32> %m8_11 = shufflevector <4 x float> %sv2, <4 x float> %sv4, <4 x i32> %m12_15 = shufflevector <4 x float> %sv2, <4 x float> %sv4, <4 x i32> store <4 x float> %m0_3, ptr %gep1, align 16 store <4 x float> %m4_7, ptr %gep2, align 16 store <4 x float> %m8_11, ptr %gep3, align 16 store <4 x float> %m12_15, ptr %gep4, align 16 ret void } ; This should be lowered into LD4 define void @aarch64_ilc_idx(ptr %ptr, i64 %idx) { entry: ;;; Check LLVM transformation ; CHECK-LABEL: @aarch64_ilc_idx( ; CHECK-DAG: [[ADD:%.+]] = add i64 %idx, 16 ; CHECK-DAG: [[LSHR:%.+]] = lshr i64 [[ADD]], 2 ; CHECK-DAG: [[GEP:%.+]] = getelementptr inbounds <4 x float>, ptr %ptr, i64 [[LSHR]] ; CHECK-DAG: [[LOAD:%.+]] = load <16 x float>, ptr [[GEP]], align 16 ; CHECK-DAG: %{{.* }}= shufflevector <16 x float> [[LOAD]], <16 x float> poison, <4 x i32> ; CHECK-DAG: %{{.* }}= shufflevector <16 x float> [[LOAD]], <16 x float> poison, <4 x i32> ; CHECK-DAG: %{{.* }}= shufflevector <16 x float> [[LOAD]], <16 x float> poison, <4 x i32> ; CHECK-DAG: %{{.* }}= shufflevector <16 x float> [[LOAD]], <16 x float> poison, <4 x i32> ; CHECK: ret void ; AS-LABEL: aarch64_ilc_idx ; AS-DAG: lsl [[LSL:x[0-9]+]], x1, #2 ; AS-DAG: add [[ADD:x[0-9]+]], [[LSL]], #64 ; AS-DAG: and [[AND:x[0-9]+]], [[ADD]], #0xfffffffffffffff0 ; AS-DAG: add [[ADR:x[0-9]+]], x0, [[AND]] ; AS-DAG: ld4 { v[[V0:[0-9]+]].4s, v[[V1:[0-9]+]].4s, v[[V2:[0-9]+]].4s, v[[V3:[0-9]+]].4s }, [[[ADR]]] ; AS-DAG: str q[[V0]] ; AS-DAG: str q[[V1]] ; AS-DAG: str q[[V2]] ; AS-DAG: str q[[V3]] ; AS: ret %a2 = add i64 %idx, 20 %idx2 = lshr i64 %a2, 2 %a3 = add i64 %idx, 24 %a1 = add i64 %idx, 16 %idx1 = lshr i64 %a1, 2 %idx3 = lshr i64 %a3, 2 %a4 = add i64 %idx, 28 %idx4 = lshr i64 %a4, 2 %gep2 = getelementptr inbounds <4 x float>, ptr %ptr, i64 %idx2 %gep4 = getelementptr inbounds <4 x float>, ptr %ptr, i64 %idx4 %gep1 = getelementptr inbounds <4 x float>, ptr %ptr, i64 %idx1 %gep3 = getelementptr inbounds <4 x float>, ptr %ptr, i64 %idx3 %ld1 = load <4 x float>, ptr %gep1, align 16 %ld2 = load <4 x float>, ptr %gep2, align 16 %ld3 = load <4 x float>, ptr %gep3, align 16 %ld4 = load <4 x float>, ptr %gep4, align 16 %sv1 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> %sv2 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> %sv3 = shufflevector <4 x float> %ld3, <4 x float> %ld4, <4 x i32> %sv4 = shufflevector <4 x float> %ld3, <4 x float> %ld4, <4 x i32> %m0_3 = shufflevector <4 x float> %sv1, <4 x float> %sv3, <4 x i32> %m4_7 = shufflevector <4 x float> %sv1, <4 x float> %sv3, <4 x i32> %m8_11 = shufflevector <4 x float> %sv2, <4 x float> %sv4, <4 x i32> %m12_15 = shufflevector <4 x float> %sv2, <4 x float> %sv4, <4 x i32> store <4 x float> %m0_3, ptr %gep1, align 16 store <4 x float> %m4_7, ptr %gep2, align 16 store <4 x float> %m8_11, ptr %gep3, align 16 store <4 x float> %m12_15, ptr %gep4, align 16 ret void } ; This should be lowered into LD4, a offset of has to be taken into account %struct.ilc = type <{ float, [0 x <4 x float>] }> define void @aarch64_ilc_struct(ptr %ptr, i64 %idx) { entry: ;;; Check LLVM transformation ; CHECK-LABEL: @aarch64_ilc_struct( ; CHECK-DAG: [[LSHR:%.+]] = lshr i64 %idx, 2 ; CHECK-DAG: [[GEP:%.+]] = getelementptr %struct.ilc, ptr %ptr, i32 0, i32 1, i64 [[LSHR]] ; CHECK-DAG: [[LOAD:%.+]] = load <16 x float>, ptr [[GEP]], align 4 ; CHECK-DAG: %{{.* }}= shufflevector <16 x float> [[LOAD]], <16 x float> poison, <4 x i32> ; CHECK-DAG: %{{.* }}= shufflevector <16 x float> [[LOAD]], <16 x float> poison, <4 x i32> ; CHECK-DAG: %{{.* }}= shufflevector <16 x float> [[LOAD]], <16 x float> poison, <4 x i32> ; CHECK-DAG: %{{.* }}= shufflevector <16 x float> [[LOAD]], <16 x float> poison, <4 x i32> ; CHECK: ret void ; AS-LABEL: aarch64_ilc_struct ; AS-DAG: lsl [[LSL:x[0-9]+]], x1, #2 ; AS-DAG: add [[ADD:x[0-9]+]], x0, #4 ; AS-DAG: and [[AND:x[0-9]+]], [[LSL]], #0xfffffffffffffff0 ; AS-DAG: add [[ADR:x[0-9]+]], [[ADD]], [[AND]] ; AS-DAG: ld4 { v[[V0:[0-9]+]].4s, v[[V1:[0-9]+]].4s, v[[V2:[0-9]+]].4s, v[[V3:[0-9]+]].4s }, [[[ADR]]] ; AS-DAG: str q[[V0]] ; AS-DAG: str q[[V1]] ; AS-DAG: str q[[V2]] ; AS-DAG: str q[[V3]] ; AS: ret %a1 = add i64 %idx, 4 %idx2 = lshr i64 %a1, 2 %a2 = add i64 %idx, 8 %idx3 = lshr i64 %a2, 2 %a3 = add i64 %idx, 12 %idx4 = lshr i64 %a3, 2 %gep2 = getelementptr %struct.ilc, ptr %ptr, i32 0, i32 1, i64 %idx2 %gep3 = getelementptr %struct.ilc, ptr %ptr, i32 0, i32 1, i64 %idx3 %gep4 = getelementptr %struct.ilc, ptr %ptr, i32 0, i32 1, i64 %idx4 %idx1 = lshr i64 %idx, 2 %gep1 = getelementptr %struct.ilc, ptr %ptr, i32 0, i32 1, i64 %idx1 %ld1 = load <4 x float>, ptr %gep1, align 4 %ld2 = load <4 x float>, ptr %gep2, align 4 %ld3 = load <4 x float>, ptr %gep3, align 4 %ld4 = load <4 x float>, ptr %gep4, align 4 %sv1 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> %sv2 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> %sv3 = shufflevector <4 x float> %ld3, <4 x float> %ld4, <4 x i32> %sv4 = shufflevector <4 x float> %ld3, <4 x float> %ld4, <4 x i32> %m0_3 = shufflevector <4 x float> %sv1, <4 x float> %sv3, <4 x i32> %m4_7 = shufflevector <4 x float> %sv1, <4 x float> %sv3, <4 x i32> %m8_11 = shufflevector <4 x float> %sv2, <4 x float> %sv4, <4 x i32> %m12_15 = shufflevector <4 x float> %sv2, <4 x float> %sv4, <4 x i32> store <4 x float> %m0_3, ptr %gep1, align 16 store <4 x float> %m4_7, ptr %gep2, align 16 store <4 x float> %m8_11, ptr %gep3, align 16 store <4 x float> %m12_15, ptr %gep4, align 16 ret void } ; This should be lowered into LD2 define void @aarch64_ilc_idx_ld2(ptr %ptr, i64 %idx) { entry: ; CHECK-LABEL: @aarch64_ilc_idx_ld2( ; CHECK-DAG: [[LSHR:%.+]] = lshr i64 %idx, 2 ; CHECK-DAG: [[GEP:%.+]] = getelementptr inbounds <4 x float>, ptr %ptr, i64 [[LSHR]] ; CHECK-DAG: [[LOAD:%.+]] = load <8 x float>, ptr [[GEP]], align 16 ; CHECK: %{{.* }}= shufflevector <8 x float> [[LOAD]], <8 x float> poison, <4 x i32> ; CHECK: %{{.* }}= shufflevector <8 x float> [[LOAD]], <8 x float> poison, <4 x i32> ; CHECK-DAG: ret void ; AS-LABEL: aarch64_ilc_idx_ld2 ; AS: ld2 ; AS: ret %idx1 = lshr i64 %idx, 2 %a1 = add i64 %idx, 4 %idx2 = lshr i64 %a1, 2 %gep1 = getelementptr inbounds <4 x float>, ptr %ptr, i64 %idx1 %gep2 = getelementptr inbounds <4 x float>, ptr %ptr, i64 %idx2 %ld1 = load <4 x float>, ptr %gep1, align 16 %ld2 = load <4 x float>, ptr %gep2, align 16 %m0_3 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> %m4_7 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> store <4 x float> %m0_3, ptr %gep1 store <4 x float> %m4_7, ptr %gep2 ret void } ; This should be lowered into LD3 define void @aarch64_ilc_idx_ld3(ptr %ptr, i64 %idx) { entry: ; CHECK-LABEL: @aarch64_ilc_idx_ld3( ; CHECK-DAG: [[LSHR:%.+]] = lshr i64 %idx, 2 ; CHECK-DAG: [[GEP:%.+]] = getelementptr inbounds <4 x float>, ptr %ptr, i64 [[LSHR]] ; CHECK-DAG: [[LOAD:%.+]] = load <12 x float>, ptr [[GEP]], align 16 ; CHECK: %{{.* }}= shufflevector <12 x float> [[LOAD]], <12 x float> poison, <4 x i32> ; CHECK: %{{.* }}= shufflevector <12 x float> [[LOAD]], <12 x float> poison, <4 x i32> ; CHECK: %{{.* }}= shufflevector <12 x float> [[LOAD]], <12 x float> poison, <4 x i32> ; CHECK-DAG: ret void ; AS-LABEL: aarch64_ilc_idx_ld3 ; AS: ld3 ; AS: ret %idx1 = lshr i64 %idx, 2 %a1 = add i64 %idx, 4 %idx2 = lshr i64 %a1, 2 %a2 = add i64 %idx, 8 %idx3 = lshr i64 %a2, 2 %gep1 = getelementptr inbounds <4 x float>, ptr %ptr, i64 %idx1 %gep2 = getelementptr inbounds <4 x float>, ptr %ptr, i64 %idx2 %gep3 = getelementptr inbounds <4 x float>, ptr %ptr, i64 %idx3 %ld1 = load <4 x float>, ptr %gep1, align 16 %ld2 = load <4 x float>, ptr %gep2, align 16 %ld3 = load <4 x float>, ptr %gep3, align 16 %sv1 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> %sv2 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> %sv3 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> %m0_3 = shufflevector <4 x float> %sv1, <4 x float> %ld3, <4 x i32> %m4_7 = shufflevector <4 x float> %sv2, <4 x float> %ld3, <4 x i32> %m8_11 = shufflevector <4 x float> %sv3, <4 x float> %ld3, <4 x i32> store <4 x float> %m0_3, ptr %gep1, align 16 store <4 x float> %m4_7, ptr %gep2, align 16 store <4 x float> %m8_11, ptr %gep3, align 16 ret void } ; %sv3 = shufflevector <4 x float> %ld3, <4 x float> %ld4, <4 x i32> ; This must not be lowered define void @aarch64_ilc_i32_idx(ptr %ptr, i32 %idx) { ; CHECK-LABEL: @aarch64_ilc_i32_idx( ; CHECK: %idx1 = lshr i32 %idx, 2 ; CHECK-NEXT: %a1 = add i32 %idx, 4 ; CHECK-NEXT: %idx2 = lshr i32 %a1, 2 ; CHECK-NEXT: %gep1 = getelementptr inbounds <4 x float>, ptr %ptr, i32 %idx1 ; CHECK-NEXT: %gep2 = getelementptr inbounds <4 x float>, ptr %ptr, i32 %idx2 ; CHECK-NEXT: %ld1 = load <4 x float>, ptr %gep1, align 16 ; CHECK-NEXT: %ld2 = load <4 x float>, ptr %gep2, align 16 ; CHECK-NEXT: %m0_3 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> ; CHECK-NEXT: %m4_7 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> ; CHECK-NEXT: store <4 x float> %m0_3, ptr %gep1, align 16 ; CHECK-NEXT: store <4 x float> %m4_7, ptr %gep2, align 16 ; CHECK-NEXT: ret void ; AS-LABEL: aarch64_ilc_i32_idx ; AS-DAG: @function ; AS-NOT: ld2 ; AS-NOT: ld3 ; AS-NOT: ld4 ; AS-DAG: ret entry: %idx1 = lshr i32 %idx, 2 %a1 = add i32 %idx, 4 %idx2 = lshr i32 %a1, 2 %gep1 = getelementptr inbounds <4 x float>, ptr %ptr, i32 %idx1 %gep2 = getelementptr inbounds <4 x float>, ptr %ptr, i32 %idx2 %ld1 = load <4 x float>, ptr %gep1, align 16 %ld2 = load <4 x float>, ptr %gep2, align 16 %m0_3 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> %m4_7 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> store <4 x float> %m0_3, ptr %gep1, align 16 store <4 x float> %m4_7, ptr %gep2, align 16 ret void } ; Volatile loads must not be lowered define void @aarch64_ilc_volatile(ptr %ptr) { ; CHECK-LABEL: @aarch64_ilc_volatile( ; CHECK: %gep2 = getelementptr inbounds <4 x float>, ptr %ptr, i32 1 ; CHECK-NEXT: %ld1 = load volatile <4 x float>, ptr %ptr, align 16 ; CHECK-NEXT: %ld2 = load <4 x float>, ptr %gep2, align 16 ; CHECK-NEXT: %m0_3 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> ; CHECK-NEXT: %m4_7 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> ; CHECK-NEXT: store <4 x float> %m0_3, ptr %ptr, align 16 ; CHECK-NEXT: store <4 x float> %m4_7, ptr %gep2, align 16 ; CHECK-NEXT: ret void ; AS-LABEL: aarch64_ilc_volatile ; AS-DAG: @function ; AS-NOT: ld2 ; AS-NOT: ld3 ; AS-NOT: ld4 ; AS-DAG: ret entry: %gep2 = getelementptr inbounds <4 x float>, ptr %ptr, i32 1 %ld1 = load volatile <4 x float>, ptr %ptr, align 16 %ld2 = load <4 x float>, ptr %gep2, align 16 %m0_3 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> %m4_7 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> store <4 x float> %m0_3, ptr %ptr, align 16 store <4 x float> %m4_7, ptr %gep2, align 16 ret void } ; This must not be lowered define void @aarch64_ilc_depmem(ptr %ptr, i32 %idx) { entry: ; CHECK-LABEL: @aarch64_ilc_depmem( ; CHECK: %gep2 = getelementptr inbounds <4 x float>, ptr %ptr, i32 1 ; CHECK-NEXT: %ld1 = load <4 x float>, ptr %ptr, align 16 ; CHECK-NEXT: store <4 x float> %ld1, ptr %gep2, align 16 ; CHECK-NEXT: %ld2 = load <4 x float>, ptr %gep2, align 16 ; CHECK-NEXT: %m0_3 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> ; CHECK-NEXT: %m4_7 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> ; CHECK-NEXT: store <4 x float> %m0_3, ptr %ptr, align 16 ; CHECK-NEXT: store <4 x float> %m4_7, ptr %gep2, align 16 ; CHECK-NEXT: ret void ; AS-LABEL: aarch64_ilc_depmem ; AS-DAG: @function ; AS-NOT: ld2 ; AS-NOT: ld3 ; AS-NOT: ld4 ; AS-DAG: ret %gep2 = getelementptr inbounds <4 x float>, ptr %ptr, i32 1 %ld1 = load <4 x float>, ptr %ptr, align 16 store <4 x float> %ld1, ptr %gep2, align 16 %ld2 = load <4 x float>, ptr %gep2, align 16 %m0_3 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> %m4_7 = shufflevector <4 x float> %ld1, <4 x float> %ld2, <4 x i32> store <4 x float> %m0_3, ptr %ptr, align 16 store <4 x float> %m4_7, ptr %gep2, align 16 ret void } ; This cannot be converted - insertion position cannot be determined define void @aarch64_no_insertion_pos(ptr %ptr) { entry: ; CHECK-LABEL: @aarch64_no_insertion_pos( ; CHECK: %p1 = getelementptr inbounds float, ptr %ptr, i32 4 ; CHECK-NEXT: %l0 = load <5 x float>, ptr %ptr ; CHECK-NEXT: %l1 = load <5 x float>, ptr %p1 ; CHECK-NEXT: %s0 = shufflevector <5 x float> %l0, <5 x float> %l1, <4 x i32> ; CHECK-NEXT: %s1 = shufflevector <5 x float> %l0, <5 x float> %l1, <4 x i32> ; CHECK-NEXT: ret void %p1 = getelementptr inbounds float, ptr %ptr, i32 4 %l0 = load <5 x float>, ptr %ptr %l1 = load <5 x float>, ptr %p1 %s0 = shufflevector <5 x float> %l0, <5 x float> %l1, <4 x i32> %s1 = shufflevector <5 x float> %l0, <5 x float> %l1, <4 x i32> ret void } ; This cannot be converted - the insertion position does not dominate all ; uses define void @aarch64_insertpos_does_not_dominate(ptr %ptr) { entry: ; CHECK-LABEL: @aarch64_insertpos_does_not_dominate( ; CHECK: %p1 = getelementptr inbounds float, ptr %ptr, i32 1 ; CHECK-NEXT: %l1 = load <7 x float>, ptr %p1 ; CHECK-NEXT: %s1 = shufflevector <7 x float> %l1, <7 x float> poison, <4 x i32> ; CHECK-NEXT: %l0 = load <7 x float>, ptr %ptr ; CHECK-NEXT: %s0 = shufflevector <7 x float> %l0, <7 x float> poison, <4 x i32> ; CHECK-NEXT: ret void %p1 = getelementptr inbounds float, ptr %ptr, i32 1 %l1 = load <7 x float>, ptr %p1 %s1 = shufflevector <7 x float> %l1, <7 x float> poison, <4 x i32> %l0 = load <7 x float>, ptr %ptr %s0 = shufflevector <7 x float> %l0, <7 x float> poison, <4 x i32> ret void }