[HLSL][DXIL] Implement refract intrinsic (#147342)
- [x] Implement refract using HLSL source in hlsl_intrinsics.h - [x] Implement the refract SPIR-V target built-in in clang/include/clang/Basic/BuiltinsSPIRV.td - [x] Add sema checks for refract to CheckSPIRVBuiltinFunctionCall in clang/lib/Sema/SemaSPIRV.cpp - [x] Add codegen for spv refract to EmitSPIRVBuiltinExpr in CGBuiltin.cpp - [x] Add codegen tests to clang/test/CodeGenHLSL/builtins/refract.hlsl - [x] Add spv codegen test to clang/test/CodeGenSPIRV/Builtins/refract.c - [x] Add sema tests to clang/test/SemaHLSL/BuiltIns/refract-errors.hlsl - [x] Add spv sema tests to clang/test/SemaSPIRV/BuiltIns/refract-errors.c - [x] Create the int_spv_refract intrinsic in IntrinsicsSPIRV.td - [x] In SPIRVInstructionSelector.cpp create the refract lowering and map it to int_spv_refract in SPIRVInstructionSelector::selectIntrinsic. - [x] Create SPIR-V backend test case in llvm/test/CodeGen/SPIRV/hlsl-intrinsics/refract.ll - [x] Check for what OpenCL support is needed. Resolves https://github.com/llvm/llvm-project/issues/99153
This commit is contained in:
raoanag
GitHub
parent
22994edb5f
commit
056f0a10b3
@ -11,3 +11,4 @@ include "clang/Basic/BuiltinsSPIRVBase.td"
|
||||
|
||||
def reflect : SPIRVBuiltin<"void(...)", [NoThrow, Const]>;
|
||||
def faceforward : SPIRVBuiltin<"void(...)", [NoThrow, Const, CustomTypeChecking]>;
|
||||
def refract : SPIRVBuiltin<"void(...)", [NoThrow, Const, CustomTypeChecking]>;
|
||||
|
||||
@ -58,6 +58,18 @@ Value *CodeGenFunction::EmitSPIRVBuiltinExpr(unsigned BuiltinID,
|
||||
/*ReturnType=*/I->getType(), Intrinsic::spv_reflect,
|
||||
ArrayRef<Value *>{I, N}, nullptr, "spv.reflect");
|
||||
}
|
||||
case SPIRV::BI__builtin_spirv_refract: {
|
||||
Value *I = EmitScalarExpr(E->getArg(0));
|
||||
Value *N = EmitScalarExpr(E->getArg(1));
|
||||
Value *eta = EmitScalarExpr(E->getArg(2));
|
||||
assert(E->getArg(0)->getType()->hasFloatingRepresentation() &&
|
||||
E->getArg(1)->getType()->hasFloatingRepresentation() &&
|
||||
E->getArg(2)->getType()->isFloatingType() &&
|
||||
"refract operands must have a float representation");
|
||||
return Builder.CreateIntrinsic(
|
||||
/*ReturnType=*/I->getType(), Intrinsic::spv_refract,
|
||||
ArrayRef<Value *>{I, N, eta}, nullptr, "spv.refract");
|
||||
}
|
||||
case SPIRV::BI__builtin_spirv_smoothstep: {
|
||||
Value *Min = EmitScalarExpr(E->getArg(0));
|
||||
Value *Max = EmitScalarExpr(E->getArg(1));
|
||||
|
||||
@ -71,6 +71,16 @@ constexpr vector<T, L> reflect_vec_impl(vector<T, L> I, vector<T, L> N) {
|
||||
#endif
|
||||
}
|
||||
|
||||
template <typename T, typename U> constexpr T refract_impl(T I, T N, U Eta) {
|
||||
#if (__has_builtin(__builtin_spirv_refract))
|
||||
return __builtin_spirv_refract(I, N, Eta);
|
||||
#endif
|
||||
T Mul = dot(N, I);
|
||||
T K = 1 - Eta * Eta * (1 - Mul * Mul);
|
||||
T Result = (Eta * I - (Eta * Mul + sqrt(K)) * N);
|
||||
return select<T>(K < 0, static_cast<T>(0), Result);
|
||||
}
|
||||
|
||||
template <typename T> constexpr T fmod_impl(T X, T Y) {
|
||||
#if !defined(__DIRECTX__)
|
||||
return __builtin_elementwise_fmod(X, Y);
|
||||
|
||||
@ -475,6 +475,65 @@ reflect(__detail::HLSL_FIXED_VECTOR<float, L> I,
|
||||
return __detail::reflect_vec_impl(I, N);
|
||||
}
|
||||
|
||||
//===----------------------------------------------------------------------===//
|
||||
// refract builtin
|
||||
//===----------------------------------------------------------------------===//
|
||||
|
||||
/// \fn T refract(T I, T N, T eta)
|
||||
/// \brief Returns a refraction using an entering ray, \a I, a surface
|
||||
/// normal, \a N and refraction index \a eta
|
||||
/// \param I The entering ray.
|
||||
/// \param N The surface normal.
|
||||
/// \param eta The refraction index.
|
||||
///
|
||||
/// The return value is a floating-point vector that represents the refraction
|
||||
/// using the refraction index, \a eta, for the direction of the entering ray,
|
||||
/// \a I, off a surface with the normal \a N.
|
||||
///
|
||||
/// This function calculates the refraction vector using the following formulas:
|
||||
/// k = 1.0 - eta * eta * (1.0 - dot(N, I) * dot(N, I))
|
||||
/// if k < 0.0 the result is 0.0
|
||||
/// otherwise, the result is eta * I - (eta * dot(N, I) + sqrt(k)) * N
|
||||
///
|
||||
/// I and N must already be normalized in order to achieve the desired result.
|
||||
///
|
||||
/// I and N must be a scalar or vector whose component type is
|
||||
/// floating-point.
|
||||
///
|
||||
/// eta must be a 16-bit or 32-bit floating-point scalar.
|
||||
///
|
||||
/// Result type, the type of I, and the type of N must all be the same type.
|
||||
|
||||
template <typename T>
|
||||
_HLSL_16BIT_AVAILABILITY(shadermodel, 6.2)
|
||||
const inline __detail::enable_if_t<__detail::is_arithmetic<T>::Value &&
|
||||
__detail::is_same<half, T>::value,
|
||||
T> refract(T I, T N, T eta) {
|
||||
return __detail::refract_impl(I, N, eta);
|
||||
}
|
||||
|
||||
template <typename T>
|
||||
const inline __detail::enable_if_t<
|
||||
__detail::is_arithmetic<T>::Value && __detail::is_same<float, T>::value, T>
|
||||
refract(T I, T N, T eta) {
|
||||
return __detail::refract_impl(I, N, eta);
|
||||
}
|
||||
|
||||
template <int L>
|
||||
_HLSL_16BIT_AVAILABILITY(shadermodel, 6.2)
|
||||
const inline __detail::HLSL_FIXED_VECTOR<half, L> refract(
|
||||
__detail::HLSL_FIXED_VECTOR<half, L> I,
|
||||
__detail::HLSL_FIXED_VECTOR<half, L> N, half eta) {
|
||||
return __detail::refract_impl(I, N, eta);
|
||||
}
|
||||
|
||||
template <int L>
|
||||
const inline __detail::HLSL_FIXED_VECTOR<float, L>
|
||||
refract(__detail::HLSL_FIXED_VECTOR<float, L> I,
|
||||
__detail::HLSL_FIXED_VECTOR<float, L> N, float eta) {
|
||||
return __detail::refract_impl(I, N, eta);
|
||||
}
|
||||
|
||||
//===----------------------------------------------------------------------===//
|
||||
// smoothstep builtin
|
||||
//===----------------------------------------------------------------------===//
|
||||
|
||||
@ -46,6 +46,49 @@ static bool CheckAllArgsHaveSameType(Sema *S, CallExpr *TheCall) {
|
||||
return false;
|
||||
}
|
||||
|
||||
static bool CheckAllArgTypesAreCorrect(
|
||||
Sema *S, CallExpr *TheCall,
|
||||
llvm::ArrayRef<
|
||||
llvm::function_ref<bool(Sema *, SourceLocation, int, QualType)>>
|
||||
Checks) {
|
||||
unsigned NumArgs = TheCall->getNumArgs();
|
||||
assert(Checks.size() == NumArgs &&
|
||||
"Wrong number of checks for Number of args.");
|
||||
// Apply each check to the corresponding argument
|
||||
for (unsigned I = 0; I < NumArgs; ++I) {
|
||||
Expr *Arg = TheCall->getArg(I);
|
||||
if (Checks[I](S, Arg->getBeginLoc(), I + 1, Arg->getType()))
|
||||
return true;
|
||||
}
|
||||
return false;
|
||||
}
|
||||
|
||||
static bool CheckFloatOrHalfRepresentation(Sema *S, SourceLocation Loc,
|
||||
int ArgOrdinal,
|
||||
clang::QualType PassedType) {
|
||||
clang::QualType BaseType =
|
||||
PassedType->isVectorType()
|
||||
? PassedType->castAs<clang::VectorType>()->getElementType()
|
||||
: PassedType;
|
||||
if (!BaseType->isHalfType() && !BaseType->isFloat16Type() &&
|
||||
!BaseType->isFloat32Type())
|
||||
return S->Diag(Loc, diag::err_builtin_invalid_arg_type)
|
||||
<< ArgOrdinal << /* scalar or vector of */ 5 << /* no int */ 0
|
||||
<< /* half or float */ 2 << PassedType;
|
||||
return false;
|
||||
}
|
||||
|
||||
static bool CheckFloatOrHalfScalarRepresentation(Sema *S, SourceLocation Loc,
|
||||
int ArgOrdinal,
|
||||
clang::QualType PassedType) {
|
||||
if (!PassedType->isHalfType() && !PassedType->isFloat16Type() &&
|
||||
!PassedType->isFloat32Type())
|
||||
return S->Diag(Loc, diag::err_builtin_invalid_arg_type)
|
||||
<< ArgOrdinal << /* scalar */ 1 << /* no int */ 0
|
||||
<< /* half or float */ 2 << PassedType;
|
||||
return false;
|
||||
}
|
||||
|
||||
static std::optional<int>
|
||||
processConstant32BitIntArgument(Sema &SemaRef, CallExpr *Call, int Argument) {
|
||||
ExprResult Arg =
|
||||
@ -235,6 +278,43 @@ bool SemaSPIRV::CheckSPIRVBuiltinFunctionCall(const TargetInfo &TI,
|
||||
TheCall->setType(RetTy);
|
||||
break;
|
||||
}
|
||||
case SPIRV::BI__builtin_spirv_refract: {
|
||||
if (SemaRef.checkArgCount(TheCall, 3))
|
||||
return true;
|
||||
|
||||
llvm::function_ref<bool(Sema *, SourceLocation, int, QualType)>
|
||||
ChecksArr[] = {CheckFloatOrHalfRepresentation,
|
||||
CheckFloatOrHalfRepresentation,
|
||||
CheckFloatOrHalfScalarRepresentation};
|
||||
if (CheckAllArgTypesAreCorrect(&SemaRef, TheCall,
|
||||
llvm::ArrayRef(ChecksArr)))
|
||||
return true;
|
||||
// Check that first two arguments are vectors/scalars of the same type
|
||||
QualType Arg0Type = TheCall->getArg(0)->getType();
|
||||
if (!SemaRef.getASTContext().hasSameUnqualifiedType(
|
||||
Arg0Type, TheCall->getArg(1)->getType()))
|
||||
return SemaRef.Diag(TheCall->getBeginLoc(),
|
||||
diag::err_vec_builtin_incompatible_vector)
|
||||
<< TheCall->getDirectCallee() << /* first two */ 0
|
||||
<< SourceRange(TheCall->getArg(0)->getBeginLoc(),
|
||||
TheCall->getArg(1)->getEndLoc());
|
||||
|
||||
// Check that scalar type of 3rd arg is same as base type of first two args
|
||||
clang::QualType BaseType =
|
||||
Arg0Type->isVectorType()
|
||||
? Arg0Type->castAs<clang::VectorType>()->getElementType()
|
||||
: Arg0Type;
|
||||
if (!SemaRef.getASTContext().hasSameUnqualifiedType(
|
||||
BaseType, TheCall->getArg(2)->getType()))
|
||||
return SemaRef.Diag(TheCall->getBeginLoc(),
|
||||
diag::err_hlsl_builtin_scalar_vector_mismatch)
|
||||
<< /* all */ 0 << TheCall->getDirectCallee() << Arg0Type
|
||||
<< TheCall->getArg(2)->getType();
|
||||
|
||||
QualType RetTy = TheCall->getArg(0)->getType();
|
||||
TheCall->setType(RetTy);
|
||||
break;
|
||||
}
|
||||
case SPIRV::BI__builtin_spirv_smoothstep: {
|
||||
if (SemaRef.checkArgCount(TheCall, 3))
|
||||
return true;
|
||||
|
||||
244
clang/test/CodeGenHLSL/builtins/refract.hlsl
Normal file
244
clang/test/CodeGenHLSL/builtins/refract.hlsl
Normal file
@ -0,0 +1,244 @@
|
||||
// RUN: %clang_cc1 -finclude-default-header -triple \
|
||||
// RUN: dxil-pc-shadermodel6.3-library %s -fnative-half-type \
|
||||
// RUN: -emit-llvm -o - | FileCheck %s
|
||||
// RUN: %clang_cc1 -finclude-default-header -triple \
|
||||
// RUN: spirv-unknown-vulkan-compute %s -fnative-half-type \
|
||||
// RUN: -emit-llvm -o - | FileCheck %s --check-prefix=SPVCHECK
|
||||
|
||||
// CHECK-LABEL: define hidden noundef nofpclass(nan inf) half @_Z17test_refract_halfDhDhDh(
|
||||
// CHECK-SAME: half noundef nofpclass(nan inf) [[I:%.*]], half noundef nofpclass(nan inf) [[N:%.*]], half noundef nofpclass(nan inf) [[ETA:%.*]]) #[[ATTR0:[0-9]+]] {
|
||||
// CHECK: [[ENTRY:.*:]]
|
||||
// CHECK: [[MUL_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn half %{{.*}}, %{{.*}}
|
||||
// CHECK: [[MUL1_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn half %{{.*}}, %{{.*}}
|
||||
// CHECK: [[MUL2_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn half %{{.*}}, %{{.*}}
|
||||
// CHECK: [[SUB_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn half 0xH3C00, [[MUL2_I]]
|
||||
// CHECK: [[MUL3_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn half [[MUL1_I]], [[SUB_I]]
|
||||
// CHECK: [[SUB4_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn half 0xH3C00, [[MUL3_I]]
|
||||
// CHECK: [[MUL5_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn half %{{.*}}, %{{.*}}
|
||||
// CHECK: [[MUL6_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn half %{{.*}}, %{{.*}}
|
||||
// CHECK: [[TMP0:%.*]] = call reassoc nnan ninf nsz arcp afn half @llvm.sqrt.f16(half %{{.*}})
|
||||
// CHECK: [[ADD_I:%.*]] = fadd reassoc nnan ninf nsz arcp afn half [[MUL6_I]], %{{.*}}
|
||||
// CHECK: [[MUL7_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn half [[ADD_I]], %{{.*}}
|
||||
// CHECK: [[SUB8_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn half %{{.*}}, [[MUL7_I]]
|
||||
// CHECK: [[CMP_I:%.*]] = fcmp reassoc nnan ninf nsz arcp afn olt half %{{.*}}, 0xH0000
|
||||
// CHECK: [[HLSL_SELECT_I:%.*]] = select reassoc nnan ninf nsz arcp afn i1 [[CMP_I]], half 0xH0000, half %{{.*}}
|
||||
// CHECK: ret half [[HLSL_SELECT_I]]
|
||||
//
|
||||
// SPVCHECK-LABEL: define hidden spir_func noundef nofpclass(nan inf) half @_Z17test_refract_halfDhDhDh(
|
||||
// SPVCHECK-SAME: half noundef nofpclass(nan inf) [[I:%.*]], half noundef nofpclass(nan inf) [[N:%.*]], half noundef nofpclass(nan inf) [[ETA:%.*]]) #[[ATTR0:[0-9]+]]
|
||||
// SPVCHECK: [[ENTRY:.*:]]
|
||||
// SPVCHECK: [[SPV_REFRACT_I:%.*]] = call reassoc nnan ninf nsz arcp afn noundef half @llvm.spv.refract.f16.f16(half %{{.*}}, half %{{.*}}, half %{{.*}})
|
||||
// SPVCHECK: ret half [[SPV_REFRACT_I]]
|
||||
//
|
||||
half test_refract_half(half I, half N, half ETA) {
|
||||
return refract(I, N, ETA);
|
||||
}
|
||||
|
||||
// CHECK-LABEL: define hidden noundef nofpclass(nan inf) <2 x half> @_Z18test_refract_half2Dv2_DhS_Dh(
|
||||
// CHECK-SAME: <2 x half> noundef nofpclass(nan inf) [[I:%.*]], <2 x half> noundef nofpclass(nan inf) [[N:%.*]], half noundef nofpclass(nan inf) [[ETA:%.*]]) #[[ATTR0:[0-9]+]] {
|
||||
// CHECK: [[ENTRY:.*:]]
|
||||
// CHECK: [[HLSL_DOT_I:%.*]] = call reassoc nnan ninf nsz arcp afn half @llvm.dx.fdot.v2f16(<2 x half> %{{.*}}, <2 x half> %{{.*}})
|
||||
// CHECK: [[MUL_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn half %{{.*}}, %{{.*}}
|
||||
// CHECK: [[MUL3_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <2 x half> %{{.*}}, %{{.*}}
|
||||
// CHECK: [[SUB_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn <2 x half> splat (half 0xH3C00), [[MUL3_I]]
|
||||
// CHECK: [[MUL4_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <2 x half> %{{.*}}, [[SUB_I]]
|
||||
// CHECK: [[SUB5_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn <2 x half> splat (half 0xH3C00), [[MUL4_I]]
|
||||
// CHECK: [[MUL8_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <2 x half> %{{.*}}, %{{.*}}
|
||||
// CHECK: [[MUL11_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <2 x half> %{{.*}}, %{{.*}}
|
||||
// CHECK: [[TMP17:%.*]] = call reassoc nnan ninf nsz arcp afn <2 x half> @llvm.sqrt.v2f16(<2 x half> %{{.*}})
|
||||
// CHECK: [[ADD_I:%.*]] = fadd reassoc nnan ninf nsz arcp afn <2 x half> [[MUL11_I]], [[TMP17]]
|
||||
// CHECK: [[MUL12_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <2 x half> [[ADD_I]], %{{.*}}
|
||||
// CHECK: [[SUB13_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn <2 x half> [[MUL8_I]], [[MUL12_I]]
|
||||
// CHECK: [[CMP_I:%.*]] = fcmp reassoc nnan ninf nsz arcp afn olt <2 x half> %{{.*}}, zeroinitializer
|
||||
// CHECK: [[CAST:%.*]] = extractelement <2 x i1> [[CMP_I]], i32 0
|
||||
// CHECK: [[HLSL_SELECT_I:%.*]] = select reassoc nnan ninf nsz arcp afn i1 [[CAST]], <2 x half> zeroinitializer, <2 x half> %{{.*}}
|
||||
// CHECK: ret <2 x half> [[HLSL_SELECT_I]]
|
||||
//
|
||||
// SPVCHECK-LABEL: define hidden spir_func noundef nofpclass(nan inf) <2 x half> @_Z18test_refract_half2Dv2_DhS_Dh(
|
||||
// SPVCHECK-SAME: <2 x half> noundef nofpclass(nan inf) [[I:%.*]], <2 x half> noundef nofpclass(nan inf) [[N:%.*]], half noundef nofpclass(nan inf) [[ETA:%.*]]) #[[ATTR0:[0-9]+]] {
|
||||
// SPVCHECK: [[ENTRY:.*:]]
|
||||
// SPVCHECK: [[SPV_REFRACT_I:%.*]] = call reassoc nnan ninf nsz arcp afn noundef <2 x half> @llvm.spv.refract.v2f16.f16(<2 x half> %{{.*}}, <2 x half> %{{.*}}, half %{{.*}})
|
||||
// SPVCHECK: ret <2 x half> [[SPV_REFRACT_I]]
|
||||
//
|
||||
half2 test_refract_half2(half2 I, half2 N, half ETA) {
|
||||
return refract(I, N, ETA);
|
||||
}
|
||||
|
||||
// CHECK-LABEL: define hidden noundef nofpclass(nan inf) <3 x half> @_Z18test_refract_half3Dv3_DhS_Dh(
|
||||
// CHECK-SAME: <3 x half> noundef nofpclass(nan inf) [[I:%.*]], <3 x half> noundef nofpclass(nan inf) [[N:%.*]], half noundef nofpclass(nan inf) [[ETA:%.*]]) #[[ATTR0:[0-9]+]] {
|
||||
// CHECK: [[ENTRY:.*:]]
|
||||
// CHECK: [[HLSL_DOT_I:%.*]] = call reassoc nnan ninf nsz arcp afn half @llvm.dx.fdot.v3f16(<3 x half> %{{.*}}, <3 x half> %{{.*}})
|
||||
// CHECK: [[MUL_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn half %{{.*}}, %{{.*}}
|
||||
// CHECK: [[MUL3_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <3 x half> %{{.*}}, %{{.*}}
|
||||
// CHECK: [[SUB_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn <3 x half> splat (half 0xH3C00), [[MUL3_I]]
|
||||
// CHECK: [[MUL4_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <3 x half> %{{.*}}, [[SUB_I]]
|
||||
// CHECK: [[SUB5_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn <3 x half> splat (half 0xH3C00), [[MUL4_I]]
|
||||
// CHECK: [[MUL8_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <3 x half> %{{.*}}, %{{.*}}
|
||||
// CHECK: [[MUL11_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <3 x half> %{{.*}}, %{{.*}}
|
||||
// CHECK: [[TMP17:%.*]] = call reassoc nnan ninf nsz arcp afn <3 x half> @llvm.sqrt.v3f16(<3 x half> %{{.*}})
|
||||
// CHECK: [[ADD_I:%.*]] = fadd reassoc nnan ninf nsz arcp afn <3 x half> [[MUL11_I]], [[TMP17]]
|
||||
// CHECK: [[MUL12_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <3 x half> [[ADD_I]], %{{.*}}
|
||||
// CHECK: [[SUB13_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn <3 x half> [[MUL8_I]], [[MUL12_I]]
|
||||
// CHECK: [[CMP_I:%.*]] = fcmp reassoc nnan ninf nsz arcp afn olt <3 x half> %{{.*}}, zeroinitializer
|
||||
// CHECK: [[CAST:%.*]] = extractelement <3 x i1> [[CMP_I]], i32 0
|
||||
// CHECK: [[HLSL_SELECT_I:%.*]] = select reassoc nnan ninf nsz arcp afn i1 [[CAST]], <3 x half> zeroinitializer, <3 x half> %{{.*}}
|
||||
// CHECK: ret <3 x half> [[HLSL_SELECT_I]]
|
||||
//
|
||||
// SPVCHECK-LABEL: define hidden spir_func noundef nofpclass(nan inf) <3 x half> @_Z18test_refract_half3Dv3_DhS_Dh(
|
||||
// SPVCHECK-SAME: <3 x half> noundef nofpclass(nan inf) [[I:%.*]], <3 x half> noundef nofpclass(nan inf) [[N:%.*]], half noundef nofpclass(nan inf) [[ETA:%.*]]) #[[ATTR0:[0-9]+]] {
|
||||
// SPVCHECK: [[ENTRY:.*:]]
|
||||
// SPVCHECK: [[SPV_REFRACT_I:%.*]] = call reassoc nnan ninf nsz arcp afn noundef <3 x half> @llvm.spv.refract.v3f16.f16(<3 x half> %{{.*}}, <3 x half> %{{.*}}, half %{{.*}})
|
||||
// SPVCHECK: ret <3 x half> [[SPV_REFRACT_I]]
|
||||
//
|
||||
half3 test_refract_half3(half3 I, half3 N, half ETA) {
|
||||
return refract(I, N, ETA);
|
||||
}
|
||||
|
||||
// CHECK-LABEL: define hidden noundef nofpclass(nan inf) <4 x half> @_Z18test_refract_half4Dv4_DhS_Dh(
|
||||
// CHECK-SAME: <4 x half> noundef nofpclass(nan inf) [[I:%.*]], <4 x half> noundef nofpclass(nan inf) [[N:%.*]], half noundef nofpclass(nan inf) [[ETA:%.*]]) #[[ATTR0:[0-9]+]] {
|
||||
// CHECK: [[ENTRY:.*:]]
|
||||
// CHECK: [[HLSL_DOT_I:%.*]] = call reassoc nnan ninf nsz arcp afn half @llvm.dx.fdot.v4f16(<4 x half> %{{.*}}, <4 x half> %{{.*}})
|
||||
// CHECK: [[MUL_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn half %{{.*}}, %{{.*}}
|
||||
// CHECK: [[MUL3_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <4 x half> %{{.*}}, %{{.*}}
|
||||
// CHECK: [[SUB_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn <4 x half> splat (half 0xH3C00), [[MUL3_I]]
|
||||
// CHECK: [[MUL4_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <4 x half> %{{.*}}, [[SUB_I]]
|
||||
// CHECK: [[SUB5_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn <4 x half> splat (half 0xH3C00), [[MUL4_I]]
|
||||
// CHECK: [[MUL8_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <4 x half> %{{.*}}, %{{.*}}
|
||||
// CHECK: [[MUL11_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <4 x half> %{{.*}}, %{{.*}}
|
||||
// CHECK: [[TMP17:%.*]] = call reassoc nnan ninf nsz arcp afn <4 x half> @llvm.sqrt.v4f16(<4 x half> %{{.*}})
|
||||
// CHECK: [[ADD_I:%.*]] = fadd reassoc nnan ninf nsz arcp afn <4 x half> [[MUL11_I]], [[TMP17]]
|
||||
// CHECK: [[MUL12_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <4 x half> [[ADD_I]], %{{.*}}
|
||||
// CHECK: [[SUB13_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn <4 x half> [[MUL8_I]], [[MUL12_I]]
|
||||
// CHECK: [[CMP_I:%.*]] = fcmp reassoc nnan ninf nsz arcp afn olt <4 x half> %{{.*}}, zeroinitializer
|
||||
// CHECK: [[CAST:%.*]] = extractelement <4 x i1> [[CMP_I]], i32 0
|
||||
// CHECK: [[HLSL_SELECT_I:%.*]] = select reassoc nnan ninf nsz arcp afn i1 [[CAST]], <4 x half> zeroinitializer, <4 x half> %{{.*}}
|
||||
// CHECK: ret <4 x half> [[HLSL_SELECT_I]]
|
||||
//
|
||||
// SPVCHECK-LABEL: define hidden spir_func noundef nofpclass(nan inf) <4 x half> @_Z18test_refract_half4Dv4_DhS_Dh(
|
||||
// SPVCHECK-SAME: <4 x half> noundef nofpclass(nan inf) [[I:%.*]], <4 x half> noundef nofpclass(nan inf) [[N:%.*]], half noundef nofpclass(nan inf) [[ETA:%.*]]) #[[ATTR0:[0-9]+]] {
|
||||
// SPVCHECK: [[ENTRY:.*:]]
|
||||
// SPVCHECK: [[SPV_REFRACT_I:%.*]] = call reassoc nnan ninf nsz arcp afn noundef <4 x half> @llvm.spv.refract.v4f16.f16(<4 x half> %{{.*}}, <4 x half> %{{.*}}, half %{{.*}})
|
||||
// SPVCHECK: ret <4 x half> [[SPV_REFRACT_I]]
|
||||
//
|
||||
half4 test_refract_half4(half4 I, half4 N, half ETA) {
|
||||
return refract(I, N, ETA);
|
||||
}
|
||||
|
||||
// CHECK-LABEL: define hidden noundef nofpclass(nan inf) float @_Z18test_refract_floatfff(
|
||||
// CHECK-SAME: float noundef nofpclass(nan inf) [[I:%.*]], float noundef nofpclass(nan inf) [[N:%.*]], float noundef nofpclass(nan inf) [[ETA:%.*]]) #[[ATTR0:[0-9]+]] {
|
||||
// CHECK: [[ENTRY:.*:]]
|
||||
// CHECK: [[MUL_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn float %{{.*}}, %{{.*}}
|
||||
// CHECK: [[MUL1_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn float %{{.*}}, %{{.*}}
|
||||
// CHECK: [[MUL2_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn float %{{.*}}, %{{.*}}
|
||||
// CHECK: [[SUB_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn float 1.000000e+00, [[MUL2_I]]
|
||||
// CHECK: [[MUL3_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn float [[MUL1_I]], [[SUB_I]]
|
||||
// CHECK: [[SUB4_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn float 1.000000e+00, [[MUL3_I]]
|
||||
// CHECK: [[MUL5_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn float %{{.*}}, %{{.*}}
|
||||
// CHECK: [[MUL6_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn float %{{.*}}, %{{.*}}
|
||||
// CHECK: [[TMP17:%.*]] = call reassoc nnan ninf nsz arcp afn float @llvm.sqrt.f32(float %{{.*}})
|
||||
// CHECK: [[ADD_I:%.*]] = fadd reassoc nnan ninf nsz arcp afn float [[MUL6_I]], %{{.*}}
|
||||
// CHECK: [[MUL7_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn float [[ADD_I]], %{{.*}}
|
||||
// CHECK: [[SUB8_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn float %{{.*}}, [[MUL7_I]]
|
||||
// CHECK: [[CMP_I:%.*]] = fcmp reassoc nnan ninf nsz arcp afn olt float %{{.*}}, 0.000000e+00
|
||||
// CHECK: [[HLSL_SELECT_I:%.*]] = select reassoc nnan ninf nsz arcp afn i1 [[CMP_I]], float 0.000000e+00, float %{{.*}}
|
||||
// CHECK: ret float [[HLSL_SELECT_I]]
|
||||
//
|
||||
// SPVCHECK-LABEL: define hidden spir_func noundef nofpclass(nan inf) float @_Z18test_refract_floatfff(
|
||||
// SPVCHECK-SAME: float noundef nofpclass(nan inf) [[I:%.*]], float noundef nofpclass(nan inf) [[N:%.*]], float noundef nofpclass(nan inf) [[ETA:%.*]]) #[[ATTR0:[0-9]+]] {
|
||||
// SPVCHECK: [[ENTRY:.*:]]
|
||||
// SPVCHECK: [[SPV_REFRACT_I:%.*]] = call reassoc nnan ninf nsz arcp afn noundef float @llvm.spv.refract.f32.f32(float %{{.*}}, float %{{.*}}, float %{{.*}})
|
||||
// SPVCHECK: ret float [[SPV_REFRACT_I]]
|
||||
//
|
||||
float test_refract_float(float I, float N, float ETA) {
|
||||
return refract(I, N, ETA);
|
||||
}
|
||||
|
||||
// CHECK-LABEL: define hidden noundef nofpclass(nan inf) <2 x float> @_Z19test_refract_float2Dv2_fS_f(
|
||||
// CHECK-SAME: <2 x float> noundef nofpclass(nan inf) [[I:%.*]], <2 x float> noundef nofpclass(nan inf) [[N:%.*]], float noundef nofpclass(nan inf) [[ETA:%.*]]) #[[ATTR0:[0-9]+]] {
|
||||
// CHECK: [[ENTRY:.*:]]
|
||||
// CHECK: [[HLSL_DOT_I:%.*]] = call reassoc nnan ninf nsz arcp afn float @llvm.dx.fdot.v2f32(<2 x float> %{{.*}}, <2 x float> %{{.*}})
|
||||
// CHECK: [[MUL_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn float %{{.*}}, %{{.*}}
|
||||
// CHECK: [[MUL3_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <2 x float> %{{.*}}, %{{.*}}
|
||||
// CHECK: [[SUB_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn <2 x float> splat (float 1.000000e+00), [[MUL3_I]]
|
||||
// CHECK: [[MUL4_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <2 x float> %{{.*}}, [[SUB_I]]
|
||||
// CHECK: [[SUB5_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn <2 x float> splat (float 1.000000e+00), [[MUL4_I]]
|
||||
// CHECK: [[MUL8_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <2 x float> %{{.*}}, %{{.*}}
|
||||
// CHECK: [[MUL11_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <2 x float> %{{.*}}, %{{.*}}
|
||||
// CHECK: [[TMP17:%.*]] = call reassoc nnan ninf nsz arcp afn <2 x float> @llvm.sqrt.v2f32(<2 x float> %{{.*}})
|
||||
// CHECK: [[ADD_I:%.*]] = fadd reassoc nnan ninf nsz arcp afn <2 x float> [[MUL11_I]], [[TMP17]]
|
||||
// CHECK: [[MUL12_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <2 x float> [[ADD_I]], %{{.*}}
|
||||
// CHECK: [[SUB13_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn <2 x float> [[MUL8_I]], [[MUL12_I]]
|
||||
// CHECK: [[CMP_I:%.*]] = fcmp reassoc nnan ninf nsz arcp afn olt <2 x float> %{{.*}}, zeroinitializer
|
||||
// CHECK: [[CAST:%.*]] = extractelement <2 x i1> [[CMP_I]], i32 0
|
||||
// CHECK: [[HLSL_SELECT_I:%.*]] = select reassoc nnan ninf nsz arcp afn i1 [[CAST]], <2 x float> zeroinitializer, <2 x float> %{{.*}}
|
||||
// CHECK: ret <2 x float> [[HLSL_SELECT_I]]
|
||||
//
|
||||
// SPVCHECK-LABEL: define hidden spir_func noundef nofpclass(nan inf) <2 x float> @_Z19test_refract_float2Dv2_fS_f(
|
||||
// SPVCHECK-SAME: <2 x float> noundef nofpclass(nan inf) [[I:%.*]], <2 x float> noundef nofpclass(nan inf) [[N:%.*]], float noundef nofpclass(nan inf) [[ETA:%.*]]) #[[ATTR0:[0-9]+]] {
|
||||
// SPVCHECK: [[ENTRY:.*:]]
|
||||
// SPVCHECK: [[SPV_REFRACT_I:%.*]] = call reassoc nnan ninf nsz arcp afn noundef <2 x float> @llvm.spv.refract.v2f32.f32(<2 x float> %{{.*}}, <2 x float> %{{.*}}, float %{{.*}})
|
||||
// SPVCHECK: ret <2 x float> [[SPV_REFRACT_I]]
|
||||
//
|
||||
float2 test_refract_float2(float2 I, float2 N, float ETA) {
|
||||
return refract(I, N, ETA);
|
||||
}
|
||||
|
||||
// CHECK-LABEL: define hidden noundef nofpclass(nan inf) <3 x float> @_Z19test_refract_float3Dv3_fS_f(
|
||||
// CHECK-SAME: <3 x float> noundef nofpclass(nan inf) [[I:%.*]], <3 x float> noundef nofpclass(nan inf) [[N:%.*]], float noundef nofpclass(nan inf) [[ETA:%.*]]) #[[ATTR0:[0-9]+]] {
|
||||
// CHECK-NEXT: [[ENTRY:.*:]]
|
||||
// CHECK: [[HLSL_DOT_I:%.*]] = call reassoc nnan ninf nsz arcp afn float @llvm.dx.fdot.v3f32(<3 x float> %{{.*}}, <3 x float> %{{.*}})
|
||||
// CHECK: [[MUL_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn float %{{.*}}, %{{.*}}
|
||||
// CHECK: [[MUL3_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <3 x float> %{{.*}}, %{{.*}}
|
||||
// CHECK: [[SUB_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn <3 x float> splat (float 1.000000e+00), [[MUL3_I]]
|
||||
// CHECK: [[MUL4_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <3 x float> %{{.*}}, [[SUB_I]]
|
||||
// CHECK: [[SUB5_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn <3 x float> splat (float 1.000000e+00), [[MUL4_I]]
|
||||
// CHECK: [[MUL8_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <3 x float> %{{.*}}, %{{.*}}
|
||||
// CHECK: [[MUL11_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <3 x float> %{{.*}}, %{{.*}}
|
||||
// CHECK: [[TMP17:%.*]] = call reassoc nnan ninf nsz arcp afn <3 x float> @llvm.sqrt.v3f32(<3 x float> %{{.*}})
|
||||
// CHECK: [[ADD_I:%.*]] = fadd reassoc nnan ninf nsz arcp afn <3 x float> [[MUL11_I]], [[TMP17]]
|
||||
// CHECK: [[MUL12_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <3 x float> [[ADD_I]], %{{.*}}
|
||||
// CHECK: [[SUB13_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn <3 x float> [[MUL8_I]], [[MUL12_I]]
|
||||
// CHECK: [[CMP_I:%.*]] = fcmp reassoc nnan ninf nsz arcp afn olt <3 x float> %{{.*}}, zeroinitializer
|
||||
// CHECK: [[CAST:%.*]] = extractelement <3 x i1> [[CMP_I]], i32 0
|
||||
// CHECK: [[HLSL_SELECT_I:%.*]] = select reassoc nnan ninf nsz arcp afn i1 [[CAST]], <3 x float> zeroinitializer, <3 x float> %{{.*}}
|
||||
// CHECK: ret <3 x float> [[HLSL_SELECT_I]]
|
||||
//
|
||||
// SPVCHECK-LABEL: define hidden spir_func noundef nofpclass(nan inf) <3 x float> @_Z19test_refract_float3Dv3_fS_f(
|
||||
// SPVCHECK-SAME: <3 x float> noundef nofpclass(nan inf) [[I:%.*]], <3 x float> noundef nofpclass(nan inf) [[N:%.*]], float noundef nofpclass(nan inf) [[ETA:%.*]]) #[[ATTR0:[0-9]+]] {
|
||||
// SPVCHECK: [[ENTRY:.*:]]
|
||||
// SPVCHECK: [[SPV_REFRACT_I:%.*]] = call reassoc nnan ninf nsz arcp afn noundef <3 x float> @llvm.spv.refract.v3f32.f32(<3 x float> %{{.*}}, <3 x float> %{{.*}}, float %{{.*}})
|
||||
// SPVCHECK: ret <3 x float> [[SPV_REFRACT_I]]
|
||||
//
|
||||
float3 test_refract_float3(float3 I, float3 N, float ETA) {
|
||||
return refract(I, N, ETA);
|
||||
}
|
||||
|
||||
// CHECK-LABEL: define hidden noundef nofpclass(nan inf) <4 x float> @_Z19test_refract_float4Dv4_fS_f
|
||||
// CHECK-SAME: <4 x float> noundef nofpclass(nan inf) [[I:%.*]], <4 x float> noundef nofpclass(nan inf) [[N:%.*]], float noundef nofpclass(nan inf) [[ETA:%.*]]) #[[ATTR0:[0-9]+]] {
|
||||
// CHECK-NEXT: [[ENTRY:.*:]]
|
||||
// CHECK: [[HLSL_DOT_I:%.*]] = call reassoc nnan ninf nsz arcp afn float @llvm.dx.fdot.v4f32(<4 x float> %{{.*}}, <4 x float> %{{.*}})
|
||||
// CHECK: [[MUL_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn float %{{.*}}, %{{.*}}
|
||||
// CHECK: [[MUL3_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <4 x float> %{{.*}}, %{{.*}}
|
||||
// CHECK: [[SUB_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn <4 x float> splat (float 1.000000e+00), [[MUL3_I]]
|
||||
// CHECK: [[MUL4_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <4 x float> %{{.*}}, [[SUB_I]]
|
||||
// CHECK: [[SUB5_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn <4 x float> splat (float 1.000000e+00), [[MUL4_I]]
|
||||
// CHECK: [[MUL8_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <4 x float> %{{.*}}, %{{.*}}
|
||||
// CHECK: [[MUL11_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <4 x float> %{{.*}}, %{{.*}}
|
||||
// CHECK: [[TMP17:%.*]] = call reassoc nnan ninf nsz arcp afn <4 x float> @llvm.sqrt.v4f32(<4 x float> %{{.*}})
|
||||
// CHECK: [[ADD_I:%.*]] = fadd reassoc nnan ninf nsz arcp afn <4 x float> [[MUL11_I]], [[TMP17]]
|
||||
// CHECK: [[MUL12_I:%.*]] = fmul reassoc nnan ninf nsz arcp afn <4 x float> [[ADD_I]], %{{.*}}
|
||||
// CHECK: [[SUB13_I:%.*]] = fsub reassoc nnan ninf nsz arcp afn <4 x float> [[MUL8_I]], [[MUL12_I]]
|
||||
// CHECK: [[CMP_I:%.*]] = fcmp reassoc nnan ninf nsz arcp afn olt <4 x float> %{{.*}}, zeroinitializer
|
||||
// CHECK: [[CAST:%.*]] = extractelement <4 x i1> [[CMP_I]], i32 0
|
||||
// CHECK: [[HLSL_SELECT_I:%.*]] = select reassoc nnan ninf nsz arcp afn i1 [[CAST]], <4 x float> zeroinitializer, <4 x float> %{{.*}}
|
||||
// CHECK: ret <4 x float> [[HLSL_SELECT_I]]
|
||||
|
||||
// SPVCHECK-LABEL: define hidden spir_func noundef nofpclass(nan inf) <4 x float> @_Z19test_refract_float4Dv4_fS_f(
|
||||
// SPVCHECK-SAME: <4 x float> noundef nofpclass(nan inf) %{{.*}}, <4 x float> noundef nofpclass(nan inf) %{{.*}}, float noundef nofpclass(nan inf) %{{.*}}) #[[ATTR0:[0-9]+]] {
|
||||
// SPVCHECK: [[ENTRY:.*:]]
|
||||
// SPVCHECK: [[SPV_REFRACT_I:%.*]] = call reassoc nnan ninf nsz arcp afn noundef <4 x float> @llvm.spv.refract.v4f32.f32(<4 x float> %{{.*}}, <4 x float> %{{.*}}, float %{{.*}})
|
||||
// SPVCHECK: ret <4 x float> [[SPV_REFRACT_I]]
|
||||
//
|
||||
float4 test_refract_float4(float4 I, float4 N, float ETA) {
|
||||
return refract(I, N, ETA);
|
||||
}
|
||||
74
clang/test/CodeGenSPIRV/Builtins/refract.c
Normal file
74
clang/test/CodeGenSPIRV/Builtins/refract.c
Normal file
@ -0,0 +1,74 @@
|
||||
// RUN: %clang_cc1 -O1 -triple spirv-pc-vulkan-compute %s -fnative-half-type -emit-llvm -o - | FileCheck %s
|
||||
|
||||
typedef _Float16 half;
|
||||
typedef half half2 __attribute__((ext_vector_type(2)));
|
||||
typedef half half3 __attribute__((ext_vector_type(3)));
|
||||
typedef half half4 __attribute__((ext_vector_type(4)));
|
||||
typedef float float2 __attribute__((ext_vector_type(2)));
|
||||
typedef float float3 __attribute__((ext_vector_type(3)));
|
||||
typedef float float4 __attribute__((ext_vector_type(4)));
|
||||
|
||||
// CHECK-LABEL: define spir_func half @test_refract_half(
|
||||
// CHECK-SAME: half noundef [[I:%.*]], half noundef [[N:%.*]], half noundef [[ETA:%.*]]) local_unnamed_addr #[[ATTR0:[0-9]+]] {
|
||||
// CHECK-NEXT: [[ENTRY:.*:]]
|
||||
// CHECK: [[SPV_REFRACT:%.*]] = tail call half @llvm.spv.refract.f16.f16(half [[I]], half [[N]], half [[ETA]])
|
||||
// CHECK-NEXT: ret half [[SPV_REFRACT]]
|
||||
//
|
||||
half test_refract_half(half I, half N, half eta) { return __builtin_spirv_refract(I, N, eta); }
|
||||
|
||||
// CHECK-LABEL: define spir_func <2 x half> @test_refract_half2(
|
||||
// CHECK-SAME: <2 x half> noundef [[I:%.*]], <2 x half> noundef [[N:%.*]], half noundef [[ETA:%.*]]) local_unnamed_addr #[[ATTR0:[0-9]+]] {
|
||||
// CHECK-NEXT: [[ENTRY:.*:]]
|
||||
// CHECK: [[SPV_REFRACT:%.*]] = tail call <2 x half> @llvm.spv.refract.v2f16.f16(<2 x half> [[I]], <2 x half> [[N]], half [[ETA]])
|
||||
// CHECK-NEXT: ret <2 x half> [[SPV_REFRACT]]
|
||||
//
|
||||
half2 test_refract_half2(half2 I, half2 N, half eta) { return __builtin_spirv_refract(I, N, eta); }
|
||||
|
||||
// CHECK-LABEL: define spir_func <3 x half> @test_refract_half3(
|
||||
// CHECK-SAME: <3 x half> noundef [[I:%.*]], <3 x half> noundef [[N:%.*]], half noundef [[ETA:%.*]]) local_unnamed_addr #[[ATTR0]] {
|
||||
// CHECK-NEXT: [[ENTRY:.*:]]
|
||||
// CHECK-NEXT: [[SPV_REFRACT:%.*]] = tail call <3 x half> @llvm.spv.refract.v3f16.f16(<3 x half> [[I]], <3 x half> [[N]], half [[ETA]])
|
||||
// CHECK-NEXT: ret <3 x half> [[SPV_REFRACT]]
|
||||
//
|
||||
half3 test_refract_half3(half3 I, half3 N, half eta) { return __builtin_spirv_refract(I, N, eta); }
|
||||
|
||||
// CHECK-LABEL: define spir_func <4 x half> @test_refract_half4(
|
||||
// CHECK-SAME: <4 x half> noundef [[I:%.*]], <4 x half> noundef [[N:%.*]], half noundef [[ETA:%.*]]) local_unnamed_addr #[[ATTR0]] {
|
||||
// CHECK-NEXT: [[ENTRY:.*:]]
|
||||
// CHECK-NEXT: [[SPV_REFRACT:%.*]] = tail call <4 x half> @llvm.spv.refract.v4f16.f16(<4 x half> [[I]], <4 x half> [[N]], half [[ETA]])
|
||||
// CHECK-NEXT: ret <4 x half> [[SPV_REFRACT]]
|
||||
//
|
||||
half4 test_refract_half4(half4 I, half4 N, half eta) { return __builtin_spirv_refract(I, N, eta); }
|
||||
|
||||
|
||||
// CHECK-LABEL: define spir_func float @test_refract_float(
|
||||
// CHECK-SAME: float noundef [[I:%.*]], float noundef [[N:%.*]], float noundef [[ETA:%.*]]) local_unnamed_addr #[[ATTR0:[0-9]+]] {
|
||||
// CHECK-NEXT: [[ENTRY:.*:]]
|
||||
// CHECK: [[SPV_REFRACT:%.*]] = tail call float @llvm.spv.refract.f32.f32(float [[I]], float [[N]], float [[ETA]])
|
||||
// CHECK-NEXT: ret float [[SPV_REFRACT]]
|
||||
//
|
||||
float test_refract_float(float I, float N, float eta) { return __builtin_spirv_refract(I, N, eta); }
|
||||
|
||||
// CHECK-LABEL: define spir_func <2 x float> @test_refract_float2(
|
||||
// CHECK-SAME: <2 x float> noundef [[I:%.*]], <2 x float> noundef [[N:%.*]], float noundef [[ETA:%.*]]) local_unnamed_addr #[[ATTR0:[0-9]+]] {
|
||||
// CHECK-NEXT: [[ENTRY:.*:]]
|
||||
// CHECK: [[SPV_REFRACT:%.*]] = tail call <2 x float> @llvm.spv.refract.v2f32.f32(<2 x float> [[I]], <2 x float> [[N]], float [[ETA]])
|
||||
// CHECK-NEXT: ret <2 x float> [[SPV_REFRACT]]
|
||||
//
|
||||
float2 test_refract_float2(float2 I, float2 N, float eta) { return __builtin_spirv_refract(I, N, eta); }
|
||||
|
||||
// CHECK-LABEL: define spir_func <3 x float> @test_refract_float3(
|
||||
// CHECK-SAME: <3 x float> noundef [[I:%.*]], <3 x float> noundef [[N:%.*]], float noundef [[ETA:%.*]]) local_unnamed_addr #[[ATTR0]] {
|
||||
// CHECK-NEXT: [[ENTRY:.*:]]
|
||||
// CHECK-NEXT: [[SPV_REFRACT:%.*]] = tail call <3 x float> @llvm.spv.refract.v3f32.f32(<3 x float> [[I]], <3 x float> [[N]], float [[ETA]])
|
||||
// CHECK-NEXT: ret <3 x float> [[SPV_REFRACT]]
|
||||
//
|
||||
float3 test_refract_float3(float3 I, float3 N, float eta) { return __builtin_spirv_refract(I, N, eta); }
|
||||
|
||||
// CHECK-LABEL: define spir_func <4 x float> @test_refract_float4(
|
||||
// CHECK-SAME: <4 x float> noundef [[I:%.*]], <4 x float> noundef [[N:%.*]], float noundef [[ETA:%.*]]) local_unnamed_addr #[[ATTR0]] {
|
||||
// CHECK-NEXT: [[ENTRY:.*:]]
|
||||
// CHECK-NEXT: [[SPV_REFRACT:%.*]] = tail call <4 x float> @llvm.spv.refract.v4f32.f32(<4 x float> [[I]], <4 x float> [[N]], float [[ETA]])
|
||||
// CHECK-NEXT: ret <4 x float> [[SPV_REFRACT]]
|
||||
//
|
||||
float4 test_refract_float4(float4 I, float4 N, float eta) { return __builtin_spirv_refract(I, N, eta); }
|
||||
66
clang/test/SemaHLSL/BuiltIns/refract-errors.hlsl
Normal file
66
clang/test/SemaHLSL/BuiltIns/refract-errors.hlsl
Normal file
@ -0,0 +1,66 @@
|
||||
// RUN: %clang_cc1 -finclude-default-header -triple dxil-pc-shadermodel6.6-library %s -fnative-half-type -emit-llvm-only -disable-llvm-passes -verify
|
||||
|
||||
float test_no_second_arg(float3 p0) {
|
||||
return refract(p0);
|
||||
// expected-error@-1 {{no matching function for call to 'refract'}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate function template not viable: requires 3 arguments, but 1 was provided}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate function template not viable: requires 3 arguments, but 1 was provided}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate function template not viable: requires 3 arguments, but 1 was provided}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate function template not viable: requires 3 arguments, but 1 was provided}}
|
||||
}
|
||||
|
||||
float test_no_third_arg(float3 p0) {
|
||||
return refract(p0, p0);
|
||||
// expected-error@-1 {{no matching function for call to 'refract'}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate function template not viable: requires 3 arguments, but 2 were provided}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate function template not viable: requires 3 arguments, but 2 were provided}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate function template not viable: requires 3 arguments, but 2 were provided}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate function template not viable: requires 3 arguments, but 2 were provided}}
|
||||
}
|
||||
|
||||
float test_too_many_arg(float2 p0) {
|
||||
return refract(p0, p0, p0, p0);
|
||||
// expected-error@-1 {{no matching function for call to 'refract'}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate function template not viable: requires 3 arguments, but 4 were provided}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate function template not viable: requires 3 arguments, but 4 were provided}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate function template not viable: requires 3 arguments, but 4 were provided}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate function template not viable: requires 3 arguments, but 4 were provided}}
|
||||
}
|
||||
|
||||
float test_double_inputs(double p0, double p1, double p2) {
|
||||
return refract(p0, p1, p2);
|
||||
// expected-error@-1 {{no matching function for call to 'refract'}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate template ignored}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate template ignored}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate template ignored}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate template ignored}}
|
||||
}
|
||||
|
||||
float test_int_inputs(int p0, int p1, int p2) {
|
||||
return refract(p0, p1, p2);
|
||||
// expected-error@-1 {{no matching function for call to 'refract'}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate template ignored}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate template ignored}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate template ignored}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate template ignored}}
|
||||
}
|
||||
|
||||
float1 test_vec1_inputs(float1 p0, float1 p1, float1 p2) {
|
||||
return refract(p0, p1, p2);
|
||||
// expected-error@-1 {{no matching function for call to 'refract'}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate template ignored: substitution failure [with T = float1]: no type named 'Type' in 'hlsl::__detail::enable_if<false, vector<float, 1>>'}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate template ignored: substitution failure [with T = float1]: no type named 'Type' in 'hlsl::__detail::enable_if<false, vector<float, 1>>'}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate template ignored: substitution failure [with L = 1]: no type named 'Type' in 'hlsl::__detail::enable_if<false, half>'}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate template ignored: substitution failure [with L = 1]: no type named 'Type' in 'hlsl::__detail::enable_if<false, float>'}}
|
||||
}
|
||||
|
||||
typedef float float5 __attribute__((ext_vector_type(5)));
|
||||
|
||||
float5 test_vec5_inputs(float5 p0, float5 p1, float p2) {
|
||||
return refract(p0, p1, p2);
|
||||
// expected-error@-1 {{no matching function for call to 'refract'}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate template ignored: deduced conflicting types for parameter 'T' ('float5' (vector of 5 'float' values) vs. 'float')}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate template ignored: deduced conflicting types for parameter 'T' ('float5' (vector of 5 'float' values) vs. 'float')}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate template ignored: substitution failure [with L = 5]: no type named 'Type' in 'hlsl::__detail::enable_if<false, half>'}}
|
||||
// expected-note@hlsl/hlsl_intrinsics.h:* {{candidate template ignored: substitution failure [with L = 5]: no type named 'Type' in 'hlsl::__detail::enable_if<false, float>'}}
|
||||
}
|
||||
41
clang/test/SemaSPIRV/BuiltIns/refract-errors.c
Normal file
41
clang/test/SemaSPIRV/BuiltIns/refract-errors.c
Normal file
@ -0,0 +1,41 @@
|
||||
// RUN: %clang_cc1 %s -triple spirv-pc-vulkan-compute -verify
|
||||
|
||||
typedef float float2 __attribute__((ext_vector_type(2)));
|
||||
typedef float float3 __attribute__((ext_vector_type(3)));
|
||||
typedef _Float16 half;
|
||||
typedef half half2 __attribute__((ext_vector_type(2)));
|
||||
|
||||
float2 test_no_third_arg(float2 p0) {
|
||||
return __builtin_spirv_refract(p0, p0);
|
||||
// expected-error@-1 {{too few arguments to function call, expected 3, have 2}}
|
||||
}
|
||||
|
||||
float2 test_too_many_arg(float2 p0, float p1) {
|
||||
return __builtin_spirv_refract(p0, p0, p1, p1);
|
||||
// expected-error@-1 {{too many arguments to function call, expected 3, have 4}}
|
||||
}
|
||||
|
||||
float test_double_scalar_inputs(double p0, double p1, double p2) {
|
||||
return __builtin_spirv_refract(p0, p1, p2);
|
||||
// expected-error@-1 {{1st argument must be a scalar or vector of 16 or 32 bit floating-point types (was 'double')}}
|
||||
}
|
||||
|
||||
float test_int_scalar_inputs(int p0, int p1, int p2) {
|
||||
return __builtin_spirv_refract(p0, p1, p2);
|
||||
// expected-error@-1 {{1st argument must be a scalar or vector of 16 or 32 bit floating-point types (was 'int')}}
|
||||
}
|
||||
|
||||
float test_float_and_half_inputs(float2 p0, half2 p1, float p2) {
|
||||
return __builtin_spirv_refract(p0, p1, p2);
|
||||
// expected-error@-1 {{first two arguments to '__builtin_spirv_refract' must have the same type}}
|
||||
}
|
||||
|
||||
float test_float_and_half_2_inputs(float2 p0, float2 p1, half p2) {
|
||||
return __builtin_spirv_refract(p0, p1, p2);
|
||||
// expected-error@-1 {{all arguments to '__builtin_spirv_refract' must be of scalar or vector type with matching scalar element type: 'float2' (vector of 2 'float' values) vs 'half' (aka '_Float16')}}
|
||||
}
|
||||
|
||||
float2 test_mismatch_vector_size_inputs(float2 p0, float3 p1, float p2) {
|
||||
return __builtin_spirv_refract(p0, p1, p2);
|
||||
// expected-error@-1 {{first two arguments to '__builtin_spirv_refract' must have the same type}}
|
||||
}
|
||||
@ -90,7 +90,12 @@ let TargetPrefix = "spv" in {
|
||||
def int_spv_length : DefaultAttrsIntrinsic<[LLVMVectorElementType<0>], [llvm_anyfloat_ty], [IntrNoMem]>;
|
||||
def int_spv_normalize : DefaultAttrsIntrinsic<[LLVMMatchType<0>], [llvm_anyfloat_ty], [IntrNoMem]>;
|
||||
def int_spv_reflect : DefaultAttrsIntrinsic<[LLVMMatchType<0>], [llvm_anyfloat_ty, LLVMMatchType<0>], [IntrNoMem]>;
|
||||
def int_spv_rsqrt : DefaultAttrsIntrinsic<[LLVMMatchType<0>], [llvm_anyfloat_ty], [IntrNoMem]>;
|
||||
def int_spv_refract
|
||||
: DefaultAttrsIntrinsic<[LLVMMatchType<0>],
|
||||
[llvm_anyfloat_ty, LLVMMatchType<0>,
|
||||
llvm_anyfloat_ty],
|
||||
[IntrNoMem]>;
|
||||
def int_spv_rsqrt : DefaultAttrsIntrinsic<[LLVMMatchType<0>], [llvm_anyfloat_ty], [IntrNoMem]>;
|
||||
def int_spv_saturate : DefaultAttrsIntrinsic<[llvm_anyfloat_ty], [LLVMMatchType<0>], [IntrNoMem]>;
|
||||
def int_spv_smoothstep : DefaultAttrsIntrinsic<[LLVMMatchType<0>], [llvm_anyfloat_ty, LLVMMatchType<0>, LLVMMatchType<0>], [IntrNoMem]>;
|
||||
def int_spv_step : DefaultAttrsIntrinsic<[LLVMMatchType<0>], [LLVMMatchType<0>, llvm_anyfloat_ty], [IntrNoMem]>;
|
||||
|
||||
@ -3120,6 +3120,8 @@ bool SPIRVInstructionSelector::selectIntrinsic(Register ResVReg,
|
||||
return selectExtInst(ResVReg, ResType, I, CL::fract, GL::Fract);
|
||||
case Intrinsic::spv_normalize:
|
||||
return selectExtInst(ResVReg, ResType, I, CL::normalize, GL::Normalize);
|
||||
case Intrinsic::spv_refract:
|
||||
return selectExtInst(ResVReg, ResType, I, GL::Refract);
|
||||
case Intrinsic::spv_reflect:
|
||||
return selectExtInst(ResVReg, ResType, I, GL::Reflect);
|
||||
case Intrinsic::spv_rsqrt:
|
||||
|
||||
36
llvm/test/CodeGen/SPIRV/hlsl-intrinsics/refract.ll
Normal file
36
llvm/test/CodeGen/SPIRV/hlsl-intrinsics/refract.ll
Normal file
@ -0,0 +1,36 @@
|
||||
; RUN: llc -O0 -mtriple=spirv-unknown-vulkan %s -o - | FileCheck %s
|
||||
; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv-unknown-vulkan %s -o - -filetype=obj | spirv-val %}
|
||||
|
||||
; Make sure SPIRV operation function calls for refract are lowered correctly.
|
||||
|
||||
; CHECK-DAG: %[[#op_ext_glsl:]] = OpExtInstImport "GLSL.std.450"
|
||||
; CHECK-DAG: %[[#float_16:]] = OpTypeFloat 16
|
||||
; CHECK-DAG: %[[#vec4_float_16:]] = OpTypeVector %[[#float_16]] 4
|
||||
; CHECK-DAG: %[[#float_32:]] = OpTypeFloat 32
|
||||
; CHECK-DAG: %[[#vec4_float_32:]] = OpTypeVector %[[#float_32]] 4
|
||||
|
||||
define noundef <4 x half> @refract_half(<4 x half> noundef %I, <4 x half> noundef %N, half noundef %ETA) {
|
||||
entry:
|
||||
; CHECK: %[[#]] = OpFunction %[[#vec4_float_16]] None %[[#]]
|
||||
; CHECK: %[[#arg0:]] = OpFunctionParameter %[[#vec4_float_16]]
|
||||
; CHECK: %[[#arg1:]] = OpFunctionParameter %[[#vec4_float_16]]
|
||||
; CHECK: %[[#arg2:]] = OpFunctionParameter %[[#float_16:]]
|
||||
; CHECK: %[[#]] = OpExtInst %[[#vec4_float_16]] %[[#op_ext_glsl]] Refract %[[#arg0]] %[[#arg1]] %[[#arg2]]
|
||||
%spv.refract.i = tail call reassoc nnan ninf nsz arcp afn noundef <4 x half> @llvm.spv.refract.v4f16.f16(<4 x half> %I, <4 x half> %N, half %ETA)
|
||||
ret <4 x half> %spv.refract.i
|
||||
}
|
||||
|
||||
define noundef <4 x float> @refract_float4(<4 x float> noundef %I, <4 x float> noundef %N, float noundef %ETA) {
|
||||
entry:
|
||||
%conv.i = fpext reassoc nnan ninf nsz arcp afn float %ETA to double
|
||||
; CHECK: %[[#]] = OpFunction %[[#vec4_float_32]] None %[[#]]
|
||||
; CHECK: %[[#arg0:]] = OpFunctionParameter %[[#vec4_float_32]]
|
||||
; CHECK: %[[#arg1:]] = OpFunctionParameter %[[#vec4_float_32]]
|
||||
; CHECK: %[[#arg2:]] = OpFunctionParameter %[[#float_32:]]
|
||||
; CHECK: %[[#]] = OpExtInst %[[#vec4_float_32]] %[[#op_ext_glsl]] Refract %[[#arg0]] %[[#arg1]] %[[#arg2]]
|
||||
%spv.refract.i = tail call reassoc nnan ninf nsz arcp afn noundef <4 x float> @llvm.spv.refract.v4f32.f32(<4 x float> %I, <4 x float> %N, float %ETA)
|
||||
ret <4 x float> %spv.refract.i
|
||||
}
|
||||
|
||||
declare <4 x half> @llvm.spv.refract.v4f16.f16(<4 x half>, <4 x half>, half)
|
||||
declare <4 x float> @llvm.spv.reflect.v4f32.f32(<4 x float>, <4 x float>, float)
|
||||
12
llvm/test/CodeGen/SPIRV/opencl/refract-error.ll
Normal file
12
llvm/test/CodeGen/SPIRV/opencl/refract-error.ll
Normal file
@ -0,0 +1,12 @@
|
||||
; RUN: not llc -verify-machineinstrs -O0 -mtriple=spirv64-unknown-unknown %s -o /dev/null 2>&1 | FileCheck %s
|
||||
; RUN: not llc -verify-machineinstrs -O0 -mtriple=spirv32-unknown-unknown %s -o /dev/null 2>&1 | FileCheck %s
|
||||
|
||||
; CHECK: LLVM ERROR: %{{.*}} = G_INTRINSIC intrinsic(@llvm.spv.refract), %{{.*}}, %{{.*}}, %{{.*}} is only supported with the GLSL extended instruction set.
|
||||
|
||||
define noundef <4 x float> @refract_float4(<4 x float> noundef %I, <4 x float> noundef %N, float noundef %ETA) {
|
||||
entry:
|
||||
%spv.refract = call <4 x float> @llvm.spv.refract.f32(<4 x float> %I, <4 x float> %N, float %ETA)
|
||||
ret <4 x float> %spv.refract
|
||||
}
|
||||
|
||||
declare <4 x float> @llvm.spv.refract.f32(<4 x float>, <4 x float>, float)
|
||||
Loading…
x
Reference in New Issue
Block a user