e1d2ff6caf
This patch implements the SPIR-V lowering for the following HLSL intrinsics: - SampleBias - SampleGrad - SampleLevel - SampleCmp - SampleCmpLevelZero It defines the required LLVM intrinsics in 'IntrinsicsDirectX.td' and 'IntrinsicsSPIRV.td'. It updates 'SPIRVInstructionSelector.cpp' to handle the new intrinsics and generates the correct 'OpImageSample*' instructions with the required operands (Bias, Grad, Lod, ConstOffset, MinLod, etc.). CodeGen tests are added to verify the implementation for images with dimension 1D, 2D, 3D, and Cube. Assisted-by: Gemini
52 lines
3.9 KiB
LLVM
52 lines
3.9 KiB
LLVM
; RUN: llc -O0 -mtriple=spirv-vulkan-compute %s -o - | FileCheck %s
|
|
; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv-vulkan-compute %s -o - -filetype=obj | spirv-val %}
|
|
|
|
; CHECK-DAG: OpCapability Shader
|
|
|
|
; CHECK-DAG: %[[float:[0-9]+]] = OpTypeFloat 32
|
|
; CHECK-DAG: %[[v4float:[0-9]+]] = OpTypeVector %[[float]] 4
|
|
; CHECK-DAG: %[[image:[0-9]+]] = OpTypeImage %[[float]] Cube 2 0 0 1 Unknown
|
|
; CHECK-DAG: %[[sampled_image:[0-9]+]] = OpTypeSampledImage %[[image]]
|
|
; CHECK-DAG: %[[sampler:[0-9]+]] = OpTypeSampler
|
|
; CHECK-DAG: %[[v3float:[0-9]+]] = OpTypeVector %[[float]] 3
|
|
; CHECK-DAG: %[[coord0:[0-9]+]] = OpConstantNull %[[v3float]]
|
|
; CHECK-DAG: %[[minlod_val:[0-9]+]] = OpConstant %[[float]] 1
|
|
|
|
@.str = private unnamed_addr constant [4 x i8] c"img\00", align 1
|
|
@.str.1 = private unnamed_addr constant [5 x i8] c"samp\00", align 1
|
|
@.str.2 = private unnamed_addr constant [4 x i8] c"out\00", align 1
|
|
|
|
define void @main() {
|
|
entry:
|
|
%img = tail call target("spirv.Image", float, 3, 2, 0, 0, 1, 0) @llvm.spv.resource.handlefrombinding.tspirv.Image_f32_3_2_0_0_1_0t(i32 0, i32 0, i32 1, i32 0, ptr @.str)
|
|
%sampler = tail call target("spirv.Sampler") @llvm.spv.resource.handlefrombinding.tspirv.Samplert(i32 0, i32 1, i32 1, i32 0, ptr @.str.1)
|
|
|
|
; CHECK: %[[img_val:[0-9]+]] = OpLoad %[[image]]
|
|
; CHECK: %[[sampler_val:[0-9]+]] = OpLoad %[[sampler]]
|
|
; CHECK: %[[si:[0-9]+]] = OpSampledImage %[[sampled_image]] %[[img_val]] %[[sampler_val]]
|
|
; CHECK: %[[res0:[0-9]+]] = OpImageSampleImplicitLod %[[v4float]] %[[si]] %[[coord0]]{{[ ]*$}}
|
|
%res0 = call <4 x float> @llvm.spv.resource.sample.v4f32.tspirv.Image_f32_3_2_0_0_1_0t.tspirv.Samplert.v3f32.v3i32(target("spirv.Image", float, 3, 2, 0, 0, 1, 0) %img, target("spirv.Sampler") %sampler, <3 x float> <float 0.0, float 0.0, float 0.0>, <3 x i32> zeroinitializer)
|
|
|
|
; CHECK: %[[img_val3:[0-9]+]] = OpLoad %[[image]]
|
|
; CHECK: %[[sampler_val3:[0-9]+]] = OpLoad %[[sampler]]
|
|
; CHECK: %[[si3:[0-9]+]] = OpSampledImage %[[sampled_image]] %[[img_val3]] %[[sampler_val3]]
|
|
; CHECK: %[[res2:[0-9]+]] = OpImageSampleImplicitLod %[[v4float]] %[[si3]] %[[coord0]] MinLod %[[minlod_val]]{{[ ]*$}}
|
|
%res2 = call <4 x float> @llvm.spv.resource.sample.clamp.v4f32.tspirv.Image_f32_3_2_0_0_1_0t.tspirv.Samplert.v3f32.v3i32(target("spirv.Image", float, 3, 2, 0, 0, 1, 0) %img, target("spirv.Sampler") %sampler, <3 x float> <float 0.0, float 0.0, float 0.0>, <3 x i32> zeroinitializer, float 1.0)
|
|
|
|
%tmp0 = fadd <4 x float> %res0, %res2
|
|
%res = fadd <4 x float> %tmp0, %tmp0
|
|
|
|
; CHECK: %[[out_handle:[0-9]+]] = OpLoad {{.*}}
|
|
; CHECK: OpImageWrite %[[out_handle]] {{.*}} %[[final_res:[0-9]+]]
|
|
%out = tail call target("spirv.Image", float, 5, 2, 0, 0, 2, 1) @llvm.spv.resource.handlefrombinding.tspirv.Image_f32_5_2_0_0_2_1t(i32 0, i32 2, i32 1, i32 0, ptr @.str.2)
|
|
%out_ptr = call ptr addrspace(11) @llvm.spv.resource.getpointer.p11.tspirv.Image_f32_5_2_0_0_2_1t(target("spirv.Image", float, 5, 2, 0, 0, 2, 1) %out, i32 0)
|
|
store <4 x float> %res, ptr addrspace(11) %out_ptr
|
|
ret void
|
|
}
|
|
|
|
declare target("spirv.Image", float, 3, 2, 0, 0, 1, 0) @llvm.spv.resource.handlefrombinding.tspirv.Image_f32_3_2_0_0_1_0t(i32, i32, i32, i32, ptr)
|
|
declare target("spirv.Sampler") @llvm.spv.resource.handlefrombinding.tspirv.Samplert(i32, i32, i32, i32, ptr)
|
|
declare <4 x float> @llvm.spv.resource.sample.v4f32.tspirv.Image_f32_3_2_0_0_1_0t.tspirv.Samplert.v3f32.v3i32(target("spirv.Image", float, 3, 2, 0, 0, 1, 0), target("spirv.Sampler"), <3 x float>, <3 x i32>)
|
|
declare <4 x float> @llvm.spv.resource.sample.clamp.v4f32.tspirv.Image_f32_3_2_0_0_1_0t.tspirv.Samplert.v3f32.v3i32(target("spirv.Image", float, 3, 2, 0, 0, 1, 0), target("spirv.Sampler"), <3 x float>, <3 x i32>, float)
|
|
declare target("spirv.Image", float, 5, 2, 0, 0, 2, 1) @llvm.spv.resource.handlefrombinding.tspirv.Image_f32_5_2_0_0_2_1t(i32, i32, i32, i32, ptr)
|
|
declare ptr addrspace(11) @llvm.spv.resource.getpointer.p11.tspirv.Image_f32_5_2_0_0_2_1t(target("spirv.Image", float, 5, 2, 0, 0, 2, 1), i32) |