aboutsummaryrefslogtreecommitdiff
path: root/llvm/test/CodeGen/DirectX/BufferStore-sm61.ll
blob: dff28dbde82be9281cedadbb09fe6c6840dc1220 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
; RUN: opt -S -dxil-op-lower %s | FileCheck %s
; Before SM6.2 ByteAddressBuffer and StructuredBuffer lower to bufferStore.

target triple = "dxil-pc-shadermodel6.1-compute"

; CHECK-LABEL: define void @storef32_struct
define void @storef32_struct(i32 %index, float %data) {
  %buffer = call target("dx.RawBuffer", float, 1, 0, 0)
      @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_f32_0_0_0(
          i32 0, i32 0, i32 1, i32 0, i1 false, ptr null)

  ; CHECK: call void @dx.op.bufferStore.f32(i32 69, %dx.types.Handle %{{.*}}, i32 %index, i32 0, float %data, float undef, float undef, float undef, i8 1)
  call void @llvm.dx.resource.store.rawbuffer.f32(
      target("dx.RawBuffer", float, 1, 0, 0) %buffer,
      i32 %index, i32 0, float %data)

  ret void
}

; CHECK-LABEL: define void @storef32_byte
define void @storef32_byte(i32 %offset, float %data) {
  %buffer = call target("dx.RawBuffer", i8, 1, 0, 0)
      @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_i8_0_0_0(
          i32 0, i32 0, i32 1, i32 0, i1 false, ptr null)

  ; CHECK: call void @dx.op.bufferStore.f32(i32 69, %dx.types.Handle %{{.*}}, i32 %offset, i32 0, float %data, float undef, float undef, float undef, i8 1)
  call void @llvm.dx.resource.store.rawbuffer.f32(
      target("dx.RawBuffer", i8, 1, 0, 0) %buffer,
      i32 %offset, i32 0, float %data)

  ret void
}

; CHECK-LABEL: define void @storev4f32_struct
define void @storev4f32_struct(i32 %index, <4 x float> %data) {
  %buffer = call target("dx.RawBuffer", <4 x float>, 1, 0, 0)
      @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_v4f32_0_0_0(
          i32 0, i32 0, i32 1, i32 0, i1 false, ptr null)

  ; CHECK: [[DATA0:%.*]] = extractelement <4 x float> %data, i32 0
  ; CHECK: [[DATA1:%.*]] = extractelement <4 x float> %data, i32 1
  ; CHECK: [[DATA2:%.*]] = extractelement <4 x float> %data, i32 2
  ; CHECK: [[DATA3:%.*]] = extractelement <4 x float> %data, i32 3
  ; CHECK: call void @dx.op.bufferStore.f32(i32 69, %dx.types.Handle %{{.*}}, i32 %index, i32 0, float [[DATA0]], float [[DATA1]], float [[DATA2]], float [[DATA3]], i8 15)
  call void @llvm.dx.resource.store.rawbuffer.v4f32(
      target("dx.RawBuffer", <4 x float>, 1, 0, 0) %buffer,
      i32 %index, i32 0, <4 x float> %data)

  ret void
}

; CHECK-LABEL: define void @storev4f32_byte
define void @storev4f32_byte(i32 %offset, <4 x float> %data) {
  %buffer = call target("dx.RawBuffer", i8, 1, 0, 0)
      @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_i8_0_0_0(
          i32 0, i32 0, i32 1, i32 0, i1 false, ptr null)

  ; CHECK: [[DATA0:%.*]] = extractelement <4 x float> %data, i32 0
  ; CHECK: [[DATA1:%.*]] = extractelement <4 x float> %data, i32 1
  ; CHECK: [[DATA2:%.*]] = extractelement <4 x float> %data, i32 2
  ; CHECK: [[DATA3:%.*]] = extractelement <4 x float> %data, i32 3
  ; CHECK: call void @dx.op.bufferStore.f32(i32 69, %dx.types.Handle %{{.*}}, i32 %offset, i32 0, float [[DATA0]], float [[DATA1]], float [[DATA2]], float [[DATA3]], i8 15)
  call void @llvm.dx.resource.store.rawbuffer.v4f32(
      target("dx.RawBuffer", i8, 1, 0, 0) %buffer,
      i32 %offset, i32 0, <4 x float> %data)

  ret void
}

