c474f8f240
The patch introduce __builtin_spirv_generic_cast_to_ptr_explicit which is lowered to the llvm.spv.generic.cast.to.ptr.explicit intrinsic. The SPIR-V builtins are now split into 3 differents file: BuiltinsSPIRVCore.td, BuiltinsSPIRVVK.td for Vulkan specific builtins, BuiltinsSPIRVCL.td for OpenCL specific builtins and BuiltinsSPIRVCommon.td for common ones. The patch also introduces a new header defining its SPIR-V friendly equivalent (__spirv_GenericCastToPtrExplicit_ToGlobal, __spirv_GenericCastToPtrExplicit_ToLocal and __spirv_GenericCastToPtrExplicit_ToPrivate). The functions are declared as aliases to the new builtin allowing C-like languages to have a definition to rely on as well as gaining proper front-end diagnostics. The motivation for the header is to provide a stable binding for applications or library (such as SYCL) and allows non SPIR-V targets to provide an implementation (via libclc or similar to how it is done for gpuintrin.h).
103 lines
4.6 KiB
C++
103 lines
4.6 KiB
C++
//===--------- SPIR.cpp - Emit LLVM Code for builtins ---------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This contains code to emit Builtin calls as LLVM code.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "CGHLSLRuntime.h"
|
|
#include "CodeGenFunction.h"
|
|
#include "clang/Basic/TargetBuiltins.h"
|
|
#include "llvm/IR/Intrinsics.h"
|
|
|
|
using namespace clang;
|
|
using namespace CodeGen;
|
|
using namespace llvm;
|
|
|
|
Value *CodeGenFunction::EmitSPIRVBuiltinExpr(unsigned BuiltinID,
|
|
const CallExpr *E) {
|
|
switch (BuiltinID) {
|
|
case SPIRV::BI__builtin_spirv_distance: {
|
|
Value *X = EmitScalarExpr(E->getArg(0));
|
|
Value *Y = EmitScalarExpr(E->getArg(1));
|
|
assert(E->getArg(0)->getType()->hasFloatingRepresentation() &&
|
|
E->getArg(1)->getType()->hasFloatingRepresentation() &&
|
|
"Distance operands must have a float representation");
|
|
assert(E->getArg(0)->getType()->isVectorType() &&
|
|
E->getArg(1)->getType()->isVectorType() &&
|
|
"Distance operands must be a vector");
|
|
return Builder.CreateIntrinsic(
|
|
/*ReturnType=*/X->getType()->getScalarType(), Intrinsic::spv_distance,
|
|
ArrayRef<Value *>{X, Y}, nullptr, "spv.distance");
|
|
}
|
|
case SPIRV::BI__builtin_spirv_length: {
|
|
Value *X = EmitScalarExpr(E->getArg(0));
|
|
assert(E->getArg(0)->getType()->hasFloatingRepresentation() &&
|
|
"length operand must have a float representation");
|
|
assert(E->getArg(0)->getType()->isVectorType() &&
|
|
"length operand must be a vector");
|
|
return Builder.CreateIntrinsic(
|
|
/*ReturnType=*/X->getType()->getScalarType(), Intrinsic::spv_length,
|
|
ArrayRef<Value *>{X}, nullptr, "spv.length");
|
|
}
|
|
case SPIRV::BI__builtin_spirv_reflect: {
|
|
Value *I = EmitScalarExpr(E->getArg(0));
|
|
Value *N = EmitScalarExpr(E->getArg(1));
|
|
assert(E->getArg(0)->getType()->hasFloatingRepresentation() &&
|
|
E->getArg(1)->getType()->hasFloatingRepresentation() &&
|
|
"Reflect operands must have a float representation");
|
|
assert(E->getArg(0)->getType()->isVectorType() &&
|
|
E->getArg(1)->getType()->isVectorType() &&
|
|
"Reflect operands must be a vector");
|
|
return Builder.CreateIntrinsic(
|
|
/*ReturnType=*/I->getType(), Intrinsic::spv_reflect,
|
|
ArrayRef<Value *>{I, N}, nullptr, "spv.reflect");
|
|
}
|
|
case SPIRV::BI__builtin_spirv_smoothstep: {
|
|
Value *Min = EmitScalarExpr(E->getArg(0));
|
|
Value *Max = EmitScalarExpr(E->getArg(1));
|
|
Value *X = EmitScalarExpr(E->getArg(2));
|
|
assert(E->getArg(0)->getType()->hasFloatingRepresentation() &&
|
|
E->getArg(1)->getType()->hasFloatingRepresentation() &&
|
|
E->getArg(2)->getType()->hasFloatingRepresentation() &&
|
|
"SmoothStep operands must have a float representation");
|
|
return Builder.CreateIntrinsic(
|
|
/*ReturnType=*/Min->getType(), Intrinsic::spv_smoothstep,
|
|
ArrayRef<Value *>{Min, Max, X}, /*FMFSource=*/nullptr,
|
|
"spv.smoothstep");
|
|
}
|
|
case SPIRV::BI__builtin_spirv_faceforward: {
|
|
Value *N = EmitScalarExpr(E->getArg(0));
|
|
Value *I = EmitScalarExpr(E->getArg(1));
|
|
Value *Ng = EmitScalarExpr(E->getArg(2));
|
|
assert(E->getArg(0)->getType()->hasFloatingRepresentation() &&
|
|
E->getArg(1)->getType()->hasFloatingRepresentation() &&
|
|
E->getArg(2)->getType()->hasFloatingRepresentation() &&
|
|
"FaceForward operands must have a float representation");
|
|
return Builder.CreateIntrinsic(
|
|
/*ReturnType=*/N->getType(), Intrinsic::spv_faceforward,
|
|
ArrayRef<Value *>{N, I, Ng}, /*FMFSource=*/nullptr, "spv.faceforward");
|
|
}
|
|
case SPIRV::BI__builtin_spirv_generic_cast_to_ptr_explicit: {
|
|
Value *Ptr = EmitScalarExpr(E->getArg(0));
|
|
assert(E->getArg(0)->getType()->hasPointerRepresentation() &&
|
|
E->getArg(1)->getType()->hasIntegerRepresentation() &&
|
|
"GenericCastToPtrExplicit takes a pointer and an int");
|
|
llvm::Type *Res = getTypes().ConvertType(E->getType());
|
|
assert(Res->isPointerTy() &&
|
|
"GenericCastToPtrExplicit doesn't return a pointer");
|
|
llvm::CallInst *Call = Builder.CreateIntrinsic(
|
|
/*ReturnType=*/Res, Intrinsic::spv_generic_cast_to_ptr_explicit,
|
|
ArrayRef<Value *>{Ptr}, nullptr, "spv.generic_cast");
|
|
Call->addRetAttr(llvm::Attribute::AttrKind::NoUndef);
|
|
return Call;
|
|
}
|
|
}
|
|
return nullptr;
|
|
}
|