1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
//===- DXILOpBuilder.h - Helper class for build DIXLOp functions ----------===//
//
// 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
//
//===----------------------------------------------------------------------===//
///
/// \file This file contains class to help build DXIL op functions.
//===----------------------------------------------------------------------===//
#ifndef LLVM_LIB_TARGET_DIRECTX_DXILOPBUILDER_H
#define LLVM_LIB_TARGET_DIRECTX_DXILOPBUILDER_H
#include "DXILConstants.h"
#include "llvm/ADT/SmallVector.h"
#include "llvm/IR/IRBuilder.h"
#include "llvm/Support/DXILABI.h"
#include "llvm/Support/Error.h"
#include "llvm/TargetParser/Triple.h"
namespace llvm {
class Module;
class IRBuilderBase;
class CallInst;
class Constant;
class Value;
class Type;
class FunctionType;
namespace dxil {
class DXILOpBuilder {
public:
DXILOpBuilder(Module &M);
IRBuilder<> &getIRB() { return IRB; }
/// Create a call instruction for the given DXIL op. The arguments
/// must be valid for an overload of the operation.
CallInst *createOp(dxil::OpCode Op, ArrayRef<Value *> Args,
const Twine &Name = "", Type *RetTy = nullptr);
/// Try to create a call instruction for the given DXIL op. Fails if the
/// overload is invalid.
Expected<CallInst *> tryCreateOp(dxil::OpCode Op, ArrayRef<Value *> Args,
const Twine &Name = "",
Type *RetTy = nullptr);
/// Get a `%dx.types.ResRet` type with the given element type.
StructType *getResRetType(Type *ElementTy);
/// Get a `%dx.types.CBufRet` type with the given element type.
StructType *getCBufRetType(Type *ElementTy);
/// Get the `%dx.types.Handle` type.
StructType *getHandleType();
/// Get a constant `%dx.types.ResBind` value.
Constant *getResBind(uint32_t LowerBound, uint32_t UpperBound,
uint32_t SpaceID, dxil::ResourceClass RC);
/// Get a constant `%dx.types.ResourceProperties` value.
Constant *getResProps(uint32_t Word0, uint32_t Word1);
/// Return the name of the given opcode.
static const char *getOpCodeName(dxil::OpCode DXILOp);
private:
/// Gets a specific overload type of the function for the given DXIL op. If
/// the operation is not overloaded, \c OverloadType may be nullptr.
FunctionType *getOpFunctionType(dxil::OpCode OpCode,
Type *OverloadType = nullptr);
Module &M;
IRBuilder<> IRB;
VersionTuple DXILVersion;
Triple::EnvironmentType ShaderStage;
};
} // namespace dxil
} // namespace llvm
#endif
|
