69 lines · plain
1; RUN: llc -O0 -verify-machineinstrs -mtriple=spirv-unknown-vulkan %s -o - | FileCheck %s2; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv-unknown-vulkan %s -o - -filetype=obj | spirv-val %}3 4; CHECK-DAG: %[[#op_ext_glsl:]] = OpExtInstImport "GLSL.std.450"5 6; CHECK-DAG: %[[#float_32:]] = OpTypeFloat 327; CHECK-DAG: %[[#float_16:]] = OpTypeFloat 168; CHECK-DAG: %[[#float_64:]] = OpTypeFloat 649 10; CHECK-DAG: %[[#vec4_float_32:]] = OpTypeVector %[[#float_32]] 411; CHECK-DAG: %[[#vec4_float_16:]] = OpTypeVector %[[#float_16]] 412; CHECK-DAG: %[[#vec4_float_64:]] = OpTypeVector %[[#float_64]] 413 14define noundef float @rsqrt_float(float noundef %a) {15entry:16; CHECK: %[[#float_32_arg:]] = OpFunctionParameter %[[#float_32]]17; CHECK: %[[#]] = OpExtInst %[[#float_32]] %[[#op_ext_glsl]] InverseSqrt %[[#float_32_arg]]18 %elt.rsqrt = call float @llvm.spv.rsqrt.f32(float %a)19 ret float %elt.rsqrt20}21 22define noundef half @rsqrt_half(half noundef %a) {23entry:24; CHECK: %[[#float_16_arg:]] = OpFunctionParameter %[[#float_16]]25; CHECK: %[[#]] = OpExtInst %[[#float_16]] %[[#op_ext_glsl]] InverseSqrt %[[#float_16_arg]]26 %elt.rsqrt = call half @llvm.spv.rsqrt.f16(half %a)27 ret half %elt.rsqrt28}29 30define noundef double @rsqrt_double(double noundef %a) {31entry:32; CHECK: %[[#float_64_arg:]] = OpFunctionParameter %[[#float_64]]33; CHECK: %[[#]] = OpExtInst %[[#float_64]] %[[#op_ext_glsl]] InverseSqrt %[[#float_64_arg]]34 %elt.rsqrt = call double @llvm.spv.rsqrt.f64(double %a)35 ret double %elt.rsqrt36}37 38define noundef <4 x float> @rsqrt_float_vector(<4 x float> noundef %a) {39entry:40; CHECK: %[[#vec4_float_32_arg:]] = OpFunctionParameter %[[#vec4_float_32]]41; CHECK: %[[#]] = OpExtInst %[[#vec4_float_32]] %[[#op_ext_glsl]] InverseSqrt %[[#vec4_float_32_arg]]42 %elt.rsqrt = call <4 x float> @llvm.spv.rsqrt.v4f32(<4 x float> %a)43 ret <4 x float> %elt.rsqrt44}45 46define noundef <4 x half> @rsqrt_half_vector(<4 x half> noundef %a) {47entry:48; CHECK: %[[#vec4_float_16_arg:]] = OpFunctionParameter %[[#vec4_float_16]]49; CHECK: %[[#]] = OpExtInst %[[#vec4_float_16]] %[[#op_ext_glsl]] InverseSqrt %[[#vec4_float_16_arg]]50 %elt.rsqrt = call <4 x half> @llvm.spv.rsqrt.v4f16(<4 x half> %a)51 ret <4 x half> %elt.rsqrt52}53 54define noundef <4 x double> @rsqrt_double_vector(<4 x double> noundef %a) {55entry:56; CHECK: %[[#vec4_float_64_arg:]] = OpFunctionParameter %[[#vec4_float_64]]57; CHECK: %[[#]] = OpExtInst %[[#vec4_float_64]] %[[#op_ext_glsl]] InverseSqrt %[[#vec4_float_64_arg]]58 %elt.rsqrt = call <4 x double> @llvm.spv.rsqrt.v4f64(<4 x double> %a)59 ret <4 x double> %elt.rsqrt60}61 62declare half @llvm.spv.rsqrt.f16(half)63declare float @llvm.spv.rsqrt.f32(float)64declare double @llvm.spv.rsqrt.f64(double)65 66declare <4 x float> @llvm.spv.rsqrt.v4f32(<4 x float>)67declare <4 x half> @llvm.spv.rsqrt.v4f16(<4 x half>)68declare <4 x double> @llvm.spv.rsqrt.v4f64(<4 x double>)69