168 lines · plain
1; RUN: opt -S -dxil-resource-access %s | FileCheck %s2 3target triple = "dxil-pc-shadermodel6.6-compute"4 5declare void @f32_user(float)6declare void @v4f32_user(<4 x float>)7declare void @i32_user(i32)8declare void @v4i32_user(<4 x i32>)9declare void @v3f16_user(<3 x half>)10declare void @v4f64_user(<4 x double>)11 12; CHECK-LABEL: define void @loadf32_struct13define void @loadf32_struct(i32 %index) {14 %buffer = call target("dx.RawBuffer", float, 0, 0, 0)15 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_f32_0_0_0(16 i32 0, i32 0, i32 1, i32 0, ptr null)17 18 ; CHECK-NOT: @llvm.dx.resource.getpointer19 %ptr = call ptr @llvm.dx.resource.getpointer(20 target("dx.RawBuffer", float, 0, 0, 0) %buffer, i32 %index)21 22 ; CHECK: %[[LOAD:.*]] = call { float, i1 } @llvm.dx.resource.load.rawbuffer.f32.tdx.RawBuffer_f32_0_0_0t(target("dx.RawBuffer", float, 0, 0, 0) %buffer, i32 %index, i32 0)23 ; CHECK: %[[VAL:.*]] = extractvalue { float, i1 } %[[LOAD]], 024 ; CHECK: call void @f32_user(float %[[VAL]])25 %data = load float, ptr %ptr26 call void @f32_user(float %data)27 28 ret void29}30 31; CHECK-LABEL: define void @loadf32_byte32define void @loadf32_byte(i32 %offset) {33 %buffer = call target("dx.RawBuffer", i8, 0, 0, 0)34 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_i8_0_0_0(35 i32 0, i32 0, i32 1, i32 0, ptr null)36 37 ; CHECK-NOT: @llvm.dx.resource.getpointer38 %ptr = call ptr @llvm.dx.resource.getpointer(39 target("dx.RawBuffer", i8, 0, 0, 0) %buffer, i32 %offset)40 41 ; CHECK: %[[LOAD:.*]] = call { float, i1 } @llvm.dx.resource.load.rawbuffer.f32.tdx.RawBuffer_i8_0_0_0t(target("dx.RawBuffer", i8, 0, 0, 0) %buffer, i32 %offset, i32 0)42 ; CHECK: %[[VAL:.*]] = extractvalue { float, i1 } %[[LOAD]], 043 ; CHECK: call void @f32_user(float %[[VAL]])44 %data = load float, ptr %ptr45 call void @f32_user(float %data)46 47 ret void48}49 50; CHECK-LABEL: define void @loadv4f32_struct51define void @loadv4f32_struct(i32 %index) {52 %buffer = call target("dx.RawBuffer", <4 x float>, 0, 0, 0)53 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_v4f32_0_0_0(54 i32 0, i32 0, i32 1, i32 0, ptr null)55 56 ; CHECK-NOT: @llvm.dx.resource.getpointer57 %ptr = call ptr @llvm.dx.resource.getpointer(58 target("dx.RawBuffer", <4 x float>, 0, 0, 0) %buffer, i32 %index)59 60 ; CHECK: %[[LOAD:.*]] = call { <4 x float>, i1 } @llvm.dx.resource.load.rawbuffer.v4f32.tdx.RawBuffer_v4f32_0_0_0t(target("dx.RawBuffer", <4 x float>, 0, 0, 0) %buffer, i32 %index, i32 0)61 ; CHECK: %[[VAL:.*]] = extractvalue { <4 x float>, i1 } %[[LOAD]], 062 ; CHECK: call void @v4f32_user(<4 x float> %[[VAL]])63 %data = load <4 x float>, ptr %ptr64 call void @v4f32_user(<4 x float> %data)65 66 ret void67}68 69; CHECK-LABEL: define void @loadv4f32_byte70define void @loadv4f32_byte(i32 %offset) {71 %buffer = call target("dx.RawBuffer", i8, 0, 0, 0)72 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_i8_0_0_0(73 i32 0, i32 0, i32 1, i32 0, ptr null)74 75 ; CHECK-NOT: @llvm.dx.resource.getpointer76 %ptr = call ptr @llvm.dx.resource.getpointer(77 target("dx.RawBuffer", i8, 0, 0, 0) %buffer, i32 %offset)78 79 ; CHECK: %[[LOAD:.*]] = call { <4 x float>, i1 } @llvm.dx.resource.load.rawbuffer.v4f32.tdx.RawBuffer_i8_0_0_0t(target("dx.RawBuffer", i8, 0, 0, 0) %buffer, i32 %offset, i32 0)80 ; CHECK: %[[VAL:.*]] = extractvalue { <4 x float>, i1 } %[[LOAD]], 081 ; CHECK: call void @v4f32_user(<4 x float> %[[VAL]]82 %data = load <4 x float>, ptr %ptr83 call void @v4f32_user(<4 x float> %data)84 85 ret void86}87 88; CHECK-LABEL: define void @loadelements89define void @loadelements(i32 %index) {90 %buffer = call target("dx.RawBuffer", {<4 x float>, <4 x i32>}, 0, 0, 0)91 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_sl_v4f32v4i32s_0_0_0(92 i32 0, i32 0, i32 1, i32 0, ptr null)93 94 ; CHECK-NOT: @llvm.dx.resource.getpointer95 %ptr = call ptr @llvm.dx.resource.getpointer(96 target("dx.RawBuffer", {<4 x float>, <4 x i32>}, 0, 0, 0) %buffer,97 i32 %index)98 99 ; CHECK: %[[LOADF32:.*]] = call { <4 x float>, i1 } @llvm.dx.resource.load.rawbuffer.v4f32.tdx.RawBuffer_sl_v4f32v4i32s_0_0_0t(target("dx.RawBuffer", { <4 x float>, <4 x i32> }, 0, 0, 0) %buffer, i32 %index, i32 0)100 ; CHECK: %[[VALF32:.*]] = extractvalue { <4 x float>, i1 } %[[LOADF32]], 0101 ; CHECK: call void @v4f32_user(<4 x float> %[[VALF32]]102 %dataf32 = load <4 x float>, ptr %ptr103 call void @v4f32_user(<4 x float> %dataf32)104 105 ; CHECK: %[[LOADI32:.*]] = call { <4 x i32>, i1 } @llvm.dx.resource.load.rawbuffer.v4i32.tdx.RawBuffer_sl_v4f32v4i32s_0_0_0t(target("dx.RawBuffer", { <4 x float>, <4 x i32> }, 0, 0, 0) %buffer, i32 %index, i32 16)106 ; CHECK: %[[VALI32:.*]] = extractvalue { <4 x i32>, i1 } %[[LOADI32]], 0107 ; CHECK: call void @v4i32_user(<4 x i32> %[[VALI32]]108 %addri32 = getelementptr inbounds nuw i8, ptr %ptr, i32 16109 %datai32 = load <4 x i32>, ptr %addri32110 call void @v4i32_user(<4 x i32> %datai32)111 112 ret void113}114 115; CHECK-LABEL: define void @loadnested116define void @loadnested(i32 %index) {117 %buffer = call118 target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 0, 0, 0)119 @llvm.dx.resource.handlefrombinding(i32 0, i32 0, i32 1, i32 0, ptr null)120 121 ; CHECK-NOT: @llvm.dx.resource.getpointer122 %ptr = call ptr @llvm.dx.resource.getpointer(123 target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 0, 0, 0) %buffer,124 i32 %index)125 126 ; CHECK: %[[LOADI32:.*]] = call { i32, i1 } @llvm.dx.resource.load.rawbuffer.i32.tdx.RawBuffer_sl_i32sl_v4f32v3f16ss_0_0_0t(target("dx.RawBuffer", { i32, { <4 x float>, <3 x half> } }, 0, 0, 0) %buffer, i32 %index, i32 0)127 ; CHECK: %[[VALI32:.*]] = extractvalue { i32, i1 } %[[LOADI32]], 0128 ; CHECK: call void @i32_user(i32 %[[VALI32]])129 %datai32 = load i32, ptr %ptr130 call void @i32_user(i32 %datai32)131 132 ; CHECK: %[[LOADF32:.*]] = call { <4 x float>, i1 } @llvm.dx.resource.load.rawbuffer.v4f32.tdx.RawBuffer_sl_i32sl_v4f32v3f16ss_0_0_0t(target("dx.RawBuffer", { i32, { <4 x float>, <3 x half> } }, 0, 0, 0) %buffer, i32 %index, i32 4)133 ; CHECK: %[[VALF32:.*]] = extractvalue { <4 x float>, i1 } %[[LOADF32]], 0134 ; CHECK: call void @v4f32_user(<4 x float> %[[VALF32]])135 %addrf32 = getelementptr inbounds nuw i8, ptr %ptr, i32 4136 %dataf32 = load <4 x float>, ptr %addrf32137 call void @v4f32_user(<4 x float> %dataf32)138 139 ; CHECK: %[[LOADF16:.*]] = call { <3 x half>, i1 } @llvm.dx.resource.load.rawbuffer.v3f16.tdx.RawBuffer_sl_i32sl_v4f32v3f16ss_0_0_0t(target("dx.RawBuffer", { i32, { <4 x float>, <3 x half> } }, 0, 0, 0) %buffer, i32 %index, i32 20)140 ; CHECK: %[[VALF16:.*]] = extractvalue { <3 x half>, i1 } %[[LOADF16]], 0141 ; CHECK: call void @v3f16_user(<3 x half> %[[VALF16]])142 %addrf16 = getelementptr inbounds nuw i8, ptr %ptr, i32 20143 %dataf16 = load <3 x half>, ptr %addrf16144 call void @v3f16_user(<3 x half> %dataf16)145 146 ret void147}148 149; byteaddressbuf.Load<int64_t4>150; CHECK-LABEL: define void @loadv4f64_byte151define void @loadv4f64_byte(i32 %offset) {152 %buffer = call target("dx.RawBuffer", i8, 0, 0, 0)153 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_i8_0_0_0(154 i32 0, i32 0, i32 1, i32 0, ptr null)155 156 ; CHECK-NOT: @llvm.dx.resource.getpointer157 %ptr = call ptr @llvm.dx.resource.getpointer(158 target("dx.RawBuffer", i8, 0, 0, 0) %buffer, i32 %offset)159 160 ; CHECK: %[[LOAD:.*]] = call { <4 x double>, i1 } @llvm.dx.resource.load.rawbuffer.v4f64.tdx.RawBuffer_i8_0_0_0t(target("dx.RawBuffer", i8, 0, 0, 0) %buffer, i32 %offset, i32 0)161 ; CHECK: %[[VAL:.*]] = extractvalue { <4 x double>, i1 } %[[LOAD]], 0162 ; CHECK: call void @v4f64_user(<4 x double> %[[VAL]])163 %data = load <4 x double>, ptr %ptr164 call void @v4f64_user(<4 x double> %data)165 166 ret void167}168