1; RUN: opt -S -passes='dxil-data-scalarization,function(scalarizer<load-store>),dxil-op-lower' -mtriple=dxil-pc-shadermodel6.3-library %s | FileCheck %s 2 3; Make sure we can load groupshared, static vectors and arrays of vectors 4 5@arrayofVecData = local_unnamed_addr addrspace(3) global [2 x <3 x float>] zeroinitializer, align 16 6@vecData = external addrspace(3) global <4 x i32>, align 4 7@staticArrayOfVecData = internal global [3 x <4 x i32>] [<4 x i32> <i32 1, i32 2, i32 3, i32 4>, <4 x i32> <i32 5, i32 6, i32 7, i32 8>, <4 x i32> <i32 9, i32 10, i32 11, i32 12>], align 4 8@groushared2dArrayofVectors = local_unnamed_addr addrspace(3) global [3 x [ 3 x <4 x i32>]] zeroinitializer, align 16 9 10; CHECK: @arrayofVecData.scalarized = local_unnamed_addr addrspace(3) global [2 x [3 x float]] zeroinitializer, align 16 11; CHECK: @vecData.scalarized = external addrspace(3) global [4 x i32], align 4 12; CHECK: @staticArrayOfVecData.scalarized = internal global [3 x [4 x i32]] {{\[}}[4 x i32] [i32 1, i32 2, i32 3, i32 4], [4 x i32] [i32 5, i32 6, i32 7, i32 8], [4 x i32] [i32 9, i32 10, i32 11, i32 12]], align 4 13; CHECK: @groushared2dArrayofVectors.scalarized = local_unnamed_addr addrspace(3) global [3 x [3 x [4 x i32]]] zeroinitializer, align 16 14 15; CHECK-NOT: @arrayofVecData 16; CHECK-NOT: @vecData 17; CHECK-NOT: @staticArrayOfVecData 18; CHECK-NOT: @groushared2dArrayofVectors 19 20 21; CHECK-LABEL: load_array_vec_test 22define <3 x float> @load_array_vec_test() #0 { 23 ; CHECK-COUNT-6: load float, ptr addrspace(3) {{(.*@arrayofVecData.scalarized.*|%.*)}}, align 4 24 %1 = load <3 x float>, <3 x float> addrspace(3)* getelementptr inbounds ([2 x <3 x float>], [2 x <3 x float>] addrspace(3)* @"arrayofVecData", i32 0, i32 0), align 4 25 %2 = load <3 x float>, <3 x float> addrspace(3)* getelementptr inbounds ([2 x <3 x float>], [2 x <3 x float>] addrspace(3)* @"arrayofVecData", i32 0, i32 1), align 4 26 %3 = fadd <3 x float> %1, %2 27 ret <3 x float> %3 28} 29 30; CHECK-LABEL: load_vec_test 31define <4 x i32> @load_vec_test() #0 { 32 ; CHECK-COUNT-4: load i32, ptr addrspace(3) {{(@vecData.scalarized|getelementptr \(i32, ptr addrspace\(3\) @vecData.scalarized, i32 .*\)|%.*)}}, align {{.*}} 33 %1 = load <4 x i32>, <4 x i32> addrspace(3)* @"vecData", align 4 34 ret <4 x i32> %1 35} 36 37; CHECK-LABEL: load_static_array_of_vec_test 38define <4 x i32> @load_static_array_of_vec_test(i32 %index) #0 { 39 ; CHECK: getelementptr inbounds [3 x [4 x i32]], ptr @staticArrayOfVecData.scalarized, i32 0, i32 %index 40 ; CHECK: load i32, ptr {{.*}}, align 4 41 ; CHECK: getelementptr i32, ptr {{.*}}, i32 1 42 ; CHECK: load i32, ptr {{.*}}, align 4 43 ; CHECK: getelementptr i32, ptr {{.*}}, i32 2 44 ; CHECK: load i32, ptr {{.*}}, align 4 45 ; CHECK: getelementptr i32, ptr {{.*}}, i32 3 46 ; CHECK: load i32, ptr {{.*}}, align 4 47 %3 = getelementptr inbounds [3 x <4 x i32>], [3 x <4 x i32>]* @staticArrayOfVecData, i32 0, i32 %index 48 %4 = load <4 x i32>, <4 x i32>* %3, align 4 49 ret <4 x i32> %4 50} 51 52; CHECK-LABEL: multid_load_test 53define <4 x i32> @multid_load_test() #0 { 54 ; CHECK-COUNT-8: load i32, ptr addrspace(3) {{(.*@groushared2dArrayofVectors.scalarized.*|%.*)}}, align 4 55 %1 = load <4 x i32>, <4 x i32> addrspace(3)* getelementptr inbounds ([3 x [3 x <4 x i32>]], [3 x [3 x <4 x i32>]] addrspace(3)* @"groushared2dArrayofVectors", i32 0, i32 0, i32 0), align 4 56 %2 = load <4 x i32>, <4 x i32> addrspace(3)* getelementptr inbounds ([3 x [3 x <4 x i32>]], [3 x [3 x <4 x i32>]] addrspace(3)* @"groushared2dArrayofVectors", i32 0, i32 1, i32 1), align 4 57 %3 = add <4 x i32> %1, %2 58 ret <4 x i32> %3 59} 60 61attributes #0 = { convergent norecurse nounwind "hlsl.export"} 62