; CHECK-LABEL: define void @storeelements
define void @storeelements(i32 %index, <4 x float> %data0, <4 x i32> %data1) {
  %buffer = call target("dx.RawBuffer", {<4 x float>, <4 x i32>}, 1, 0, 0)
      @llvm.dx.resource.handlefrombinding.tdx.RawBuffer_sl_v4f32v4i32s_0_0_0(
          i32 0, i32 0, i32 1, i32 0, i1 false, ptr null)

  ; CHECK: [[DATA0_0:%.*]] = extractelement <4 x float> %data0, i32 0
  ; CHECK: [[DATA0_1:%.*]] = extractelement <4 x float> %data0, i32 1
  ; CHECK: [[DATA0_2:%.*]] = extractelement <4 x float> %data0, i32 2
  ; CHECK: [[DATA0_3:%.*]] = extractelement <4 x float> %data0, i32 3
  ; CHECK: call void @dx.op.bufferStore.f32(i32 69, %dx.types.Handle %{{.*}}, i32 %index, i32 0, float [[DATA0_0]], float [[DATA0_1]], float [[DATA0_2]], float [[DATA0_3]], i8 15)
  call void @llvm.dx.resource.store.rawbuffer.v4f32(
      target("dx.RawBuffer", {<4 x float>, <4 x i32>}, 1, 0, 0) %buffer,
      i32 %index, i32 0, <4 x float> %data0)

  ; CHECK: [[DATA1_0:%.*]] = extractelement <4 x i32> %data1, i32 0
  ; CHECK: [[DATA1_1:%.*]] = extractelement <4 x i32> %data1, i32 1
  ; CHECK: [[DATA1_2:%.*]] = extractelement <4 x i32> %data1, i32 2
  ; CHECK: [[DATA1_3:%.*]] = extractelement <4 x i32> %data1, i32 3
  ; CHECK: call void @dx.op.bufferStore.i32(i32 69, %dx.types.Handle %{{.*}}, i32 %index, i32 16, i32 [[DATA1_0]], i32 [[DATA1_1]], i32 [[DATA1_2]], i32 [[DATA1_3]], i8 15)
  call void @llvm.dx.resource.store.rawbuffer.v4i32(
      target("dx.RawBuffer", {<4 x float>, <4 x i32>}, 1, 0, 0) %buffer,
      i32 %index, i32 16, <4 x i32> %data1)

  ret void
}

; CHECK-LABEL: define void @storenested
define void @storenested(i32 %index, i32 %data0, <4 x float> %data1, <3 x half> %data2) {
  %buffer = call
      target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 1, 0, 0)
      @llvm.dx.resource.handlefrombinding(i32 0, i32 0, i32 1, i32 0, i1 false, ptr null)

  ; CHECK: call void @dx.op.bufferStore.i32(i32 69, %dx.types.Handle %{{.*}}, i32 %index, i32 0, i32 %data0, i32 undef, i32 undef, i32 undef, i8 1)
  call void @llvm.dx.resource.store.rawbuffer.i32(
      target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 1, 0, 0) %buffer,
      i32 %index, i32 0, i32 %data0)

  ; CHECK: [[DATA1_0:%.*]] = extractelement <4 x float> %data1, i32 0
  ; CHECK: [[DATA1_1:%.*]] = extractelement <4 x float> %data1, i32 1
  ; CHECK: [[DATA1_2:%.*]] = extractelement <4 x float> %data1, i32 2
  ; CHECK: [[DATA1_3:%.*]] = extractelement <4 x float> %data1, i32 3
  ; CHECK: call void @dx.op.bufferStore.f32(i32 69, %dx.types.Handle %{{.*}}, i32 %index, i32 4, float [[DATA1_0]], float [[DATA1_1]], float [[DATA1_2]], float [[DATA1_3]], i8 15)
  call void @llvm.dx.resource.store.rawbuffer.v4f32(
      target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 1, 0, 0) %buffer,
      i32 %index, i32 4, <4 x float> %data1)

  ; CHECK: [[DATA2_0:%.*]] = extractelement <3 x half> %data2, i32 0
  ; CHECK: [[DATA2_1:%.*]] = extractelement <3 x half> %data2, i32 1
  ; CHECK: [[DATA2_2:%.*]] = extractelement <3 x half> %data2, i32 2
  ; CHECK: call void @dx.op.bufferStore.f16(i32 69, %dx.types.Handle %{{.*}}, i32 %index, i32 20, half [[DATA2_0]], half [[DATA2_1]], half [[DATA2_2]], half undef, i8 7)
  call void @llvm.dx.resource.store.rawbuffer.v3f16(
      target("dx.RawBuffer", {i32, {<4 x float>, <3 x half>}}, 1, 0, 0) %buffer,
      i32 %index, i32 20, <3 x half> %data2)

  ret void
}