1; RUN: opt -S -dxil-op-lower %s | FileCheck %s 2 3target triple = "dxil-pc-shadermodel6.6-compute" 4 5; CHECK-LABEL: define void @storef32_struct 6define void @storef32_struct(i32 %index, float %data) { 7 %buffer = call target("dx.RawBuffer", float, 1, 0, 0) 8 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_f32_0_0_0( 9 i32 0, i32 0, i32 1, i32 0, i1 false) 10 11 ; CHECK: call void @dx.op.rawBufferStore.f32(i32 140, %dx.types.Handle %buffer_annot, i32 %index, i32 0, float %data, float undef, float undef, float undef, i8 1, i32 4) 12 call void @llvm.dx.resource.store.rawbuffer.f32( 13 target("dx.RawBuffer", float, 1, 0, 0) %buffer, 14 i32 %index, i32 0, float %data) 15 16 ret void 17} 18 19; CHECK-LABEL: define void @storef32_byte 20define void @storef32_byte(i32 %offset, float %data) { 21 %buffer = call target("dx.RawBuffer", i8, 1, 0, 0) 22 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_i8_0_0_0( 23 i32 0, i32 0, i32 1, i32 0, i1 false) 24 25 ; CHECK: call void @dx.op.rawBufferStore.f32(i32 140, %dx.types.Handle %buffer_annot, i32 %offset, i32 0, float %data, float undef, float undef, float undef, i8 1, i32 4) 26 call void @llvm.dx.resource.store.rawbuffer.f32( 27 target("dx.RawBuffer", i8, 1, 0, 0) %buffer, 28 i32 %offset, i32 0, float %data) 29 30 ret void 31} 32 33; CHECK-LABEL: define void @storev4f32_struct 34define void @storev4f32_struct(i32 %index, <4 x float> %data) { 35 %buffer = call target("dx.RawBuffer", <4 x float>, 1, 0, 0) 36 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_v4f32_0_0_0( 37 i32 0, i32 0, i32 1, i32 0, i1 false) 38 39 ; CHECK: [[DATA0:%.*]] = extractelement <4 x float> %data, i32 0 40 ; CHECK: [[DATA1:%.*]] = extractelement <4 x float> %data, i32 1 41 ; CHECK: [[DATA2:%.*]] = extractelement <4 x float> %data, i32 2 42 ; CHECK: [[DATA3:%.*]] = extractelement <4 x float> %data, i32 3 43 ; CHECK: call void @dx.op.rawBufferStore.f32(i32 140, %dx.types.Handle %buffer_annot, i32 %index, i32 0, float [[DATA0]], float [[DATA1]], float [[DATA2]], float [[DATA3]], i8 15, i32 4) 44 call void @llvm.dx.resource.store.rawbuffer.v4f32( 45 target("dx.RawBuffer", <4 x float>, 1, 0, 0) %buffer, 46 i32 %index, i32 0, <4 x float> %data) 47 48 ret void 49} 50 51; CHECK-LABEL: define void @storev4f32_byte 52define void @storev4f32_byte(i32 %offset, <4 x float> %data) { 53 %buffer = call target("dx.RawBuffer", i8, 1, 0, 0) 54 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_i8_0_0_0( 55 i32 0, i32 0, i32 1, i32 0, i1 false) 56 57 ; CHECK: [[DATA0:%.*]] = extractelement <4 x float> %data, i32 0 58 ; CHECK: [[DATA1:%.*]] = extractelement <4 x float> %data, i32 1 59 ; CHECK: [[DATA2:%.*]] = extractelement <4 x float> %data, i32 2 60 ; CHECK: [[DATA3:%.*]] = extractelement <4 x float> %data, i32 3 61 ; CHECK: call void @dx.op.rawBufferStore.f32(i32 140, %dx.types.Handle %buffer_annot, i32 %offset, i32 0, float [[DATA0]], float [[DATA1]], float [[DATA2]], float [[DATA3]], i8 15, i32 4) 62 call void @llvm.dx.resource.store.rawbuffer.v4f32( 63 target("dx.RawBuffer", i8, 1, 0, 0) %buffer, 64 i32 %offset, i32 0, <4 x float> %data) 65 66 ret void 67} 68 69; CHECK-LABEL: define void @storeelements 70define void @storeelements(i32 %index, <4 x float> %data0, <4 x i32> %data1) { 71 %buffer = call target("dx.RawBuffer", {<4 x float>, <4 x i32>}, 1, 0, 0) 72 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_sl_v4f32v4i32s_0_0_0( 73 i32 0, i32 0, i32 1, i32 0, i1 false) 74 75 ; CHECK: [[DATA0_0:%.*]] = extractelement <4 x float> %data0, i32 0 76 ; CHECK: [[DATA0_1:%.*]] = extractelement <4 x float> %data0, i32 1 77 ; CHECK: [[DATA0_2:%.*]] = extractelement <4 x float> %data0, i32 2 78 ; CHECK: [[DATA0_3:%.*]] = extractelement <4 x float> %data0, i32 3 79 ; CHECK: call void @dx.op.rawBufferStore.f32(i32 140, %dx.types.Handle %buffer_annot, i32 %index, i32 0, float [[DATA0_0]], float [[DATA0_1]], float [[DATA0_2]], float [[DATA0_3]], i8 15, i32 4) 80 call void @llvm.dx.resource.store.rawbuffer.v4f32( 81 target("dx.RawBuffer", {<4 x float>, <4 x i32>}, 1, 0, 0) %buffer, 82 i32 %index, i32 0, <4 x float> %data0) 83 84 ; CHECK: [[DATA1_0:%.*]] = extractelement <4 x i32> %data1, i32 0 85 ; CHECK: [[DATA1_1:%.*]] = extractelement <4 x i32> %data1, i32 1 86 ; CHECK: [[DATA1_2:%.*]] = extractelement <4 x i32> %data1, i32 2 87 ; CHECK: [[DATA1_3:%.*]] = extractelement <4 x i32> %data1, i32 3 88 ; CHECK: call void @dx.op.rawBufferStore.i32(i32 140, %dx.types.Handle %buffer_annot, i32 %index, i32 16, i32 [[DATA1_0]], i32 [[DATA1_1]], i32 [[DATA1_2]], i32 [[DATA1_3]], i8 15, i32 4) 89 call void @llvm.dx.resource.store.rawbuffer.v4i32( 90 target("dx.RawBuffer", {<4 x float>, <4 x i32>}, 1, 0, 0) %buffer, 91 i32 %index, i32 16, <4 x i32> %data1) 92 93 ret void 94} 95 96; CHECK-LABEL: define void @storenested 97define void @storenested(i32 %index, i32 %data0, <4 x float> %data1, <3 x half> %data2) { 98 %buffer = call 99 target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 1, 0, 0) 100 @llvm.dx.resource.handlefrombinding(i32 0, i32 0, i32 1, i32 0, i1 false) 101 102 ; CHECK: call void @dx.op.rawBufferStore.i32(i32 140, %dx.types.Handle %buffer_annot, i32 %index, i32 0, i32 %data0, i32 undef, i32 undef, i32 undef, i8 1, i32 4) 103 call void @llvm.dx.resource.store.rawbuffer.i32( 104 target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 1, 0, 0) %buffer, 105 i32 %index, i32 0, i32 %data0) 106 107 ; CHECK: [[DATA1_0:%.*]] = extractelement <4 x float> %data1, i32 0 108 ; CHECK: [[DATA1_1:%.*]] = extractelement <4 x float> %data1, i32 1 109 ; CHECK: [[DATA1_2:%.*]] = extractelement <4 x float> %data1, i32 2 110 ; CHECK: [[DATA1_3:%.*]] = extractelement <4 x float> %data1, i32 3 111 ; CHECK: call void @dx.op.rawBufferStore.f32(i32 140, %dx.types.Handle %buffer_annot, i32 %index, i32 4, float [[DATA1_0]], float [[DATA1_1]], float [[DATA1_2]], float [[DATA1_3]], i8 15, i32 4) 112 call void @llvm.dx.resource.store.rawbuffer.v4f32( 113 target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 1, 0, 0) %buffer, 114 i32 %index, i32 4, <4 x float> %data1) 115 116 ; CHECK: [[DATA2_0:%.*]] = extractelement <3 x half> %data2, i32 0 117 ; CHECK: [[DATA2_1:%.*]] = extractelement <3 x half> %data2, i32 1 118 ; CHECK: [[DATA2_2:%.*]] = extractelement <3 x half> %data2, i32 2 119 ; CHECK: call void @dx.op.rawBufferStore.f16(i32 140, %dx.types.Handle %buffer_annot, i32 %index, i32 20, half [[DATA2_0]], half [[DATA2_1]], half [[DATA2_2]], half undef, i8 7, i32 2) 120 call void @llvm.dx.resource.store.rawbuffer.v3f16( 121 target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 1, 0, 0) %buffer, 122 i32 %index, i32 20, <3 x half> %data2) 123 124 ret void 125} 126 127; byteaddressbuf.Store<int64_t4> 128; CHECK-LABEL: define void @storev4f64_byte 129define void @storev4f64_byte(i32 %offset, <4 x double> %data) { 130 %buffer = call target("dx.RawBuffer", i8, 1, 0, 0) 131 @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_i8_0_0_0( 132 i32 0, i32 0, i32 1, i32 0, i1 false) 133 134 ; CHECK: [[DATA0:%.*]] = extractelement <4 x double> %data, i32 0 135 ; CHECK: [[DATA1:%.*]] = extractelement <4 x double> %data, i32 1 136 ; CHECK: [[DATA2:%.*]] = extractelement <4 x double> %data, i32 2 137 ; CHECK: [[DATA3:%.*]] = extractelement <4 x double> %data, i32 3 138 ; CHECK: call void @dx.op.rawBufferStore.f64(i32 140, %dx.types.Handle %buffer_annot, i32 %offset, i32 0, double [[DATA0]], double [[DATA1]], double [[DATA2]], double [[DATA3]], i8 15, i32 8) 139 call void @llvm.dx.resource.store.rawbuffer.v4i64( 140 target("dx.RawBuffer", i8, 1, 0, 0) %buffer, 141 i32 %offset, i32 0, <4 x double> %data) 142 143 ret void 144} 145