clang 23.0.0git
CGHLSLBuiltins.cpp
Go to the documentation of this file.
1//===------- CGHLSLBuiltins.cpp - Emit LLVM Code for HLSL builtins --------===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// This contains code to emit HLSL Builtin calls as LLVM code.
10//
11//===----------------------------------------------------------------------===//
12
13#include "CGBuiltin.h"
14#include "CGHLSLRuntime.h"
15#include "CodeGenFunction.h"
16#include "llvm/IR/MatrixBuilder.h"
17
18using namespace clang;
19using namespace CodeGen;
20using namespace llvm;
21
25 "asdouble operands types mismatch");
26 Value *OpLowBits = CGF.EmitScalarExpr(E->getArg(0));
27 Value *OpHighBits = CGF.EmitScalarExpr(E->getArg(1));
28
29 llvm::Type *ResultType = CGF.DoubleTy;
30 int N = 1;
31 if (auto *VTy = E->getArg(0)->getType()->getAs<clang::VectorType>()) {
32 N = VTy->getNumElements();
33 ResultType = llvm::FixedVectorType::get(CGF.DoubleTy, N);
34 }
35
36 if (CGF.CGM.getTarget().getTriple().isDXIL())
37 return CGF.Builder.CreateIntrinsic(
38 /*ReturnType=*/ResultType, Intrinsic::dx_asdouble,
39 {OpLowBits, OpHighBits}, nullptr, "hlsl.asdouble");
40
41 if (!E->getArg(0)->getType()->isVectorType()) {
42 OpLowBits = CGF.Builder.CreateVectorSplat(1, OpLowBits);
43 OpHighBits = CGF.Builder.CreateVectorSplat(1, OpHighBits);
44 }
45
47 for (int i = 0; i < N; i++) {
48 Mask.push_back(i);
49 Mask.push_back(i + N);
50 }
51
52 Value *BitVec = CGF.Builder.CreateShuffleVector(OpLowBits, OpHighBits, Mask);
53
54 return CGF.Builder.CreateBitCast(BitVec, ResultType);
55}
56
58 Value *Op0 = CGF->EmitScalarExpr(E->getArg(0));
59
60 Constant *FZeroConst = ConstantFP::getZero(CGF->FloatTy);
61 Value *CMP;
62 Value *LastInstr;
63
64 if (const auto *VecTy = E->getArg(0)->getType()->getAs<clang::VectorType>()) {
65 FZeroConst = ConstantVector::getSplat(
66 ElementCount::getFixed(VecTy->getNumElements()), FZeroConst);
67 auto *FCompInst = CGF->Builder.CreateFCmpOLT(Op0, FZeroConst);
68 CMP = CGF->Builder.CreateIntrinsic(
69 CGF->Builder.getInt1Ty(), CGF->CGM.getHLSLRuntime().getAnyIntrinsic(),
70 {FCompInst});
71 } else {
72 CMP = CGF->Builder.CreateFCmpOLT(Op0, FZeroConst);
73 }
74
75 if (CGF->CGM.getTarget().getTriple().isDXIL()) {
76 LastInstr = CGF->Builder.CreateIntrinsic(Intrinsic::dx_discard, {CMP});
77 } else if (CGF->CGM.getTarget().getTriple().isSPIRV()) {
78 BasicBlock *LT0 = CGF->createBasicBlock("lt0", CGF->CurFn);
79 BasicBlock *End = CGF->createBasicBlock("end", CGF->CurFn);
80
81 CGF->Builder.CreateCondBr(CMP, LT0, End);
82
83 CGF->Builder.SetInsertPoint(LT0);
84
85 CGF->Builder.CreateIntrinsic(Intrinsic::spv_discard, {});
86
87 LastInstr = CGF->Builder.CreateBr(End);
88 CGF->Builder.SetInsertPoint(End);
89 } else {
90 llvm_unreachable("Backend Codegen not supported.");
91 }
92
93 return LastInstr;
94}
95
97 Value *Op0 = CGF->EmitScalarExpr(E->getArg(0));
98 const auto *OutArg1 = dyn_cast<HLSLOutArgExpr>(E->getArg(1));
99 const auto *OutArg2 = dyn_cast<HLSLOutArgExpr>(E->getArg(2));
100
101 CallArgList Args;
102 LValue Op1TmpLValue =
103 CGF->EmitHLSLOutArgExpr(OutArg1, Args, OutArg1->getType());
104 LValue Op2TmpLValue =
105 CGF->EmitHLSLOutArgExpr(OutArg2, Args, OutArg2->getType());
106
108 Args.reverseWritebacks();
109
110 Value *LowBits = nullptr;
111 Value *HighBits = nullptr;
112
113 if (CGF->CGM.getTarget().getTriple().isDXIL()) {
114 llvm::Type *RetElementTy = CGF->Int32Ty;
115 if (auto *Op0VecTy = E->getArg(0)->getType()->getAs<clang::VectorType>())
116 RetElementTy = llvm::VectorType::get(
117 CGF->Int32Ty, ElementCount::getFixed(Op0VecTy->getNumElements()));
118 auto *RetTy = llvm::StructType::get(RetElementTy, RetElementTy);
119
120 CallInst *CI = CGF->Builder.CreateIntrinsic(
121 RetTy, Intrinsic::dx_splitdouble, {Op0}, nullptr, "hlsl.splitdouble");
122
123 LowBits = CGF->Builder.CreateExtractValue(CI, 0);
124 HighBits = CGF->Builder.CreateExtractValue(CI, 1);
125 } else {
126 // For Non DXIL targets we generate the instructions.
127
128 if (!Op0->getType()->isVectorTy()) {
129 FixedVectorType *DestTy = FixedVectorType::get(CGF->Int32Ty, 2);
130 Value *Bitcast = CGF->Builder.CreateBitCast(Op0, DestTy);
131
132 LowBits = CGF->Builder.CreateExtractElement(Bitcast, (uint64_t)0);
133 HighBits = CGF->Builder.CreateExtractElement(Bitcast, 1);
134 } else {
135 int NumElements = 1;
136 if (const auto *VecTy =
138 NumElements = VecTy->getNumElements();
139
140 FixedVectorType *Uint32VecTy =
141 FixedVectorType::get(CGF->Int32Ty, NumElements * 2);
142 Value *Uint32Vec = CGF->Builder.CreateBitCast(Op0, Uint32VecTy);
143 if (NumElements == 1) {
144 LowBits = CGF->Builder.CreateExtractElement(Uint32Vec, (uint64_t)0);
145 HighBits = CGF->Builder.CreateExtractElement(Uint32Vec, 1);
146 } else {
147 SmallVector<int> EvenMask, OddMask;
148 for (int I = 0, E = NumElements; I != E; ++I) {
149 EvenMask.push_back(I * 2);
150 OddMask.push_back(I * 2 + 1);
151 }
152 LowBits = CGF->Builder.CreateShuffleVector(Uint32Vec, EvenMask);
153 HighBits = CGF->Builder.CreateShuffleVector(Uint32Vec, OddMask);
154 }
155 }
156 }
157 CGF->Builder.CreateStore(LowBits, Op1TmpLValue.getAddress());
158 auto *LastInst =
159 CGF->Builder.CreateStore(HighBits, Op2TmpLValue.getAddress());
160 CGF->EmitWritebacks(Args);
161 return LastInst;
162}
163
165 const CallExpr *E) {
166 Value *Cond = CGF.EmitScalarExpr(E->getArg(0));
167 llvm::Type *I32 = CGF.Int32Ty;
168
169 llvm::Type *Vec4I32 = llvm::FixedVectorType::get(I32, 4);
170 [[maybe_unused]] llvm::StructType *Struct4I32 =
171 llvm::StructType::get(CGF.getLLVMContext(), {I32, I32, I32, I32});
172
173 if (CGF.CGM.getTarget().getTriple().isDXIL()) {
174 // Call DXIL intrinsic: returns { i32, i32, i32, i32 }
175 llvm::Function *Fn = CGF.CGM.getIntrinsic(Intrinsic::dx_wave_ballot, {I32});
176
177 Value *StructVal = CGF.EmitRuntimeCall(Fn, Cond);
178 assert(StructVal->getType() == Struct4I32 &&
179 "dx.wave.ballot must return {i32,i32,i32,i32}");
180
181 // Reassemble struct to <4 x i32>
182 llvm::Value *VecVal = llvm::PoisonValue::get(Vec4I32);
183 for (unsigned I = 0; I < 4; ++I) {
184 Value *Elt = CGF.Builder.CreateExtractValue(StructVal, I);
185 VecVal =
186 CGF.Builder.CreateInsertElement(VecVal, Elt, CGF.Builder.getInt32(I));
187 }
188
189 return VecVal;
190 }
191
192 if (CGF.CGM.getTarget().getTriple().isSPIRV())
193 return CGF.EmitRuntimeCall(
194 CGF.CGM.getIntrinsic(Intrinsic::spv_subgroup_ballot), Cond);
195
196 llvm_unreachable(
197 "WaveActiveBallot is only supported for DXIL and SPIRV targets");
198}
199
201 const CallExpr *E) {
202 Value *Op0 = CGF.EmitScalarExpr(E->getArg(0));
203 QualType Op0Ty = E->getArg(0)->getType();
204 llvm::Type *ResType = CGF.FloatTy;
205 uint64_t NumElements = 0;
206 if (Op0->getType()->isVectorTy()) {
207 NumElements =
208 E->getArg(0)->getType()->castAs<clang::VectorType>()->getNumElements();
209 ResType =
210 llvm::VectorType::get(ResType, ElementCount::getFixed(NumElements));
211 }
213 llvm_unreachable(
214 "f16tof32 operand must have an unsigned int representation");
215
216 if (CGF.CGM.getTriple().isDXIL())
217 return CGF.Builder.CreateIntrinsic(ResType, Intrinsic::dx_legacyf16tof32,
218 ArrayRef<Value *>{Op0}, nullptr,
219 "hlsl.f16tof32");
220
221 if (CGF.CGM.getTriple().isSPIRV()) {
222 // We use the SPIRV UnpackHalf2x16 operation to avoid the need for the
223 // Int16 and Float16 capabilities
224 auto *UnpackType =
225 llvm::VectorType::get(CGF.FloatTy, ElementCount::getFixed(2));
226
227 if (NumElements == 0) {
228 // a scalar input - simply extract the first element of the unpacked
229 // vector
230 Value *Unpack = CGF.Builder.CreateIntrinsic(
231 UnpackType, Intrinsic::spv_unpackhalf2x16, ArrayRef<Value *>{Op0});
232 return CGF.Builder.CreateExtractElement(Unpack, (uint64_t)0);
233 }
234
235 // a vector input - build a congruent output vector by iterating through
236 // the input vector calling unpackhalf2x16 for each element
237 Value *Result = PoisonValue::get(ResType);
238 for (uint64_t I = 0; I < NumElements; I++) {
239 Value *InVal = CGF.Builder.CreateExtractElement(Op0, I);
240 Value *Unpack = CGF.Builder.CreateIntrinsic(
241 UnpackType, Intrinsic::spv_unpackhalf2x16, ArrayRef<Value *>{InVal});
242 Value *Res = CGF.Builder.CreateExtractElement(Unpack, (uint64_t)0);
243 Result = CGF.Builder.CreateInsertElement(Result, Res, I);
244 }
245 return Result;
246 }
247
248 llvm_unreachable("Intrinsic F16ToF32 not supported by target architecture");
249}
250
252 const CallExpr *E) {
253 Value *Op0 = CGF.EmitScalarExpr(E->getArg(0));
254 QualType Op0Ty = E->getArg(0)->getType();
255 llvm::Type *ResType = CGF.IntTy;
256 uint64_t NumElements = 0;
257 if (Op0->getType()->isVectorTy()) {
258 NumElements =
259 E->getArg(0)->getType()->castAs<clang::VectorType>()->getNumElements();
260 ResType =
261 llvm::VectorType::get(ResType, ElementCount::getFixed(NumElements));
262 }
263 if (!Op0Ty->hasFloatingRepresentation())
264 llvm_unreachable("f32tof16 operand must have a float representation");
265
266 if (CGF.CGM.getTriple().isDXIL())
267 return CGF.Builder.CreateIntrinsic(ResType, Intrinsic::dx_legacyf32tof16,
268 ArrayRef<Value *>{Op0}, nullptr,
269 "hlsl.f32tof16");
270
271 if (CGF.CGM.getTriple().isSPIRV()) {
272 // We use the SPIRV PackHalf2x16 operation to avoid the need for the
273 // Int16 and Float16 capabilities
274 auto *PackType =
275 llvm::VectorType::get(CGF.FloatTy, ElementCount::getFixed(2));
276
277 if (NumElements == 0) {
278 // a scalar input - simply insert the scalar in the first element
279 // of the 2 element float vector
280 Value *Float2 = Constant::getNullValue(PackType);
281 Float2 = CGF.Builder.CreateInsertElement(Float2, Op0, (uint64_t)0);
282 Value *Result = CGF.Builder.CreateIntrinsic(
283 ResType, Intrinsic::spv_packhalf2x16, ArrayRef<Value *>{Float2});
284 return Result;
285 }
286
287 // a vector input - build a congruent output vector by iterating through
288 // the input vector calling packhalf2x16 for each element
289 Value *Result = PoisonValue::get(ResType);
290 for (uint64_t I = 0; I < NumElements; I++) {
291 Value *Float2 = Constant::getNullValue(PackType);
292 Value *InVal = CGF.Builder.CreateExtractElement(Op0, I);
293 Float2 = CGF.Builder.CreateInsertElement(Float2, InVal, (uint64_t)0);
294 Value *Res = CGF.Builder.CreateIntrinsic(
295 CGF.IntTy, Intrinsic::spv_packhalf2x16, ArrayRef<Value *>{Float2});
296 Result = CGF.Builder.CreateInsertElement(Result, Res, I);
297 }
298 return Result;
299 }
300
301 llvm_unreachable("Intrinsic F32ToF16 not supported by target architecture");
302}
303
304static Value *emitBufferStride(CodeGenFunction *CGF, const Expr *HandleExpr,
305 LValue &Stride) {
306 // Figure out the stride of the buffer elements from the handle type.
307 auto *HandleTy =
309 QualType ElementTy = HandleTy->getContainedType();
310 Value *StrideValue = CGF->getTypeSize(ElementTy);
311 return CGF->Builder.CreateStore(StrideValue, Stride.getAddress());
312}
313
314// Return dot product intrinsic that corresponds to the QT scalar type
315static Intrinsic::ID getDotProductIntrinsic(CGHLSLRuntime &RT, QualType QT) {
316 if (QT->isFloatingType())
317 return RT.getFDotIntrinsic();
318 if (QT->isSignedIntegerType())
319 return RT.getSDotIntrinsic();
320 assert(QT->isUnsignedIntegerType());
321 return RT.getUDotIntrinsic();
322}
323
324static Intrinsic::ID getFirstBitHighIntrinsic(CGHLSLRuntime &RT, QualType QT) {
326 return RT.getFirstBitSHighIntrinsic();
327 }
328
330 return RT.getFirstBitUHighIntrinsic();
331}
332
333// Return wave active sum that corresponds to the QT scalar type
334static Intrinsic::ID getWaveActiveSumIntrinsic(llvm::Triple::ArchType Arch,
335 QualType QT) {
336 switch (Arch) {
337 case llvm::Triple::spirv:
338 return Intrinsic::spv_wave_reduce_sum;
339 case llvm::Triple::dxil: {
340 if (QT->isUnsignedIntegerType())
341 return Intrinsic::dx_wave_reduce_usum;
342 return Intrinsic::dx_wave_reduce_sum;
343 }
344 default:
345 llvm_unreachable("Intrinsic WaveActiveSum"
346 " not supported by target architecture");
347 }
348}
349
350// Return wave active product that corresponds to the QT scalar type
351static Intrinsic::ID getWaveActiveProductIntrinsic(llvm::Triple::ArchType Arch,
352 QualType QT) {
353 switch (Arch) {
354 case llvm::Triple::spirv:
355 return Intrinsic::spv_wave_product;
356 case llvm::Triple::dxil: {
357 if (QT->isUnsignedIntegerType())
358 return Intrinsic::dx_wave_uproduct;
359 return Intrinsic::dx_wave_product;
360 }
361 default:
362 llvm_unreachable("Intrinsic WaveActiveProduct"
363 " not supported by target architecture");
364 }
365}
366
367static Intrinsic::ID getPrefixCountBitsIntrinsic(llvm::Triple::ArchType Arch) {
368 switch (Arch) {
369 case llvm::Triple::spirv:
370 return Intrinsic::spv_subgroup_prefix_bit_count;
371 case llvm::Triple::dxil: {
372 return Intrinsic::dx_wave_prefix_bit_count;
373 }
374 default:
375 llvm_unreachable(
376 "WavePrefixOp instruction not supported by target architecture");
377 }
378}
379
380// Return wave prefix sum that corresponds to the QT scalar type
381static Intrinsic::ID getWavePrefixSumIntrinsic(llvm::Triple::ArchType Arch,
382 QualType QT) {
383 switch (Arch) {
384 case llvm::Triple::spirv:
385 return Intrinsic::spv_wave_prefix_sum;
386 case llvm::Triple::dxil: {
387 if (QT->isUnsignedIntegerType())
388 return Intrinsic::dx_wave_prefix_usum;
389 return Intrinsic::dx_wave_prefix_sum;
390 }
391 default:
392 llvm_unreachable("Intrinsic WavePrefixSum"
393 " not supported by target architecture");
394 }
395}
396
397// Return wave prefix product that corresponds to the QT scalar type
398static Intrinsic::ID getWavePrefixProductIntrinsic(llvm::Triple::ArchType Arch,
399 QualType QT) {
400 switch (Arch) {
401 case llvm::Triple::spirv:
402 return Intrinsic::spv_wave_prefix_product;
403 case llvm::Triple::dxil: {
404 if (QT->isUnsignedIntegerType())
405 return Intrinsic::dx_wave_prefix_uproduct;
406 return Intrinsic::dx_wave_prefix_product;
407 }
408 default:
409 llvm_unreachable("Intrinsic WavePrefixProduct"
410 " not supported by target architecture");
411 }
412}
413
414// Returns the mangled name for a builtin function that the SPIR-V backend
415// will expand into a spec Constant.
416static std::string getSpecConstantFunctionName(clang::QualType SpecConstantType,
417 ASTContext &Context) {
418 // The parameter types for our conceptual intrinsic function.
419 QualType ClangParamTypes[] = {Context.IntTy, SpecConstantType};
420
421 // Create a temporary FunctionDecl for the builtin fuction. It won't be
422 // added to the AST.
424 QualType FnType =
425 Context.getFunctionType(SpecConstantType, ClangParamTypes, EPI);
426 DeclarationName FuncName = &Context.Idents.get("__spirv_SpecConstant");
427 FunctionDecl *FnDeclForMangling = FunctionDecl::Create(
428 Context, Context.getTranslationUnitDecl(), SourceLocation(),
429 SourceLocation(), FuncName, FnType, /*TSI=*/nullptr, SC_Extern);
430
431 // Attach the created parameter declarations to the function declaration.
433 for (QualType ParamType : ClangParamTypes) {
435 Context, FnDeclForMangling, SourceLocation(), SourceLocation(),
436 /*IdentifierInfo*/ nullptr, ParamType, /*TSI*/ nullptr, SC_None,
437 /*DefaultArg*/ nullptr);
438 ParamDecls.push_back(PD);
439 }
440 FnDeclForMangling->setParams(ParamDecls);
441
442 // Get the mangled name.
443 std::string Name;
444 llvm::raw_string_ostream MangledNameStream(Name);
445 std::unique_ptr<MangleContext> Mangler(Context.createMangleContext());
446 Mangler->mangleName(FnDeclForMangling, MangledNameStream);
447 MangledNameStream.flush();
448
449 return Name;
450}
451
452static llvm::Type *getOffsetType(CodeGenModule &CGM, llvm::Type *CoordTy) {
453 llvm::Type *Int32Ty = CGM.Int32Ty;
454 if (auto *VT = dyn_cast<llvm::FixedVectorType>(CoordTy))
455 return llvm::FixedVectorType::get(Int32Ty, VT->getNumElements());
456 return Int32Ty;
457}
458
460 unsigned OffsetArgIndex, llvm::Type *OffsetTy) {
461 if (E->getNumArgs() > OffsetArgIndex)
462 return CGF.EmitScalarExpr(E->getArg(OffsetArgIndex));
463
464 return llvm::Constant::getNullValue(OffsetTy);
465}
466
468 unsigned ClampArgIndex) {
469 Value *Clamp = CGF.EmitScalarExpr(E->getArg(ClampArgIndex));
470 // The builtin is defined with variadic arguments, so the clamp parameter
471 // might have been promoted to double. The intrinsic requires a 32-bit
472 // float.
473 if (Clamp->getType() != CGF.Builder.getFloatTy())
474 Clamp = CGF.Builder.CreateFPCast(Clamp, CGF.Builder.getFloatTy());
475 return Clamp;
476}
477
479 const CallExpr *E,
481 if (!getLangOpts().HLSL)
482 return nullptr;
483
484 switch (BuiltinID) {
485 case Builtin::BI__builtin_hlsl_adduint64: {
486 Value *OpA = EmitScalarExpr(E->getArg(0));
487 Value *OpB = EmitScalarExpr(E->getArg(1));
488 QualType Arg0Ty = E->getArg(0)->getType();
489 uint64_t NumElements = Arg0Ty->castAs<VectorType>()->getNumElements();
490 assert(Arg0Ty == E->getArg(1)->getType() &&
491 "AddUint64 operand types must match");
492 assert(Arg0Ty->hasIntegerRepresentation() &&
493 "AddUint64 operands must have an integer representation");
494 assert((NumElements == 2 || NumElements == 4) &&
495 "AddUint64 operands must have 2 or 4 elements");
496
497 llvm::Value *LowA;
498 llvm::Value *HighA;
499 llvm::Value *LowB;
500 llvm::Value *HighB;
501
502 // Obtain low and high words of inputs A and B
503 if (NumElements == 2) {
504 LowA = Builder.CreateExtractElement(OpA, (uint64_t)0, "LowA");
505 HighA = Builder.CreateExtractElement(OpA, (uint64_t)1, "HighA");
506 LowB = Builder.CreateExtractElement(OpB, (uint64_t)0, "LowB");
507 HighB = Builder.CreateExtractElement(OpB, (uint64_t)1, "HighB");
508 } else {
509 LowA = Builder.CreateShuffleVector(OpA, {0, 2}, "LowA");
510 HighA = Builder.CreateShuffleVector(OpA, {1, 3}, "HighA");
511 LowB = Builder.CreateShuffleVector(OpB, {0, 2}, "LowB");
512 HighB = Builder.CreateShuffleVector(OpB, {1, 3}, "HighB");
513 }
514
515 // Use an uadd_with_overflow to compute the sum of low words and obtain a
516 // carry value
517 llvm::Value *Carry;
518 llvm::Value *LowSum = EmitOverflowIntrinsic(
519 *this, Intrinsic::uadd_with_overflow, LowA, LowB, Carry);
520 llvm::Value *ZExtCarry =
521 Builder.CreateZExt(Carry, HighA->getType(), "CarryZExt");
522
523 // Sum the high words and the carry
524 llvm::Value *HighSum = Builder.CreateAdd(HighA, HighB, "HighSum");
525 llvm::Value *HighSumPlusCarry =
526 Builder.CreateAdd(HighSum, ZExtCarry, "HighSumPlusCarry");
527
528 if (NumElements == 4) {
529 return Builder.CreateShuffleVector(LowSum, HighSumPlusCarry, {0, 2, 1, 3},
530 "hlsl.AddUint64");
531 }
532
533 llvm::Value *Result = PoisonValue::get(OpA->getType());
534 Result = Builder.CreateInsertElement(Result, LowSum, (uint64_t)0,
535 "hlsl.AddUint64.upto0");
536 Result = Builder.CreateInsertElement(Result, HighSumPlusCarry, (uint64_t)1,
537 "hlsl.AddUint64");
538 return Result;
539 }
540 case Builtin::BI__builtin_hlsl_resource_getpointer:
541 case Builtin::BI__builtin_hlsl_resource_getpointer_typed: {
542 Value *HandleOp = EmitScalarExpr(E->getArg(0));
543 Value *IndexOp = EmitScalarExpr(E->getArg(1));
544
545 llvm::Type *RetTy = ConvertType(E->getType());
546 return Builder.CreateIntrinsic(
547 RetTy, CGM.getHLSLRuntime().getCreateResourceGetPointerIntrinsic(),
548 ArrayRef<Value *>{HandleOp, IndexOp});
549 }
550 case Builtin::BI__builtin_hlsl_resource_sample: {
551 Value *HandleOp = EmitScalarExpr(E->getArg(0));
552 Value *SamplerOp = EmitScalarExpr(E->getArg(1));
553 Value *CoordOp = EmitScalarExpr(E->getArg(2));
554
556 Args.push_back(HandleOp);
557 Args.push_back(SamplerOp);
558 Args.push_back(CoordOp);
559 Args.push_back(
560 emitHlslOffset(*this, E, 3, getOffsetType(CGM, CoordOp->getType())));
561
562 llvm::Type *RetTy = ConvertType(E->getType());
563 if (E->getNumArgs() <= 4) {
564 return Builder.CreateIntrinsic(
565 RetTy, CGM.getHLSLRuntime().getSampleIntrinsic(), Args);
566 }
567
568 Args.push_back(emitHlslClamp(*this, E, 4));
569 return Builder.CreateIntrinsic(
570 RetTy, CGM.getHLSLRuntime().getSampleClampIntrinsic(), Args);
571 }
572 case Builtin::BI__builtin_hlsl_resource_sample_bias: {
573 Value *HandleOp = EmitScalarExpr(E->getArg(0));
574 Value *SamplerOp = EmitScalarExpr(E->getArg(1));
575 Value *CoordOp = EmitScalarExpr(E->getArg(2));
576 Value *BiasOp = EmitScalarExpr(E->getArg(3));
577 if (BiasOp->getType() != Builder.getFloatTy())
578 BiasOp = Builder.CreateFPCast(BiasOp, Builder.getFloatTy());
579
580 SmallVector<Value *, 6> Args; // Max 6 arguments for SampleBias
581 Args.push_back(HandleOp);
582 Args.push_back(SamplerOp);
583 Args.push_back(CoordOp);
584 Args.push_back(BiasOp);
585 Args.push_back(
586 emitHlslOffset(*this, E, 4, getOffsetType(CGM, CoordOp->getType())));
587
588 llvm::Type *RetTy = ConvertType(E->getType());
589 if (E->getNumArgs() <= 5)
590 return Builder.CreateIntrinsic(
591 RetTy, CGM.getHLSLRuntime().getSampleBiasIntrinsic(), Args);
592
593 Args.push_back(emitHlslClamp(*this, E, 5));
594 return Builder.CreateIntrinsic(
595 RetTy, CGM.getHLSLRuntime().getSampleBiasClampIntrinsic(), Args);
596 }
597 case Builtin::BI__builtin_hlsl_resource_sample_grad: {
598 Value *HandleOp = EmitScalarExpr(E->getArg(0));
599 Value *SamplerOp = EmitScalarExpr(E->getArg(1));
600 Value *CoordOp = EmitScalarExpr(E->getArg(2));
601 Value *DDXOp = EmitScalarExpr(E->getArg(3));
602 Value *DDYOp = EmitScalarExpr(E->getArg(4));
603
605 Args.push_back(HandleOp);
606 Args.push_back(SamplerOp);
607 Args.push_back(CoordOp);
608 Args.push_back(DDXOp);
609 Args.push_back(DDYOp);
610 Args.push_back(
611 emitHlslOffset(*this, E, 5, getOffsetType(CGM, CoordOp->getType())));
612
613 llvm::Type *RetTy = ConvertType(E->getType());
614
615 if (E->getNumArgs() <= 6) {
616 return Builder.CreateIntrinsic(
617 RetTy, CGM.getHLSLRuntime().getSampleGradIntrinsic(), Args);
618 }
619
620 Args.push_back(emitHlslClamp(*this, E, 6));
621 return Builder.CreateIntrinsic(
622 RetTy, CGM.getHLSLRuntime().getSampleGradClampIntrinsic(), Args);
623 }
624 case Builtin::BI__builtin_hlsl_resource_sample_level: {
625 Value *HandleOp = EmitScalarExpr(E->getArg(0));
626 Value *SamplerOp = EmitScalarExpr(E->getArg(1));
627 Value *CoordOp = EmitScalarExpr(E->getArg(2));
628 Value *LODOp = EmitScalarExpr(E->getArg(3));
629 if (LODOp->getType() != Builder.getFloatTy())
630 LODOp = Builder.CreateFPCast(LODOp, Builder.getFloatTy());
631
632 SmallVector<Value *, 5> Args; // Max 5 arguments for SampleLevel
633 Args.push_back(HandleOp);
634 Args.push_back(SamplerOp);
635 Args.push_back(CoordOp);
636 Args.push_back(LODOp);
637 Args.push_back(
638 emitHlslOffset(*this, E, 4, getOffsetType(CGM, CoordOp->getType())));
639
640 llvm::Type *RetTy = ConvertType(E->getType());
641 return Builder.CreateIntrinsic(
642 RetTy, CGM.getHLSLRuntime().getSampleLevelIntrinsic(), Args);
643 }
644 case Builtin::BI__builtin_hlsl_resource_load_level: {
645 Value *HandleOp = EmitScalarExpr(E->getArg(0));
646 Value *CoordLODOp = EmitScalarExpr(E->getArg(1));
647
648 auto *CoordLODVecTy = cast<llvm::FixedVectorType>(CoordLODOp->getType());
649 unsigned NumElts = CoordLODVecTy->getNumElements();
650 assert(NumElts >= 2 && "CoordLOD must have at least 2 elements");
651
652 // Split CoordLOD into Coord and LOD
654 for (unsigned I = 0; I < NumElts - 1; ++I)
655 Mask.push_back(I);
656
657 Value *CoordOp =
658 Builder.CreateShuffleVector(CoordLODOp, Mask, "hlsl.load.coord");
659 Value *LODOp =
660 Builder.CreateExtractElement(CoordLODOp, NumElts - 1, "hlsl.load.lod");
661
663 Args.push_back(HandleOp);
664 Args.push_back(CoordOp);
665 Args.push_back(LODOp);
666 Args.push_back(
667 emitHlslOffset(*this, E, 2, getOffsetType(CGM, CoordOp->getType())));
668
669 llvm::Type *RetTy = ConvertType(E->getType());
670 return Builder.CreateIntrinsic(
671 RetTy, CGM.getHLSLRuntime().getLoadLevelIntrinsic(), Args);
672 }
673 case Builtin::BI__builtin_hlsl_resource_sample_cmp: {
674 Value *HandleOp = EmitScalarExpr(E->getArg(0));
675 Value *SamplerOp = EmitScalarExpr(E->getArg(1));
676 Value *CoordOp = EmitScalarExpr(E->getArg(2));
677 Value *CmpOp = EmitScalarExpr(E->getArg(3));
678 if (CmpOp->getType() != Builder.getFloatTy())
679 CmpOp = Builder.CreateFPCast(CmpOp, Builder.getFloatTy());
680
681 SmallVector<Value *, 6> Args; // Max 6 arguments for SampleCmp
682 Args.push_back(HandleOp);
683 Args.push_back(SamplerOp);
684 Args.push_back(CoordOp);
685 Args.push_back(CmpOp);
686 Args.push_back(
687 emitHlslOffset(*this, E, 4, getOffsetType(CGM, CoordOp->getType())));
688
689 llvm::Type *RetTy = ConvertType(E->getType());
690 if (E->getNumArgs() <= 5) {
691 return Builder.CreateIntrinsic(
692 RetTy, CGM.getHLSLRuntime().getSampleCmpIntrinsic(), Args);
693 }
694
695 Args.push_back(emitHlslClamp(*this, E, 5));
696 return Builder.CreateIntrinsic(
697 RetTy, CGM.getHLSLRuntime().getSampleCmpClampIntrinsic(), Args);
698 }
699 case Builtin::BI__builtin_hlsl_resource_sample_cmp_level_zero: {
700 Value *HandleOp = EmitScalarExpr(E->getArg(0));
701 Value *SamplerOp = EmitScalarExpr(E->getArg(1));
702 Value *CoordOp = EmitScalarExpr(E->getArg(2));
703 Value *CmpOp = EmitScalarExpr(E->getArg(3));
704 if (CmpOp->getType() != Builder.getFloatTy())
705 CmpOp = Builder.CreateFPCast(CmpOp, Builder.getFloatTy());
706
708 Args.push_back(HandleOp);
709 Args.push_back(SamplerOp);
710 Args.push_back(CoordOp);
711 Args.push_back(CmpOp);
712
713 Args.push_back(
714 emitHlslOffset(*this, E, 4, getOffsetType(CGM, CoordOp->getType())));
715
716 llvm::Type *RetTy = ConvertType(E->getType());
717 return Builder.CreateIntrinsic(
718 RetTy, CGM.getHLSLRuntime().getSampleCmpLevelZeroIntrinsic(), Args);
719 }
720 case Builtin::BI__builtin_hlsl_resource_gather: {
721 Value *HandleOp = EmitScalarExpr(E->getArg(0));
722 Value *SamplerOp = EmitScalarExpr(E->getArg(1));
723 Value *CoordOp = EmitScalarExpr(E->getArg(2));
724 Value *ComponentOp = EmitScalarExpr(E->getArg(3));
725 if (ComponentOp->getType() != Builder.getInt32Ty())
726 ComponentOp = Builder.CreateIntCast(ComponentOp, Builder.getInt32Ty(),
727 /*isSigned=*/false);
728
730 Args.push_back(HandleOp);
731 Args.push_back(SamplerOp);
732 Args.push_back(CoordOp);
733 Args.push_back(ComponentOp);
734 Args.push_back(
735 emitHlslOffset(*this, E, 4, getOffsetType(CGM, CoordOp->getType())));
736
737 llvm::Type *RetTy = ConvertType(E->getType());
738 return Builder.CreateIntrinsic(
739 RetTy, CGM.getHLSLRuntime().getGatherIntrinsic(), Args);
740 }
741 case Builtin::BI__builtin_hlsl_resource_gather_cmp: {
742 Value *HandleOp = EmitScalarExpr(E->getArg(0));
743 Value *SamplerOp = EmitScalarExpr(E->getArg(1));
744 Value *CoordOp = EmitScalarExpr(E->getArg(2));
745 Value *CompareOp = EmitScalarExpr(E->getArg(3));
746 if (CompareOp->getType() != Builder.getFloatTy())
747 CompareOp = Builder.CreateFPCast(CompareOp, Builder.getFloatTy());
748
750 Args.push_back(HandleOp);
751 Args.push_back(SamplerOp);
752 Args.push_back(CoordOp);
753 Args.push_back(CompareOp);
754
755 if (CGM.getTarget().getTriple().isDXIL()) {
756 Value *ComponentOp = EmitScalarExpr(E->getArg(4));
757 if (ComponentOp->getType() != Builder.getInt32Ty())
758 ComponentOp = Builder.CreateIntCast(ComponentOp, Builder.getInt32Ty(),
759 /*isSigned=*/false);
760 Args.push_back(ComponentOp);
761 }
762
763 Args.push_back(
764 emitHlslOffset(*this, E, 5, getOffsetType(CGM, CoordOp->getType())));
765
766 llvm::Type *RetTy = ConvertType(E->getType());
767 return Builder.CreateIntrinsic(
768 RetTy, CGM.getHLSLRuntime().getGatherCmpIntrinsic(), Args);
769 }
770 case Builtin::BI__builtin_hlsl_resource_load_with_status:
771 case Builtin::BI__builtin_hlsl_resource_load_with_status_typed: {
772 Value *HandleOp = EmitScalarExpr(E->getArg(0));
773 Value *IndexOp = EmitScalarExpr(E->getArg(1));
774
775 // Get the *address* of the status argument to write to it by reference
776 LValue StatusLVal = EmitLValue(E->getArg(2));
777 Address StatusAddr = StatusLVal.getAddress();
778
779 QualType HandleTy = E->getArg(0)->getType();
780 const HLSLAttributedResourceType *RT =
781 HandleTy->getAs<HLSLAttributedResourceType>();
782 assert(CGM.getTarget().getTriple().getArch() == llvm::Triple::dxil &&
783 "Only DXIL currently implements load with status");
784
785 Intrinsic::ID IntrID = RT->getAttrs().RawBuffer
786 ? llvm::Intrinsic::dx_resource_load_rawbuffer
787 : llvm::Intrinsic::dx_resource_load_typedbuffer;
788
789 llvm::Type *DataTy = ConvertType(E->getType());
790 llvm::Type *RetTy = llvm::StructType::get(Builder.getContext(),
791 {DataTy, Builder.getInt1Ty()});
792
794 Args.push_back(HandleOp);
795 Args.push_back(IndexOp);
796
797 if (RT->isRaw()) {
798 Value *Offset = Builder.getInt32(0);
799 // The offset parameter needs to be poison for ByteAddressBuffer
800 if (!RT->isStructured())
801 Offset = llvm::PoisonValue::get(Builder.getInt32Ty());
802 Args.push_back(Offset);
803 }
804
805 // The load intrinsics give us a (T value, i1 status) pair -
806 // shepherd these into the return value and out reference respectively.
807 Value *ResRet =
808 Builder.CreateIntrinsic(RetTy, IntrID, Args, {}, "ld.struct");
809 Value *LoadedValue = Builder.CreateExtractValue(ResRet, {0}, "ld.value");
810 Value *StatusBit = Builder.CreateExtractValue(ResRet, {1}, "ld.status");
811 Value *ExtendedStatus =
812 Builder.CreateZExt(StatusBit, Builder.getInt32Ty(), "ld.status.ext");
813 Builder.CreateStore(ExtendedStatus, StatusAddr);
814
815 return LoadedValue;
816 }
817 case Builtin::BI__builtin_hlsl_resource_uninitializedhandle: {
818 llvm::Type *HandleTy = CGM.getTypes().ConvertType(E->getType());
819 return llvm::PoisonValue::get(HandleTy);
820 }
821 case Builtin::BI__builtin_hlsl_resource_handlefrombinding: {
822 llvm::Type *HandleTy = CGM.getTypes().ConvertType(E->getType());
823 Value *RegisterOp = EmitScalarExpr(E->getArg(1));
824 Value *SpaceOp = EmitScalarExpr(E->getArg(2));
825 Value *RangeOp = EmitScalarExpr(E->getArg(3));
826 Value *IndexOp = EmitScalarExpr(E->getArg(4));
827 Value *Name = EmitScalarExpr(E->getArg(5));
828 llvm::Intrinsic::ID IntrinsicID =
829 CGM.getHLSLRuntime().getCreateHandleFromBindingIntrinsic();
830 SmallVector<Value *> Args{SpaceOp, RegisterOp, RangeOp, IndexOp, Name};
831 return Builder.CreateIntrinsic(HandleTy, IntrinsicID, Args);
832 }
833 case Builtin::BI__builtin_hlsl_resource_handlefromimplicitbinding: {
834 llvm::Type *HandleTy = CGM.getTypes().ConvertType(E->getType());
835 Value *OrderID = EmitScalarExpr(E->getArg(1));
836 Value *SpaceOp = EmitScalarExpr(E->getArg(2));
837 Value *RangeOp = EmitScalarExpr(E->getArg(3));
838 Value *IndexOp = EmitScalarExpr(E->getArg(4));
839 Value *Name = EmitScalarExpr(E->getArg(5));
840 llvm::Intrinsic::ID IntrinsicID =
841 CGM.getHLSLRuntime().getCreateHandleFromImplicitBindingIntrinsic();
842 SmallVector<Value *> Args{OrderID, SpaceOp, RangeOp, IndexOp, Name};
843 return Builder.CreateIntrinsic(HandleTy, IntrinsicID, Args);
844 }
845 case Builtin::BI__builtin_hlsl_resource_counterhandlefromimplicitbinding: {
846 Value *MainHandle = EmitScalarExpr(E->getArg(0));
847 if (!CGM.getTriple().isSPIRV())
848 return MainHandle;
849
850 llvm::Type *HandleTy = CGM.getTypes().ConvertType(E->getType());
851 Value *OrderID = EmitScalarExpr(E->getArg(1));
852 Value *SpaceOp = EmitScalarExpr(E->getArg(2));
853 llvm::Intrinsic::ID IntrinsicID =
854 llvm::Intrinsic::spv_resource_counterhandlefromimplicitbinding;
855 SmallVector<Value *> Args{MainHandle, OrderID, SpaceOp};
856 return Builder.CreateIntrinsic(HandleTy, IntrinsicID, Args);
857 }
858 case Builtin::BI__builtin_hlsl_resource_nonuniformindex: {
859 Value *IndexOp = EmitScalarExpr(E->getArg(0));
860 llvm::Type *RetTy = ConvertType(E->getType());
861 return Builder.CreateIntrinsic(
862 RetTy, CGM.getHLSLRuntime().getNonUniformResourceIndexIntrinsic(),
863 ArrayRef<Value *>{IndexOp});
864 }
865 case Builtin::BI__builtin_hlsl_resource_getdimensions_x: {
866 Value *Handle = EmitScalarExpr(E->getArg(0));
867 LValue Dim = EmitLValue(E->getArg(1));
868 llvm::Type *RetTy = llvm::Type::getInt32Ty(getLLVMContext());
869 Value *DimValue = Builder.CreateIntrinsic(
870 RetTy, CGM.getHLSLRuntime().getGetDimensionsXIntrinsic(),
871 ArrayRef<Value *>{Handle});
872 return Builder.CreateStore(DimValue, Dim.getAddress());
873 }
874 case Builtin::BI__builtin_hlsl_resource_getstride: {
875 LValue Stride = EmitLValue(E->getArg(1));
876 return emitBufferStride(this, E->getArg(0), Stride);
877 }
878 case Builtin::BI__builtin_hlsl_all: {
879 Value *Op0 = EmitScalarExpr(E->getArg(0));
880 return Builder.CreateIntrinsic(
881 /*ReturnType=*/llvm::Type::getInt1Ty(getLLVMContext()),
882 CGM.getHLSLRuntime().getAllIntrinsic(), ArrayRef<Value *>{Op0}, nullptr,
883 "hlsl.all");
884 }
885 case Builtin::BI__builtin_hlsl_and: {
886 Value *Op0 = EmitScalarExpr(E->getArg(0));
887 Value *Op1 = EmitScalarExpr(E->getArg(1));
888 return Builder.CreateAnd(Op0, Op1, "hlsl.and");
889 }
890 case Builtin::BI__builtin_hlsl_or: {
891 Value *Op0 = EmitScalarExpr(E->getArg(0));
892 Value *Op1 = EmitScalarExpr(E->getArg(1));
893 return Builder.CreateOr(Op0, Op1, "hlsl.or");
894 }
895 case Builtin::BI__builtin_hlsl_any: {
896 Value *Op0 = EmitScalarExpr(E->getArg(0));
897 return Builder.CreateIntrinsic(
898 /*ReturnType=*/llvm::Type::getInt1Ty(getLLVMContext()),
899 CGM.getHLSLRuntime().getAnyIntrinsic(), ArrayRef<Value *>{Op0}, nullptr,
900 "hlsl.any");
901 }
902 case Builtin::BI__builtin_hlsl_asdouble:
903 return handleAsDoubleBuiltin(*this, E);
904 case Builtin::BI__builtin_hlsl_elementwise_clamp: {
905 Value *OpX = EmitScalarExpr(E->getArg(0));
906 Value *OpMin = EmitScalarExpr(E->getArg(1));
907 Value *OpMax = EmitScalarExpr(E->getArg(2));
908
909 QualType Ty = E->getArg(0)->getType();
910 if (auto *VecTy = Ty->getAs<VectorType>())
911 Ty = VecTy->getElementType();
912
913 Intrinsic::ID Intr;
914 if (Ty->isFloatingType()) {
915 Intr = CGM.getHLSLRuntime().getNClampIntrinsic();
916 } else if (Ty->isUnsignedIntegerType()) {
917 Intr = CGM.getHLSLRuntime().getUClampIntrinsic();
918 } else {
919 assert(Ty->isSignedIntegerType());
920 Intr = CGM.getHLSLRuntime().getSClampIntrinsic();
921 }
922 return Builder.CreateIntrinsic(
923 /*ReturnType=*/OpX->getType(), Intr,
924 ArrayRef<Value *>{OpX, OpMin, OpMax}, nullptr, "hlsl.clamp");
925 }
926 case Builtin::BI__builtin_hlsl_crossf16:
927 case Builtin::BI__builtin_hlsl_crossf32: {
928 Value *Op0 = EmitScalarExpr(E->getArg(0));
929 Value *Op1 = EmitScalarExpr(E->getArg(1));
930 assert(E->getArg(0)->getType()->hasFloatingRepresentation() &&
932 "cross operands must have a float representation");
933 // make sure each vector has exactly 3 elements
934 assert(
935 E->getArg(0)->getType()->castAs<VectorType>()->getNumElements() == 3 &&
936 E->getArg(1)->getType()->castAs<VectorType>()->getNumElements() == 3 &&
937 "input vectors must have 3 elements each");
938 return Builder.CreateIntrinsic(
939 /*ReturnType=*/Op0->getType(), CGM.getHLSLRuntime().getCrossIntrinsic(),
940 ArrayRef<Value *>{Op0, Op1}, nullptr, "hlsl.cross");
941 }
942 case Builtin::BI__builtin_hlsl_dot: {
943 Value *Op0 = EmitScalarExpr(E->getArg(0));
944 Value *Op1 = EmitScalarExpr(E->getArg(1));
945 llvm::Type *T0 = Op0->getType();
946 llvm::Type *T1 = Op1->getType();
947
948 // If the arguments are scalars, just emit a multiply
949 if (!T0->isVectorTy() && !T1->isVectorTy()) {
950 if (T0->isFloatingPointTy())
951 return Builder.CreateFMul(Op0, Op1, "hlsl.dot");
952
953 if (T0->isIntegerTy())
954 return Builder.CreateMul(Op0, Op1, "hlsl.dot");
955
956 llvm_unreachable(
957 "Scalar dot product is only supported on ints and floats.");
958 }
959 // For vectors, validate types and emit the appropriate intrinsic
960 assert(CGM.getContext().hasSameUnqualifiedType(E->getArg(0)->getType(),
961 E->getArg(1)->getType()) &&
962 "Dot product operands must have the same type.");
963
964 auto *VecTy0 = E->getArg(0)->getType()->castAs<VectorType>();
965 assert(VecTy0 && "Dot product argument must be a vector.");
966
967 return Builder.CreateIntrinsic(
968 /*ReturnType=*/T0->getScalarType(),
969 getDotProductIntrinsic(CGM.getHLSLRuntime(), VecTy0->getElementType()),
970 ArrayRef<Value *>{Op0, Op1}, nullptr, "hlsl.dot");
971 }
972 case Builtin::BI__builtin_hlsl_dot4add_i8packed: {
973 Value *X = EmitScalarExpr(E->getArg(0));
974 Value *Y = EmitScalarExpr(E->getArg(1));
975 Value *Acc = EmitScalarExpr(E->getArg(2));
976
977 Intrinsic::ID ID = CGM.getHLSLRuntime().getDot4AddI8PackedIntrinsic();
978 // Note that the argument order disagrees between the builtin and the
979 // intrinsic here.
980 return Builder.CreateIntrinsic(
981 /*ReturnType=*/Acc->getType(), ID, ArrayRef<Value *>{Acc, X, Y},
982 nullptr, "hlsl.dot4add.i8packed");
983 }
984 case Builtin::BI__builtin_hlsl_dot4add_u8packed: {
985 Value *X = EmitScalarExpr(E->getArg(0));
986 Value *Y = EmitScalarExpr(E->getArg(1));
987 Value *Acc = EmitScalarExpr(E->getArg(2));
988
989 Intrinsic::ID ID = CGM.getHLSLRuntime().getDot4AddU8PackedIntrinsic();
990 // Note that the argument order disagrees between the builtin and the
991 // intrinsic here.
992 return Builder.CreateIntrinsic(
993 /*ReturnType=*/Acc->getType(), ID, ArrayRef<Value *>{Acc, X, Y},
994 nullptr, "hlsl.dot4add.u8packed");
995 }
996 case Builtin::BI__builtin_hlsl_elementwise_firstbithigh: {
997 Value *X = EmitScalarExpr(E->getArg(0));
998
999 return Builder.CreateIntrinsic(
1000 /*ReturnType=*/ConvertType(E->getType()),
1001 getFirstBitHighIntrinsic(CGM.getHLSLRuntime(), E->getArg(0)->getType()),
1002 ArrayRef<Value *>{X}, nullptr, "hlsl.firstbithigh");
1003 }
1004 case Builtin::BI__builtin_hlsl_elementwise_firstbitlow: {
1005 Value *X = EmitScalarExpr(E->getArg(0));
1006
1007 return Builder.CreateIntrinsic(
1008 /*ReturnType=*/ConvertType(E->getType()),
1009 CGM.getHLSLRuntime().getFirstBitLowIntrinsic(), ArrayRef<Value *>{X},
1010 nullptr, "hlsl.firstbitlow");
1011 }
1012 case Builtin::BI__builtin_hlsl_lerp: {
1013 Value *X = EmitScalarExpr(E->getArg(0));
1014 Value *Y = EmitScalarExpr(E->getArg(1));
1015 Value *S = EmitScalarExpr(E->getArg(2));
1016 if (!E->getArg(0)->getType()->hasFloatingRepresentation())
1017 llvm_unreachable("lerp operand must have a float representation");
1018 return Builder.CreateIntrinsic(
1019 /*ReturnType=*/X->getType(), CGM.getHLSLRuntime().getLerpIntrinsic(),
1020 ArrayRef<Value *>{X, Y, S}, nullptr, "hlsl.lerp");
1021 }
1022 case Builtin::BI__builtin_hlsl_normalize: {
1023 Value *X = EmitScalarExpr(E->getArg(0));
1024
1025 assert(E->getArg(0)->getType()->hasFloatingRepresentation() &&
1026 "normalize operand must have a float representation");
1027
1028 return Builder.CreateIntrinsic(
1029 /*ReturnType=*/X->getType(),
1030 CGM.getHLSLRuntime().getNormalizeIntrinsic(), ArrayRef<Value *>{X},
1031 nullptr, "hlsl.normalize");
1032 }
1033 case Builtin::BI__builtin_hlsl_elementwise_degrees: {
1034 Value *X = EmitScalarExpr(E->getArg(0));
1035
1036 assert(E->getArg(0)->getType()->hasFloatingRepresentation() &&
1037 "degree operand must have a float representation");
1038
1039 return Builder.CreateIntrinsic(
1040 /*ReturnType=*/X->getType(), CGM.getHLSLRuntime().getDegreesIntrinsic(),
1041 ArrayRef<Value *>{X}, nullptr, "hlsl.degrees");
1042 }
1043 case Builtin::BI__builtin_hlsl_elementwise_f16tof32: {
1044 return handleElementwiseF16ToF32(*this, E);
1045 }
1046 case Builtin::BI__builtin_hlsl_elementwise_f32tof16: {
1047 return handleElementwiseF32ToF16(*this, E);
1048 }
1049 case Builtin::BI__builtin_hlsl_elementwise_frac: {
1050 Value *Op0 = EmitScalarExpr(E->getArg(0));
1051 if (!E->getArg(0)->getType()->hasFloatingRepresentation())
1052 llvm_unreachable("frac operand must have a float representation");
1053 return Builder.CreateIntrinsic(
1054 /*ReturnType=*/Op0->getType(), CGM.getHLSLRuntime().getFracIntrinsic(),
1055 ArrayRef<Value *>{Op0}, nullptr, "hlsl.frac");
1056 }
1057 case Builtin::BI__builtin_hlsl_elementwise_isinf: {
1058 Value *Op0 = EmitScalarExpr(E->getArg(0));
1059 llvm::Type *Xty = Op0->getType();
1060 llvm::Type *retType = llvm::Type::getInt1Ty(this->getLLVMContext());
1061 if (Xty->isVectorTy()) {
1062 auto *XVecTy = E->getArg(0)->getType()->castAs<VectorType>();
1063 retType = llvm::VectorType::get(
1064 retType, ElementCount::getFixed(XVecTy->getNumElements()));
1065 }
1066 if (!E->getArg(0)->getType()->hasFloatingRepresentation())
1067 llvm_unreachable("isinf operand must have a float representation");
1068 return Builder.CreateIntrinsic(
1069 retType, CGM.getHLSLRuntime().getIsInfIntrinsic(),
1070 ArrayRef<Value *>{Op0}, nullptr, "hlsl.isinf");
1071 }
1072 case Builtin::BI__builtin_hlsl_elementwise_isnan: {
1073 Value *Op0 = EmitScalarExpr(E->getArg(0));
1074 llvm::Type *Xty = Op0->getType();
1075 llvm::Type *retType = llvm::Type::getInt1Ty(this->getLLVMContext());
1076 if (Xty->isVectorTy()) {
1077 auto *XVecTy = E->getArg(0)->getType()->castAs<VectorType>();
1078 retType = llvm::VectorType::get(
1079 retType, ElementCount::getFixed(XVecTy->getNumElements()));
1080 }
1081 if (!E->getArg(0)->getType()->hasFloatingRepresentation())
1082 llvm_unreachable("isnan operand must have a float representation");
1083 return Builder.CreateIntrinsic(
1084 retType, CGM.getHLSLRuntime().getIsNaNIntrinsic(),
1085 ArrayRef<Value *>{Op0}, nullptr, "hlsl.isnan");
1086 }
1087 case Builtin::BI__builtin_hlsl_mad: {
1088 Value *M = EmitScalarExpr(E->getArg(0));
1089 Value *A = EmitScalarExpr(E->getArg(1));
1090 Value *B = EmitScalarExpr(E->getArg(2));
1092 return Builder.CreateIntrinsic(
1093 /*ReturnType*/ M->getType(), Intrinsic::fmuladd,
1094 ArrayRef<Value *>{M, A, B}, nullptr, "hlsl.fmad");
1095
1097 if (CGM.getTarget().getTriple().getArch() == llvm::Triple::dxil)
1098 return Builder.CreateIntrinsic(
1099 /*ReturnType*/ M->getType(), Intrinsic::dx_imad,
1100 ArrayRef<Value *>{M, A, B}, nullptr, "dx.imad");
1101
1102 Value *Mul = Builder.CreateNSWMul(M, A);
1103 return Builder.CreateNSWAdd(Mul, B);
1104 }
1106 if (CGM.getTarget().getTriple().getArch() == llvm::Triple::dxil)
1107 return Builder.CreateIntrinsic(
1108 /*ReturnType=*/M->getType(), Intrinsic::dx_umad,
1109 ArrayRef<Value *>{M, A, B}, nullptr, "dx.umad");
1110
1111 Value *Mul = Builder.CreateNUWMul(M, A);
1112 return Builder.CreateNUWAdd(Mul, B);
1113 }
1114 case Builtin::BI__builtin_hlsl_mul: {
1115 Value *Op0 = EmitScalarExpr(E->getArg(0));
1116 Value *Op1 = EmitScalarExpr(E->getArg(1));
1117 QualType QTy0 = E->getArg(0)->getType();
1118 QualType QTy1 = E->getArg(1)->getType();
1119
1120 bool IsVec0 = QTy0->isVectorType();
1121 bool IsVec1 = QTy1->isVectorType();
1122 bool IsMat0 = QTy0->isConstantMatrixType();
1123 bool IsMat1 = QTy1->isConstantMatrixType();
1124
1125 // The matrix multiply intrinsic only operates on column-major order
1126 // matrices. Therefore matrix memory layout transforms must be inserted
1127 // before and after matrix multiply intrinsics.
1128 bool IsRowMajor = getLangOpts().getDefaultMatrixMemoryLayout() ==
1130
1131 llvm::MatrixBuilder MB(Builder);
1132 if (IsVec0 && IsMat1) {
1133 unsigned N = QTy0->castAs<VectorType>()->getNumElements();
1134 auto *MatTy = QTy1->castAs<ConstantMatrixType>();
1135 unsigned Rows = MatTy->getNumRows();
1136 unsigned Cols = MatTy->getNumColumns();
1137 assert(N == Rows && "vector length must match matrix row count");
1138 if (IsRowMajor)
1139 Op1 = MB.CreateRowMajorToColumnMajorTransform(Op1, Rows, Cols);
1140 return MB.CreateMatrixMultiply(Op0, Op1, 1, N, Cols, "hlsl.mul");
1141 }
1142 if (IsMat0 && IsVec1) {
1143 auto *MatTy = QTy0->castAs<ConstantMatrixType>();
1144 unsigned Rows = MatTy->getNumRows();
1145 unsigned Cols = MatTy->getNumColumns();
1146 assert(QTy1->castAs<VectorType>()->getNumElements() == Cols &&
1147 "vector length must match matrix column count");
1148 if (IsRowMajor)
1149 Op0 = MB.CreateRowMajorToColumnMajorTransform(Op0, Rows, Cols);
1150 return MB.CreateMatrixMultiply(Op0, Op1, Rows, Cols, 1, "hlsl.mul");
1151 }
1152 assert(IsMat0 && IsMat1);
1153 auto *MatTy0 = QTy0->castAs<ConstantMatrixType>();
1154 auto *MatTy1 = QTy1->castAs<ConstantMatrixType>();
1155 unsigned Rows0 = MatTy0->getNumRows();
1156 unsigned Rows1 = MatTy1->getNumRows();
1157 unsigned Cols0 = MatTy0->getNumColumns();
1158 unsigned Cols1 = MatTy1->getNumColumns();
1159 assert(Cols0 == Rows1 &&
1160 "inner matrix dimensions must match for multiplication");
1161 if (IsRowMajor) {
1162 Op0 = MB.CreateRowMajorToColumnMajorTransform(Op0, Rows0, Cols0);
1163 Op1 = MB.CreateRowMajorToColumnMajorTransform(Op1, Rows1, Cols1);
1164 }
1165 Value *Result =
1166 MB.CreateMatrixMultiply(Op0, Op1, Rows0, Cols0, Cols1, "hlsl.mul");
1167 if (IsRowMajor)
1168 Result = MB.CreateColumnMajorToRowMajorTransform(Result, Rows0, Cols1);
1169 return Result;
1170 }
1171 case Builtin::BI__builtin_hlsl_transpose: {
1172 Value *Op0 = EmitScalarExpr(E->getArg(0));
1173 auto *MatTy = E->getArg(0)->getType()->castAs<ConstantMatrixType>();
1174 unsigned Rows = MatTy->getNumRows();
1175 unsigned Cols = MatTy->getNumColumns();
1176 llvm::MatrixBuilder MB(Builder);
1177 // The matrix transpose intrinsic operates on column-major matrices.
1178 // For row-major, a row-major RxC matrix is equivalent to a column-major
1179 // CxR matrix, so transposing with swapped dimensions produces the correct
1180 // row-major CxR result directly.
1181 bool IsRowMajor = getLangOpts().getDefaultMatrixMemoryLayout() ==
1183 if (IsRowMajor)
1184 return MB.CreateMatrixTranspose(Op0, Cols, Rows);
1185 return MB.CreateMatrixTranspose(Op0, Rows, Cols);
1186 }
1187 case Builtin::BI__builtin_hlsl_elementwise_rcp: {
1188 Value *Op0 = EmitScalarExpr(E->getArg(0));
1189 if (!E->getArg(0)->getType()->hasFloatingRepresentation())
1190 llvm_unreachable("rcp operand must have a float representation");
1191 llvm::Type *Ty = Op0->getType();
1192 llvm::Type *EltTy = Ty->getScalarType();
1193 Constant *One = Ty->isVectorTy()
1194 ? ConstantVector::getSplat(
1195 ElementCount::getFixed(
1196 cast<FixedVectorType>(Ty)->getNumElements()),
1197 ConstantFP::get(EltTy, 1.0))
1198 : ConstantFP::get(EltTy, 1.0);
1199 return Builder.CreateFDiv(One, Op0, "hlsl.rcp");
1200 }
1201 case Builtin::BI__builtin_hlsl_elementwise_rsqrt: {
1202 Value *Op0 = EmitScalarExpr(E->getArg(0));
1203 if (!E->getArg(0)->getType()->hasFloatingRepresentation())
1204 llvm_unreachable("rsqrt operand must have a float representation");
1205 return Builder.CreateIntrinsic(
1206 /*ReturnType=*/Op0->getType(), CGM.getHLSLRuntime().getRsqrtIntrinsic(),
1207 ArrayRef<Value *>{Op0}, nullptr, "hlsl.rsqrt");
1208 }
1209 case Builtin::BI__builtin_hlsl_elementwise_saturate: {
1210 Value *Op0 = EmitScalarExpr(E->getArg(0));
1211 assert(E->getArg(0)->getType()->hasFloatingRepresentation() &&
1212 "saturate operand must have a float representation");
1213 return Builder.CreateIntrinsic(
1214 /*ReturnType=*/Op0->getType(),
1215 CGM.getHLSLRuntime().getSaturateIntrinsic(), ArrayRef<Value *>{Op0},
1216 nullptr, "hlsl.saturate");
1217 }
1218 case Builtin::BI__builtin_hlsl_wave_prefix_count_bits: {
1219 Value *Op = EmitScalarExpr(E->getArg(0));
1220 assert(Op->getType()->isIntegerTy(1) &&
1221 "WavePrefixBitCount operand must be a boolean type");
1222
1223 Intrinsic::ID IID =
1225
1226 return EmitRuntimeCall(
1227 Intrinsic::getOrInsertDeclaration(&CGM.getModule(), IID), ArrayRef{Op},
1228 "hlsl.wave.prefix.bit.count");
1229 }
1230 case Builtin::BI__builtin_hlsl_select: {
1231 Value *OpCond = EmitScalarExpr(E->getArg(0));
1232 RValue RValTrue = EmitAnyExpr(E->getArg(1));
1233 Value *OpTrue =
1234 RValTrue.isScalar()
1235 ? RValTrue.getScalarVal()
1236 : Builder.CreateLoad(RValTrue.getAggregateAddress(), "true_val");
1237 RValue RValFalse = EmitAnyExpr(E->getArg(2));
1238 Value *OpFalse =
1239 RValFalse.isScalar()
1240 ? RValFalse.getScalarVal()
1241 : Builder.CreateLoad(RValFalse.getAggregateAddress(), "false_val");
1242 if (auto *VTy = E->getType()->getAs<VectorType>()) {
1243 if (!OpTrue->getType()->isVectorTy())
1244 OpTrue =
1245 Builder.CreateVectorSplat(VTy->getNumElements(), OpTrue, "splat");
1246 if (!OpFalse->getType()->isVectorTy())
1247 OpFalse =
1248 Builder.CreateVectorSplat(VTy->getNumElements(), OpFalse, "splat");
1249 }
1250
1251 Value *SelectVal =
1252 Builder.CreateSelect(OpCond, OpTrue, OpFalse, "hlsl.select");
1253 if (!RValTrue.isScalar())
1254 Builder.CreateStore(SelectVal, ReturnValue.getAddress(),
1255 ReturnValue.isVolatile());
1256
1257 return SelectVal;
1258 }
1259 case Builtin::BI__builtin_hlsl_step: {
1260 Value *Op0 = EmitScalarExpr(E->getArg(0));
1261 Value *Op1 = EmitScalarExpr(E->getArg(1));
1262 assert(E->getArg(0)->getType()->hasFloatingRepresentation() &&
1264 "step operands must have a float representation");
1265 return Builder.CreateIntrinsic(
1266 /*ReturnType=*/Op0->getType(), CGM.getHLSLRuntime().getStepIntrinsic(),
1267 ArrayRef<Value *>{Op0, Op1}, nullptr, "hlsl.step");
1268 }
1269 case Builtin::BI__builtin_hlsl_wave_active_all_equal: {
1270 Value *Op = EmitScalarExpr(E->getArg(0));
1271
1272 Intrinsic::ID ID = CGM.getHLSLRuntime().getWaveActiveAllEqualIntrinsic();
1273 return EmitRuntimeCall(Intrinsic::getOrInsertDeclaration(
1274 &CGM.getModule(), ID, {Op->getType()}),
1275 {Op});
1276 }
1277 case Builtin::BI__builtin_hlsl_wave_active_all_true: {
1278 Value *Op = EmitScalarExpr(E->getArg(0));
1279 assert(Op->getType()->isIntegerTy(1) &&
1280 "Intrinsic WaveActiveAllTrue operand must be a bool");
1281
1282 Intrinsic::ID ID = CGM.getHLSLRuntime().getWaveActiveAllTrueIntrinsic();
1283 return EmitRuntimeCall(
1284 Intrinsic::getOrInsertDeclaration(&CGM.getModule(), ID), {Op});
1285 }
1286 case Builtin::BI__builtin_hlsl_wave_active_any_true: {
1287 Value *Op = EmitScalarExpr(E->getArg(0));
1288 assert(Op->getType()->isIntegerTy(1) &&
1289 "Intrinsic WaveActiveAnyTrue operand must be a bool");
1290
1291 Intrinsic::ID ID = CGM.getHLSLRuntime().getWaveActiveAnyTrueIntrinsic();
1292 return EmitRuntimeCall(
1293 Intrinsic::getOrInsertDeclaration(&CGM.getModule(), ID), {Op});
1294 }
1295 case Builtin::BI__builtin_hlsl_wave_active_bit_or: {
1296 Value *Op = EmitScalarExpr(E->getArg(0));
1297 assert(E->getArg(0)->getType()->hasUnsignedIntegerRepresentation() &&
1298 "Intrinsic WaveActiveBitOr operand must have an unsigned integer "
1299 "representation");
1300
1301 Intrinsic::ID ID = CGM.getHLSLRuntime().getWaveActiveBitOrIntrinsic();
1302 return EmitRuntimeCall(Intrinsic::getOrInsertDeclaration(
1303 &CGM.getModule(), ID, {Op->getType()}),
1304 ArrayRef{Op}, "hlsl.wave.active.bit.or");
1305 }
1306 case Builtin::BI__builtin_hlsl_wave_active_bit_xor: {
1307 Value *Op = EmitScalarExpr(E->getArg(0));
1308 assert(E->getArg(0)->getType()->hasUnsignedIntegerRepresentation() &&
1309 "Intrinsic WaveActiveBitXor operand must have an unsigned integer "
1310 "representation");
1311
1312 Intrinsic::ID ID = CGM.getHLSLRuntime().getWaveActiveBitXorIntrinsic();
1313 return EmitRuntimeCall(Intrinsic::getOrInsertDeclaration(
1314 &CGM.getModule(), ID, {Op->getType()}),
1315 ArrayRef{Op}, "hlsl.wave.active.bit.xor");
1316 }
1317 case Builtin::BI__builtin_hlsl_wave_active_bit_and: {
1318 Value *Op = EmitScalarExpr(E->getArg(0));
1319 assert(E->getArg(0)->getType()->hasUnsignedIntegerRepresentation() &&
1320 "Intrinsic WaveActiveBitAnd operand must have an unsigned integer "
1321 "representation");
1322
1323 Intrinsic::ID ID = CGM.getHLSLRuntime().getWaveActiveBitAndIntrinsic();
1324 return EmitRuntimeCall(Intrinsic::getOrInsertDeclaration(
1325 &CGM.getModule(), ID, {Op->getType()}),
1326 ArrayRef{Op}, "hlsl.wave.active.bit.and");
1327 }
1328 case Builtin::BI__builtin_hlsl_wave_active_ballot: {
1329 [[maybe_unused]] Value *Op = EmitScalarExpr(E->getArg(0));
1330 assert(Op->getType()->isIntegerTy(1) &&
1331 "Intrinsic WaveActiveBallot operand must be a bool");
1332
1333 return handleHlslWaveActiveBallot(*this, E);
1334 }
1335 case Builtin::BI__builtin_hlsl_wave_active_count_bits: {
1336 Value *OpExpr = EmitScalarExpr(E->getArg(0));
1337 Intrinsic::ID ID = CGM.getHLSLRuntime().getWaveActiveCountBitsIntrinsic();
1338 return EmitRuntimeCall(
1339 Intrinsic::getOrInsertDeclaration(&CGM.getModule(), ID),
1340 ArrayRef{OpExpr});
1341 }
1342 case Builtin::BI__builtin_hlsl_wave_active_sum: {
1343 // Due to the use of variadic arguments, explicitly retrieve argument
1344 Value *OpExpr = EmitScalarExpr(E->getArg(0));
1345 Intrinsic::ID IID = getWaveActiveSumIntrinsic(
1346 getTarget().getTriple().getArch(), E->getArg(0)->getType());
1347
1348 return EmitRuntimeCall(Intrinsic::getOrInsertDeclaration(
1349 &CGM.getModule(), IID, {OpExpr->getType()}),
1350 ArrayRef{OpExpr}, "hlsl.wave.active.sum");
1351 }
1352 case Builtin::BI__builtin_hlsl_wave_active_product: {
1353 // Due to the use of variadic arguments, explicitly retrieve argument
1354 Value *OpExpr = EmitScalarExpr(E->getArg(0));
1355 Intrinsic::ID IID = getWaveActiveProductIntrinsic(
1356 getTarget().getTriple().getArch(), E->getArg(0)->getType());
1357
1358 return EmitRuntimeCall(Intrinsic::getOrInsertDeclaration(
1359 &CGM.getModule(), IID, {OpExpr->getType()}),
1360 ArrayRef{OpExpr}, "hlsl.wave.active.product");
1361 }
1362 case Builtin::BI__builtin_hlsl_wave_active_max: {
1363 // Due to the use of variadic arguments, explicitly retrieve argument
1364 Value *OpExpr = EmitScalarExpr(E->getArg(0));
1365 QualType QT = E->getArg(0)->getType();
1366 Intrinsic::ID IID;
1367 if (QT->isUnsignedIntegerType())
1368 IID = CGM.getHLSLRuntime().getWaveActiveUMaxIntrinsic();
1369 else
1370 IID = CGM.getHLSLRuntime().getWaveActiveMaxIntrinsic();
1371
1372 return EmitRuntimeCall(Intrinsic::getOrInsertDeclaration(
1373 &CGM.getModule(), IID, {OpExpr->getType()}),
1374 ArrayRef{OpExpr}, "hlsl.wave.active.max");
1375 }
1376 case Builtin::BI__builtin_hlsl_wave_active_min: {
1377 // Due to the use of variadic arguments, explicitly retrieve argument
1378 Value *OpExpr = EmitScalarExpr(E->getArg(0));
1379 QualType QT = E->getArg(0)->getType();
1380 Intrinsic::ID IID;
1381 if (QT->isUnsignedIntegerType())
1382 IID = CGM.getHLSLRuntime().getWaveActiveUMinIntrinsic();
1383 else
1384 IID = CGM.getHLSLRuntime().getWaveActiveMinIntrinsic();
1385
1386 return EmitRuntimeCall(Intrinsic::getOrInsertDeclaration(
1387 &CGM.getModule(), IID, {OpExpr->getType()}),
1388 ArrayRef{OpExpr}, "hlsl.wave.active.min");
1389 }
1390 case Builtin::BI__builtin_hlsl_wave_get_lane_index: {
1391 // We don't define a SPIR-V intrinsic, instead it is a SPIR-V built-in
1392 // defined in SPIRVBuiltins.td. So instead we manually get the matching name
1393 // for the DirectX intrinsic and the demangled builtin name
1394 switch (CGM.getTarget().getTriple().getArch()) {
1395 case llvm::Triple::dxil:
1396 return EmitRuntimeCall(Intrinsic::getOrInsertDeclaration(
1397 &CGM.getModule(), Intrinsic::dx_wave_getlaneindex));
1398 case llvm::Triple::spirv:
1399 return EmitRuntimeCall(CGM.CreateRuntimeFunction(
1400 llvm::FunctionType::get(IntTy, {}, false),
1401 "__hlsl_wave_get_lane_index", {}, false, true));
1402 default:
1403 llvm_unreachable(
1404 "Intrinsic WaveGetLaneIndex not supported by target architecture");
1405 }
1406 }
1407 case Builtin::BI__builtin_hlsl_wave_is_first_lane: {
1408 Intrinsic::ID ID = CGM.getHLSLRuntime().getWaveIsFirstLaneIntrinsic();
1409 return EmitRuntimeCall(
1410 Intrinsic::getOrInsertDeclaration(&CGM.getModule(), ID));
1411 }
1412 case Builtin::BI__builtin_hlsl_wave_get_lane_count: {
1413 Intrinsic::ID ID = CGM.getHLSLRuntime().getWaveGetLaneCountIntrinsic();
1414 return EmitRuntimeCall(
1415 Intrinsic::getOrInsertDeclaration(&CGM.getModule(), ID));
1416 }
1417 case Builtin::BI__builtin_hlsl_wave_read_lane_at: {
1418 // Due to the use of variadic arguments we must explicitly retrieve them and
1419 // create our function type.
1420 Value *OpExpr = EmitScalarExpr(E->getArg(0));
1421 Value *OpIndex = EmitScalarExpr(E->getArg(1));
1422 return EmitRuntimeCall(
1423 Intrinsic::getOrInsertDeclaration(
1424 &CGM.getModule(), CGM.getHLSLRuntime().getWaveReadLaneAtIntrinsic(),
1425 {OpExpr->getType()}),
1426 ArrayRef{OpExpr, OpIndex}, "hlsl.wave.readlane");
1427 }
1428 case Builtin::BI__builtin_hlsl_wave_prefix_sum: {
1429 Value *OpExpr = EmitScalarExpr(E->getArg(0));
1430 Intrinsic::ID IID = getWavePrefixSumIntrinsic(
1431 getTarget().getTriple().getArch(), E->getArg(0)->getType());
1432 return EmitRuntimeCall(Intrinsic::getOrInsertDeclaration(
1433 &CGM.getModule(), IID, {OpExpr->getType()}),
1434 ArrayRef{OpExpr}, "hlsl.wave.prefix.sum");
1435 }
1436 case Builtin::BI__builtin_hlsl_wave_prefix_product: {
1437 Value *OpExpr = EmitScalarExpr(E->getArg(0));
1438 Intrinsic::ID IID = getWavePrefixProductIntrinsic(
1439 getTarget().getTriple().getArch(), E->getArg(0)->getType());
1440 return EmitRuntimeCall(Intrinsic::getOrInsertDeclaration(
1441 &CGM.getModule(), IID, {OpExpr->getType()}),
1442 ArrayRef{OpExpr}, "hlsl.wave.prefix.product");
1443 }
1444 case Builtin::BI__builtin_hlsl_quad_read_across_x: {
1445 Value *OpExpr = EmitScalarExpr(E->getArg(0));
1446 Intrinsic::ID ID = CGM.getHLSLRuntime().getQuadReadAcrossXIntrinsic();
1447 return EmitRuntimeCall(Intrinsic::getOrInsertDeclaration(
1448 &CGM.getModule(), ID, {OpExpr->getType()}),
1449 ArrayRef{OpExpr}, "hlsl.quad.read.across.x");
1450 }
1451 case Builtin::BI__builtin_hlsl_quad_read_across_y: {
1452 Value *OpExpr = EmitScalarExpr(E->getArg(0));
1453 Intrinsic::ID ID = CGM.getHLSLRuntime().getQuadReadAcrossYIntrinsic();
1454 return EmitRuntimeCall(Intrinsic::getOrInsertDeclaration(
1455 &CGM.getModule(), ID, {OpExpr->getType()}),
1456 ArrayRef{OpExpr}, "hlsl.quad.read.across.y");
1457 }
1458 case Builtin::BI__builtin_hlsl_elementwise_sign: {
1459 auto *Arg0 = E->getArg(0);
1460 Value *Op0 = EmitScalarExpr(Arg0);
1461 llvm::Type *Xty = Op0->getType();
1462 llvm::Type *retType = llvm::Type::getInt32Ty(this->getLLVMContext());
1463 if (Xty->isVectorTy()) {
1464 auto *XVecTy = Arg0->getType()->castAs<VectorType>();
1465 retType = llvm::VectorType::get(
1466 retType, ElementCount::getFixed(XVecTy->getNumElements()));
1467 }
1468 assert((Arg0->getType()->hasFloatingRepresentation() ||
1469 Arg0->getType()->hasIntegerRepresentation()) &&
1470 "sign operand must have a float or int representation");
1471
1472 if (Arg0->getType()->hasUnsignedIntegerRepresentation()) {
1473 Value *Cmp = Builder.CreateICmpEQ(Op0, ConstantInt::get(Xty, 0));
1474 return Builder.CreateSelect(Cmp, ConstantInt::get(retType, 0),
1475 ConstantInt::get(retType, 1), "hlsl.sign");
1476 }
1477
1478 return Builder.CreateIntrinsic(
1479 retType, CGM.getHLSLRuntime().getSignIntrinsic(),
1480 ArrayRef<Value *>{Op0}, nullptr, "hlsl.sign");
1481 }
1482 case Builtin::BI__builtin_hlsl_elementwise_radians: {
1483 Value *Op0 = EmitScalarExpr(E->getArg(0));
1484 assert(E->getArg(0)->getType()->hasFloatingRepresentation() &&
1485 "radians operand must have a float representation");
1486 return Builder.CreateIntrinsic(
1487 /*ReturnType=*/Op0->getType(),
1488 CGM.getHLSLRuntime().getRadiansIntrinsic(), ArrayRef<Value *>{Op0},
1489 nullptr, "hlsl.radians");
1490 }
1491 case Builtin::BI__builtin_hlsl_buffer_update_counter: {
1492 Value *ResHandle = EmitScalarExpr(E->getArg(0));
1493 Value *Offset = EmitScalarExpr(E->getArg(1));
1494 Value *OffsetI8 = Builder.CreateIntCast(Offset, Int8Ty, true);
1495 return Builder.CreateIntrinsic(
1496 /*ReturnType=*/Offset->getType(),
1497 CGM.getHLSLRuntime().getBufferUpdateCounterIntrinsic(),
1498 ArrayRef<Value *>{ResHandle, OffsetI8}, nullptr);
1499 }
1500 case Builtin::BI__builtin_hlsl_elementwise_splitdouble: {
1501
1502 assert((E->getArg(0)->getType()->hasFloatingRepresentation() &&
1505 "asuint operands types mismatch");
1506 return handleHlslSplitdouble(E, this);
1507 }
1508 case Builtin::BI__builtin_hlsl_elementwise_clip:
1509 assert(E->getArg(0)->getType()->hasFloatingRepresentation() &&
1510 "clip operands types mismatch");
1511 return handleHlslClip(E, this);
1512 case Builtin::BI__builtin_hlsl_group_memory_barrier: {
1513 Intrinsic::ID ID = CGM.getHLSLRuntime().getGroupMemoryBarrierIntrinsic();
1514 return EmitRuntimeCall(
1515 Intrinsic::getOrInsertDeclaration(&CGM.getModule(), ID));
1516 }
1517 case Builtin::BI__builtin_hlsl_group_memory_barrier_with_group_sync: {
1518 Intrinsic::ID ID =
1519 CGM.getHLSLRuntime().getGroupMemoryBarrierWithGroupSyncIntrinsic();
1520 return EmitRuntimeCall(
1521 Intrinsic::getOrInsertDeclaration(&CGM.getModule(), ID));
1522 }
1523 case Builtin::BI__builtin_hlsl_elementwise_ddx_coarse: {
1524 Value *Op0 = EmitScalarExpr(E->getArg(0));
1525 if (!E->getArg(0)->getType()->hasFloatingRepresentation())
1526 llvm_unreachable("ddx_coarse operand must have a float representation");
1527 Intrinsic::ID ID = CGM.getHLSLRuntime().getDdxCoarseIntrinsic();
1528 return Builder.CreateIntrinsic(/*ReturnType=*/Op0->getType(), ID,
1529 ArrayRef<Value *>{Op0}, nullptr,
1530 "hlsl.ddx.coarse");
1531 }
1532 case Builtin::BI__builtin_hlsl_elementwise_ddy_coarse: {
1533 Value *Op0 = EmitScalarExpr(E->getArg(0));
1534 if (!E->getArg(0)->getType()->hasFloatingRepresentation())
1535 llvm_unreachable("ddy_coarse operand must have a float representation");
1536 Intrinsic::ID ID = CGM.getHLSLRuntime().getDdyCoarseIntrinsic();
1537 return Builder.CreateIntrinsic(/*ReturnType=*/Op0->getType(), ID,
1538 ArrayRef<Value *>{Op0}, nullptr,
1539 "hlsl.ddy.coarse");
1540 }
1541 case Builtin::BI__builtin_hlsl_elementwise_ddx_fine: {
1542 Value *Op0 = EmitScalarExpr(E->getArg(0));
1543 if (!E->getArg(0)->getType()->hasFloatingRepresentation())
1544 llvm_unreachable("ddx_fine operand must have a float representation");
1545 Intrinsic::ID ID = CGM.getHLSLRuntime().getDdxFineIntrinsic();
1546 return Builder.CreateIntrinsic(/*ReturnType=*/Op0->getType(), ID,
1547 ArrayRef<Value *>{Op0}, nullptr,
1548 "hlsl.ddx.fine");
1549 }
1550 case Builtin::BI__builtin_hlsl_elementwise_ddy_fine: {
1551 Value *Op0 = EmitScalarExpr(E->getArg(0));
1552 if (!E->getArg(0)->getType()->hasFloatingRepresentation())
1553 llvm_unreachable("ddy_fine operand must have a float representation");
1554 Intrinsic::ID ID = CGM.getHLSLRuntime().getDdyFineIntrinsic();
1555 return Builder.CreateIntrinsic(/*ReturnType=*/Op0->getType(), ID,
1556 ArrayRef<Value *>{Op0}, nullptr,
1557 "hlsl.ddy.fine");
1558 }
1559 case Builtin::BI__builtin_get_spirv_spec_constant_bool:
1560 case Builtin::BI__builtin_get_spirv_spec_constant_short:
1561 case Builtin::BI__builtin_get_spirv_spec_constant_ushort:
1562 case Builtin::BI__builtin_get_spirv_spec_constant_int:
1563 case Builtin::BI__builtin_get_spirv_spec_constant_uint:
1564 case Builtin::BI__builtin_get_spirv_spec_constant_longlong:
1565 case Builtin::BI__builtin_get_spirv_spec_constant_ulonglong:
1566 case Builtin::BI__builtin_get_spirv_spec_constant_half:
1567 case Builtin::BI__builtin_get_spirv_spec_constant_float:
1568 case Builtin::BI__builtin_get_spirv_spec_constant_double: {
1569 llvm::Function *SpecConstantFn = getSpecConstantFunction(E->getType());
1570 llvm::Value *SpecId = EmitScalarExpr(E->getArg(0));
1571 llvm::Value *DefaultVal = EmitScalarExpr(E->getArg(1));
1572 llvm::Value *Args[] = {SpecId, DefaultVal};
1573 return Builder.CreateCall(SpecConstantFn, Args);
1574 }
1575 }
1576 return nullptr;
1577}
1578
1580 const clang::QualType &SpecConstantType) {
1581
1582 // Find or create the declaration for the function.
1583 llvm::Module *M = &CGM.getModule();
1584 std::string MangledName =
1585 getSpecConstantFunctionName(SpecConstantType, getContext());
1586 llvm::Function *SpecConstantFn = M->getFunction(MangledName);
1587
1588 if (!SpecConstantFn) {
1589 llvm::Type *IntType = ConvertType(getContext().IntTy);
1590 llvm::Type *RetTy = ConvertType(SpecConstantType);
1591 llvm::Type *ArgTypes[] = {IntType, RetTy};
1592 llvm::FunctionType *FnTy = llvm::FunctionType::get(RetTy, ArgTypes, false);
1593 SpecConstantFn = llvm::Function::Create(
1594 FnTy, llvm::GlobalValue::ExternalLinkage, MangledName, M);
1595 }
1596 return SpecConstantFn;
1597}
llvm::Value * EmitOverflowIntrinsic(CodeGenFunction &CGF, const Intrinsic::ID IntrinsicID, llvm::Value *X, llvm::Value *Y, llvm::Value *&Carry)
Emit a call to llvm.
static Intrinsic::ID getWavePrefixSumIntrinsic(llvm::Triple::ArchType Arch, QualType QT)
static Intrinsic::ID getDotProductIntrinsic(CGHLSLRuntime &RT, QualType QT)
static Intrinsic::ID getPrefixCountBitsIntrinsic(llvm::Triple::ArchType Arch)
static Intrinsic::ID getWaveActiveSumIntrinsic(llvm::Triple::ArchType Arch, QualType QT)
static std::string getSpecConstantFunctionName(clang::QualType SpecConstantType, ASTContext &Context)
static Value * handleHlslSplitdouble(const CallExpr *E, CodeGenFunction *CGF)
static Value * emitBufferStride(CodeGenFunction *CGF, const Expr *HandleExpr, LValue &Stride)
static Intrinsic::ID getWavePrefixProductIntrinsic(llvm::Triple::ArchType Arch, QualType QT)
static Value * emitHlslClamp(CodeGenFunction &CGF, const CallExpr *E, unsigned ClampArgIndex)
static Intrinsic::ID getFirstBitHighIntrinsic(CGHLSLRuntime &RT, QualType QT)
static llvm::Type * getOffsetType(CodeGenModule &CGM, llvm::Type *CoordTy)
static Value * handleElementwiseF16ToF32(CodeGenFunction &CGF, const CallExpr *E)
static Value * handleAsDoubleBuiltin(CodeGenFunction &CGF, const CallExpr *E)
static Value * handleHlslWaveActiveBallot(CodeGenFunction &CGF, const CallExpr *E)
static Value * emitHlslOffset(CodeGenFunction &CGF, const CallExpr *E, unsigned OffsetArgIndex, llvm::Type *OffsetTy)
static Value * handleElementwiseF32ToF16(CodeGenFunction &CGF, const CallExpr *E)
static Intrinsic::ID getWaveActiveProductIntrinsic(llvm::Triple::ArchType Arch, QualType QT)
static Value * handleHlslClip(const CallExpr *E, CodeGenFunction *CGF)
#define X(type, name)
Definition Value.h:97
static StringRef getTriple(const Command &Job)
Holds long-lived AST nodes (such as types and decls) that can be referred to throughout the semantic ...
Definition ASTContext.h:226
CallExpr - Represents a function call (C99 6.5.2.2, C++ [expr.call]).
Definition Expr.h:2946
Expr * getArg(unsigned Arg)
getArg - Return the specified argument.
Definition Expr.h:3150
unsigned getNumArgs() const
getNumArgs - Return the number of actual arguments to this call.
Definition Expr.h:3137
Like RawAddress, an abstract representation of an aligned address, but the pointer contained in this ...
Definition Address.h:128
llvm::StoreInst * CreateStore(llvm::Value *Val, Address Addr, bool IsVolatile=false)
Definition CGBuilder.h:146
CallArgList - Type for representing both the value and type of arguments in a call.
Definition CGCall.h:274
CodeGenFunction - This class organizes the per-function state that is used while generating LLVM code...
llvm::Type * ConvertType(QualType T)
llvm::BasicBlock * createBasicBlock(const Twine &name="", llvm::Function *parent=nullptr, llvm::BasicBlock *before=nullptr)
createBasicBlock - Create an LLVM basic block.
const LangOptions & getLangOpts() const
const TargetInfo & getTarget() const
llvm::Function * getSpecConstantFunction(const clang::QualType &SpecConstantType)
LValue EmitHLSLOutArgExpr(const HLSLOutArgExpr *E, CallArgList &Args, QualType Ty)
Definition CGExpr.cpp:6328
void EmitWritebacks(const CallArgList &Args)
EmitWriteback - Emit callbacks for function.
Definition CGCall.cpp:4974
llvm::Value * getTypeSize(QualType Ty)
Returns calculated size of the specified type.
RValue EmitAnyExpr(const Expr *E, AggValueSlot aggSlot=AggValueSlot::ignored(), bool ignoreResult=false)
EmitAnyExpr - Emit code to compute the specified expression which can have any type.
Definition CGExpr.cpp:273
llvm::CallInst * EmitRuntimeCall(llvm::FunctionCallee callee, const Twine &name="")
llvm::Value * EmitScalarExpr(const Expr *E, bool IgnoreResultAssign=false)
EmitScalarExpr - Emit the computation of the specified expression of LLVM scalar type,...
Address ReturnValue
ReturnValue - The temporary alloca to hold the return value.
LValue EmitLValue(const Expr *E, KnownNonNull_t IsKnownNonNull=NotKnownNonNull)
EmitLValue - Emit code to compute a designator that specifies the location of the expression.
Definition CGExpr.cpp:1707
llvm::LLVMContext & getLLVMContext()
llvm::Value * EmitHLSLBuiltinExpr(unsigned BuiltinID, const CallExpr *E, ReturnValueSlot ReturnValue)
This class organizes the cross-function state that is used while generating LLVM code.
CGHLSLRuntime & getHLSLRuntime()
Return a reference to the configured HLSL runtime.
const TargetInfo & getTarget() const
const llvm::Triple & getTriple() const
llvm::Function * getIntrinsic(unsigned IID, ArrayRef< llvm::Type * > Tys={})
LValue - This represents an lvalue references.
Definition CGValue.h:183
Address getAddress() const
Definition CGValue.h:373
RValue - This trivial value class is used to represent the result of an expression that is evaluated.
Definition CGValue.h:42
bool isScalar() const
Definition CGValue.h:64
Address getAggregateAddress() const
getAggregateAddr() - Return the Value* of the address of the aggregate.
Definition CGValue.h:84
llvm::Value * getScalarVal() const
getScalarVal() - Return the Value* of this scalar value.
Definition CGValue.h:72
ReturnValueSlot - Contains the address where the return value of a function can be stored,...
Definition CGCall.h:379
Represents a concrete matrix type with constant number of rows and columns.
Definition TypeBase.h:4437
unsigned getNumColumns() const
Returns the number of columns in the matrix.
Definition TypeBase.h:4456
unsigned getNumRows() const
Returns the number of rows in the matrix.
Definition TypeBase.h:4453
The name of a declaration.
This represents one expression.
Definition Expr.h:112
QualType getType() const
Definition Expr.h:144
Represents a function declaration or definition.
Definition Decl.h:2015
static FunctionDecl * Create(ASTContext &C, DeclContext *DC, SourceLocation StartLoc, SourceLocation NLoc, DeclarationName N, QualType T, TypeSourceInfo *TInfo, StorageClass SC, bool UsesFPIntrin=false, bool isInlineSpecified=false, bool hasWrittenPrototype=true, ConstexprSpecKind ConstexprKind=ConstexprSpecKind::Unspecified, const AssociatedConstraint &TrailingRequiresClause={})
Definition Decl.h:2204
Represents a parameter to a function.
Definition Decl.h:1805
static ParmVarDecl * Create(ASTContext &C, DeclContext *DC, SourceLocation StartLoc, SourceLocation IdLoc, const IdentifierInfo *Id, QualType T, TypeSourceInfo *TInfo, StorageClass S, Expr *DefArg)
Definition Decl.cpp:2958
A (possibly-)qualified type.
Definition TypeBase.h:937
const Type * getTypePtr() const
Retrieves a pointer to the underlying (unqualified) type.
Definition TypeBase.h:8431
Encodes a location in the source.
bool areArgsDestroyedLeftToRightInCallee() const
Are arguments to a call destroyed left to right in the callee?
const llvm::Triple & getTriple() const
Returns the target triple of the primary target.
TargetCXXABI getCXXABI() const
Get the C++ ABI currently in use.
bool isSignedIntegerType() const
Return true if this is an integer type that is signed, according to C99 6.2.5p4 [char,...
Definition Type.cpp:2231
bool hasIntegerRepresentation() const
Determine whether this type has an integer representation of some sort, e.g., it is an integer type o...
Definition Type.cpp:2084
bool isConstantMatrixType() const
Definition TypeBase.h:8835
const T * castAs() const
Member-template castAs<specific type>.
Definition TypeBase.h:9328
bool hasUnsignedIntegerRepresentation() const
Determine whether this type has an unsigned integer representation of some sort, e....
Definition Type.cpp:2329
bool hasSignedIntegerRepresentation() const
Determine whether this type has an signed integer representation of some sort, e.g....
Definition Type.cpp:2275
bool hasFloatingRepresentation() const
Determine whether this type has a floating-point representation of some sort, e.g....
Definition Type.cpp:2350
bool isVectorType() const
Definition TypeBase.h:8807
bool isFloatingType() const
Definition Type.cpp:2342
bool isUnsignedIntegerType() const
Return true if this is an integer type that is unsigned, according to C99 6.2.5p6 [which returns true...
Definition Type.cpp:2285
const T * getAs() const
Member-template getAs<specific type>'.
Definition TypeBase.h:9261
QualType getType() const
Definition Value.cpp:237
Represents a GCC generic vector type.
Definition TypeBase.h:4225
unsigned getNumElements() const
Definition TypeBase.h:4240
The JSON file list parser is used to communicate input to InstallAPI.
@ SC_Extern
Definition Specifiers.h:251
@ SC_None
Definition Specifiers.h:250
Expr * Cond
};
@ Result
The result type of a method or function.
Definition TypeBase.h:905
U cast(CodeGen::Address addr)
Definition Address.h:327
Diagnostic wrappers for TextAPI types for error reporting.
Definition Dominators.h:30
llvm::IntegerType * Int8Ty
i8, i16, i32, and i64
Extra information about a function prototype.
Definition TypeBase.h:5442