xref: /llvm-project/llvm/test/CodeGen/SPIRV/hlsl-intrinsics/step.ll (revision cf3d6fded9eaf1372ccfde1d49dd91df6762d98c)
1; RUN: llc -O0 -verify-machineinstrs -mtriple=spirv-unknown-unknown %s -o - | FileCheck %s
2; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv-unknown-unknown %s -o - -filetype=obj | spirv-val %}
3
4; Make sure SPIRV operation function calls for step are lowered correctly.
5
6; CHECK-DAG: %[[#op_ext_glsl:]] = OpExtInstImport "GLSL.std.450"
7; CHECK-DAG: %[[#float_32:]] = OpTypeFloat 32
8; CHECK-DAG: %[[#float_16:]] = OpTypeFloat 16
9; CHECK-DAG: %[[#vec4_float_16:]] = OpTypeVector %[[#float_16]] 4
10; CHECK-DAG: %[[#vec4_float_32:]] = OpTypeVector %[[#float_32]] 4
11
12define noundef <4 x half> @step_half4(<4 x half> noundef %a, <4 x half> noundef %b) {
13entry:
14  ; CHECK: %[[#]] = OpFunction %[[#vec4_float_16]] None %[[#]]
15  ; CHECK: %[[#arg0:]] = OpFunctionParameter %[[#vec4_float_16]]
16  ; CHECK: %[[#arg1:]] = OpFunctionParameter %[[#vec4_float_16]]
17  ; CHECK: %[[#]] = OpExtInst %[[#vec4_float_16]] %[[#op_ext_glsl]] Step %[[#arg0]] %[[#arg1]]
18  %hlsl.step = call <4 x half> @llvm.spv.step.v4f16(<4 x half> %a, <4 x half> %b)
19  ret <4 x half> %hlsl.step
20}
21
22define noundef <4 x float> @step_float4(<4 x float> noundef %a, <4 x float> noundef %b) {
23entry:
24  ; CHECK: %[[#]] = OpFunction %[[#vec4_float_32]] None %[[#]]
25  ; CHECK: %[[#arg0:]] = OpFunctionParameter %[[#vec4_float_32]]
26  ; CHECK: %[[#arg1:]] = OpFunctionParameter %[[#vec4_float_32]]
27  ; CHECK: %[[#]] = OpExtInst %[[#vec4_float_32]] %[[#op_ext_glsl]] Step %[[#arg0]] %[[#arg1]]
28  %hlsl.step = call <4 x float> @llvm.spv.step.v4f32(<4 x float> %a, <4 x float> %b)
29  ret <4 x float> %hlsl.step
30}
31
32declare <4 x half> @llvm.spv.step.v4f16(<4 x half>, <4 x half>)
33declare <4 x float> @llvm.spv.step.v4f32(<4 x float>, <4 x float>)
34