40c1d50024
Closes #99156. Tasks completed: - Implement `smoothstep` using HLSL source in `hlsl_intrinsics.h` - Implement the `smoothstep` SPIR-V target built-in in `clang/include/clang/Basic/BuiltinsSPIRV.td` - Add sema checks for `smoothstep` to `CheckSPIRVBuiltinFunctionCall` in `clang/lib/Sema/SemaSPIRV.cpp` - Add codegen for spv `smoothstep` to `EmitSPIRVBuiltinExpr` in `clang/lib/CodeGen/TargetBuiltins/SPIR.cpp` - Add codegen tests to `clang/test/CodeGenHLSL/builtins/smoothstep.hlsl` - Add spv codegen test to `clang/test/CodeGenSPIRV/Builtins/smoothstep.c` - Add sema tests to `clang/test/SemaHLSL/BuiltIns/smoothstep-errors.hlsl` - Add spv sema tests to `clang/test/SemaSPIRV/BuiltIns/smoothstep-errors.c` - Create the `int_spv_smoothstep` intrinsic in `IntrinsicsSPIRV.td` - In SPIRVInstructionSelector.cpp create the `smoothstep` lowering and map it to `int_spv_smoothstep` in `SPIRVInstructionSelector::selectIntrinsic` - Create SPIR-V backend test case in `llvm/test/CodeGen/SPIRV/hlsl-intrinsics/smoothstep.ll` - Create SPIR-V backend test case in `llvm/test/CodeGen/SPIRV/opencl/smoothstep.ll`
77 lines
3.3 KiB
C++
77 lines
3.3 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");
|
|
}
|
|
}
|
|
return nullptr;
|
|
}
|