38 lines · plain
1// RUN: %clang_cc1 -triple dxil-pc-shadermodel6.6-compute -finclude-default-header \2// RUN: -emit-llvm -disable-llvm-passes -o - %s | FileCheck %s3 4// This test verifies handling of local arrays of resources when used as a function argument.5 6// CHECK: @_ZL1A = internal global [3 x %"class.hlsl::RWBuffer"] poison, align 47 8RWBuffer<float> A[3] : register(u0);9RWStructuredBuffer<float> Out : register(u0);10 11// NOTE: _ZN4hlsl8RWBufferIfEixEj is the subscript operator for RWBuffer<float> and12// _ZN4hlsl18RWStructuredBufferIfEixEj is the subscript operator for RWStructuredBuffer<float>13 14// CHECK: define {{.*}} float @_Z3fooA3_N4hlsl8RWBufferIfEE(ptr noundef byval([3 x %"class.hlsl::RWBuffer"]) align 4 %LocalA)15// CHECK-NEXT: entry:16float foo(RWBuffer<float> LocalA[3]) {17// CHECK-NEXT: %[[LocalA_2_Ptr:.*]] = getelementptr inbounds [3 x %"class.hlsl::RWBuffer"], ptr %LocalA, i32 0, i32 218// CHECK-NEXT: %[[BufPtr:.*]] = call {{.*}} ptr @_ZN4hlsl8RWBufferIfEixEj(ptr {{.*}} %[[LocalA_2_Ptr]], i32 noundef 0)19// CHECK-NEXT: %[[Value:.*]] = load float, ptr %[[BufPtr]], align 420// CHECK-NEXT: ret float %[[Value]]21 return LocalA[2][0];22}23 24// CHECK: define internal void @_Z4mainv()25// CHECK-NEXT: entry:26[numthreads(4,1,1)]27void main() {28// Check that the `main` function calls `foo` with a local copy of the array29// CHECK-NEXT: %[[Tmp:.*]] = alloca [3 x %"class.hlsl::RWBuffer"], align 430// CHECK-NEXT: call void @llvm.memcpy.p0.p0.i32(ptr align 4 %[[Tmp]], ptr align 4 @_ZL1A, i32 12, i1 false)31 32// CHECK-NEXT: %[[ReturnedValue:.*]] = call {{.*}} float @_Z3fooA3_N4hlsl8RWBufferIfEE(ptr noundef byval([3 x %"class.hlsl::RWBuffer"]) align 4 %[[Tmp]])33// CHECK-NEXT: %[[OutBufPtr:.*]] = call {{.*}} ptr @_ZN4hlsl18RWStructuredBufferIfEixEj(ptr {{.*}} @_ZL3Out, i32 noundef 0)34// CHECK-NEXT: store float %[[ReturnedValue]], ptr %[[OutBufPtr]], align 435// CHECK-NEXT: ret void36 Out[0] = foo(A);37}38