Files
TheRock/compiler/amd-llvm/llvm/test/CodeGen/DirectX/RawBufferStore.ll
T
derek.holloway 7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
Flatten subprojects into a single base monorepo
2026-09-02 16:34:47 -07:00

145 lines
7.4 KiB
LLVM

; RUN: opt -S -dxil-op-lower %s | FileCheck %s
target triple = "dxil-pc-shadermodel6.6-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, ptr null)
; 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)
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, ptr null)
; CHECK: call void @dx.op.rawBufferStore.f32(i32 140, %dx.types.Handle %buffer_annot, i32 %offset, i32 undef, float %data, float undef, float undef, float undef, i8 1, i32 4)
call void @llvm.dx.resource.store.rawbuffer.f32(
target("dx.RawBuffer", i8, 1, 0, 0) %buffer,
i32 %offset, i32 poison, 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, 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.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)
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, 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.rawBufferStore.f32(i32 140, %dx.types.Handle %buffer_annot, i32 %offset, i32 undef, float [[DATA0]], float [[DATA1]], float [[DATA2]], float [[DATA3]], i8 15, i32 4)
call void @llvm.dx.resource.store.rawbuffer.v4f32(
target("dx.RawBuffer", i8, 1, 0, 0) %buffer,
i32 %offset, i32 poison, <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, 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.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)
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.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)
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, ptr null)
; 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)
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.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)
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.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)
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
}
; byteaddressbuf.Store<int64_t4>
; CHECK-LABEL: define void @storev4f64_byte
define void @storev4f64_byte(i32 %offset, <4 x double> %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, ptr null)
; CHECK: [[DATA0:%.*]] = extractelement <4 x double> %data, i32 0
; CHECK: [[DATA1:%.*]] = extractelement <4 x double> %data, i32 1
; CHECK: [[DATA2:%.*]] = extractelement <4 x double> %data, i32 2
; CHECK: [[DATA3:%.*]] = extractelement <4 x double> %data, i32 3
; CHECK: call void @dx.op.rawBufferStore.f64(i32 140, %dx.types.Handle %buffer_annot, i32 %offset, i32 undef, double [[DATA0]], double [[DATA1]], double [[DATA2]], double [[DATA3]], i8 15, i32 8)
call void @llvm.dx.resource.store.rawbuffer.v4i64(
target("dx.RawBuffer", i8, 1, 0, 0) %buffer,
i32 %offset, i32 poison, <4 x double> %data)
ret void
}