27#include "llvm/IR/IntrinsicsDirectX.h"
39 bool HasUAVWithBinding =
any_of(
41 if (!HasUAVWithBinding)
86 case Intrinsic::dx_wave_is_first_lane:
87 case Intrinsic::dx_wave_getlaneindex:
88 case Intrinsic::dx_wave_get_lane_count:
89 case Intrinsic::dx_wave_any:
90 case Intrinsic::dx_wave_all_equal:
91 case Intrinsic::dx_wave_all:
92 case Intrinsic::dx_wave_readlane:
93 case Intrinsic::dx_wave_readlane_first:
94 case Intrinsic::dx_wave_active_countbits:
95 case Intrinsic::dx_wave_ballot:
96 case Intrinsic::dx_wave_prefix_bit_count:
98 case Intrinsic::dx_wave_reduce_or:
99 case Intrinsic::dx_wave_reduce_xor:
100 case Intrinsic::dx_wave_reduce_and:
101 case Intrinsic::dx_wave_reduce_sum:
102 case Intrinsic::dx_wave_reduce_usum:
103 case Intrinsic::dx_wave_product:
104 case Intrinsic::dx_wave_uproduct:
105 case Intrinsic::dx_wave_reduce_max:
106 case Intrinsic::dx_wave_reduce_umax:
107 case Intrinsic::dx_wave_reduce_min:
108 case Intrinsic::dx_wave_reduce_umin:
110 case Intrinsic::dx_wave_prefix_sum:
111 case Intrinsic::dx_wave_prefix_usum:
112 case Intrinsic::dx_wave_prefix_product:
113 case Intrinsic::dx_wave_prefix_uproduct:
115 case Intrinsic::dx_quad_read_across_x:
116 case Intrinsic::dx_quad_read_across_y:
117 case Intrinsic::dx_quad_read_across_diagonal:
145 switch (
II.getIntrinsicID()) {
148 case Intrinsic::dx_resource_load_level:
149 case Intrinsic::dx_resource_sample:
150 case Intrinsic::dx_resource_sample_clamp:
153 case Intrinsic::dx_resource_samplebias:
154 case Intrinsic::dx_resource_samplebias_clamp:
155 case Intrinsic::dx_resource_samplelevel:
158 case Intrinsic::dx_resource_samplegrad:
159 case Intrinsic::dx_resource_samplegrad_clamp:
169 return Flag->getValue().getBoolValue();
178 assert(IID == Intrinsic::dx_resource_load_typedbuffer ||
179 IID == Intrinsic::dx_resource_load_rawbuffer &&
180 "unexpected intrinsic ID");
181 for (
const User *U :
II.users()) {
185 if (EVI->getNumIndices() == 1 && EVI->getIndices()[0] == 1)
201 CSF.Doubles =
I.getType()->getScalarType()->isDoubleTy();
204 for (
const Value *
Op :
I.operands()) {
205 if (
Op->getType()->getScalarType()->isDoubleTy()) {
213 switch (
I.getOpcode()) {
214 case Instruction::FDiv:
215 case Instruction::UIToFP:
216 case Instruction::SIToFP:
217 case Instruction::FPToUI:
218 case Instruction::FPToSI:
219 CSF.DX11_1_DoubleExtensions =
true;
224 if (!CSF.LowPrecisionPresent)
225 CSF.LowPrecisionPresent =
I.getType()->getScalarType()->isIntegerTy(16) ||
226 I.getType()->getScalarType()->isHalfTy();
228 if (!CSF.LowPrecisionPresent) {
229 for (
const Value *
Op :
I.operands()) {
230 if (
Op->getType()->getScalarType()->isIntegerTy(16) ||
231 Op->getType()->getScalarType()->isHalfTy()) {
232 CSF.LowPrecisionPresent =
true;
238 if (CSF.LowPrecisionPresent) {
239 if (CSF.NativeLowPrecisionMode)
240 CSF.NativeLowPrecision =
true;
242 CSF.MinimumPrecision =
true;
246 CSF.Int64Ops =
I.getType()->getScalarType()->isIntegerTy(64);
249 for (
const Value *
Op :
I.operands()) {
250 if (
Op->getType()->getScalarType()->isIntegerTy(64)) {
260 switch (
II->getIntrinsicID()) {
263 case Intrinsic::dx_resource_handlefrombinding: {
268 if (!CSF.ResMayNotAlias && CanSetResMayNotAlias &&
270 CSF.ResMayNotAlias =
true;
275 CSF.EnableRawAndStructuredBuffers =
true;
282 case Intrinsic::dx_resource_handlefromheap: {
285 CSF.SamplerDescriptorHeapIndexing |= IsSamplerHeap;
286 CSF.ResourceDescriptorHeapIndexing |= !IsSamplerHeap;
288 if (!CSF.ResMayNotAlias && CanSetResMayNotAlias && RTI.
isUAV() &&
290 CSF.ResMayNotAlias =
true;
294 case Intrinsic::dx_resource_load_level:
295 case Intrinsic::dx_resource_load_typedbuffer: {
296 dxil::ResourceTypeInfo &RTI =
300 if (
II->getIntrinsicID() == Intrinsic::dx_resource_load_typedbuffer &&
302 CSF.TiledResources =
true;
305 case Intrinsic::dx_resource_load_rawbuffer: {
307 CSF.TiledResources =
true;
310 case Intrinsic::dx_resource_atomic_binop:
311 case Intrinsic::dx_resource_atomic_compare_exchange: {
312 if (
II->getType()->isIntegerTy(64)) {
313 dxil::ResourceTypeInfo &RTI =
316 CSF.AtomicInt64OnTypedResource =
true;
326 if (ARMW->getValOperand()->getType()->isIntegerTy(64) &&
327 ARMW->getPointerAddressSpace() == 3)
328 CSF.AtomicInt64OnGroupShared =
true;
330 if (AXCG->getNewValOperand()->getType()->isIntegerTy(64) &&
331 AXCG->getPointerAddressSpace() == 3)
332 CSF.AtomicInt64OnGroupShared =
true;
336 const Function *CF = CI->getCalledFunction();
338 if (FunctionFlags.contains(CF))
339 CSF.
merge(FunctionFlags[CF]);
341 CSF.DX11_1_DoubleExtensions |=
349ModuleShaderFlags::gatherGlobalModuleFlags(
const Module &M,
350 const DXILResourceMap &DRM,
353 ComputedShaderFlags CSF;
360 uint32_t NumUAVs = 0;
363 if (!
UAV.hasBinding())
369 uint32_t NewNum = NumUAVs + (
Size == 0 ? ~0
U :
Size);
370 if (NewNum < NumUAVs)
376 CSF.Max64UAVs =
true;
386 M.getModuleFlag(
"dx.nativelowprec")))
388 CSF.NativeLowPrecisionMode = NativeLowPrec->getValue().getBoolValue();
393 CSF.ResMayNotAlias = !DRM.
uavs().
empty();
398 M.getModuleFlag(
"dx.allresourcesbound")))
399 if (AllResourcesBound->getValue().getBoolValue())
400 CSF.AllResourcesBound =
true;
415 M.getModuleFlag(
"dx.resmayalias")))
416 if (ResMayAlias->getValue().getBoolValue())
417 CanSetResMayNotAlias =
false;
427 const std::vector<CallGraphNode *> &CurSCC = *SCCI;
439 if (
F->isDeclaration()) {
440 assert(!
F->getName().starts_with(
"dx.op.") &&
441 "DXIL Shader Flag analysis should not be run post-lowering.");
446 for (
const auto &BB : *
F)
447 for (
const auto &
I : BB)
448 updateFunctionFlags(CSF,
I, DRTM, MMDI);
452 CSF.Barycentrics |=
llvm::any_of(Sig->Inputs, [](
const auto &E) {
453 return E.SemanticKind == dxbc::PSV::SemanticKind::Barycentrics;
455 auto HasLowPrecision = [](
const auto &Elements) {
462 if (HasLowPrecision(Sig->Inputs) || HasLowPrecision(Sig->Outputs)) {
463 CSF.LowPrecisionPresent =
true;
464 if (Sig->UseNative16Bit)
465 CSF.NativeLowPrecision =
true;
467 CSF.MinimumPrecision =
true;
471 auto HasShadingRate = [](
const auto &Elements) {
473 return E.SemanticKind == dxbc::PSV::SemanticKind::ShadingRate;
476 if (HasShadingRate(Sig->Inputs) || HasShadingRate(Sig->Outputs))
477 CSF.ShadingRate =
true;
486 CombinedSFMask.merge(SCCSF);
494 FunctionFlags[
F].merge(SCCSF);
499 uint64_t FlagVal = (uint64_t) *
this;
500 OS <<
formatv(
"; Shader Flags Value: {0:x8}\n;\n", FlagVal);
503 OS <<
"; Note: shader requires additional functionality:\n";
504#define SHADER_FEATURE_FLAG(FeatureBit, DxilModuleNum, FlagName, Str) \
506 (OS << ";").indent(7) << Str << "\n";
507#include "llvm/BinaryFormat/DXContainerConstants.def"
508 OS <<
"; Note: extra DXIL module flags:\n";
509#define DXIL_MODULE_FLAG(DxilModuleBit, FlagName, Str) \
511 (OS << ";").indent(7) << Str << "\n";
512#include "llvm/BinaryFormat/DXContainerConstants.def"
519 auto Iter = FunctionFlags.find(Func);
520 assert((Iter != FunctionFlags.end() && Iter->first == Func) &&
521 "Get Shader Flags : No Shader Flags Mask exists for function");
547 OS <<
"; Combined Shader Flags for Module\n";
550 OS <<
"; Shader Flags for Module Functions\n";
551 for (
const auto &
F : M.getFunctionList()) {
552 if (
F.isDeclaration())
555 OS <<
formatv(
"; Function {0} : {1:x8}\n;\n",
F.getName(),
573 MSFI.initialize(M, DRTM, DRM, MMDI,
589 "DXIL Shader Flag Analysis",
true,
true)
assert(UImm &&(UImm !=~static_cast< T >(0)) &&"Invalid immediate!")
This file contains the simple types necessary to represent the attributes associated with functions a...
This file provides interfaces used to build and manipulate a call graph, which is a very useful tool ...
bool checkIfStatusIsExtracted(const IntrinsicInst &II)
static bool isOptimizationDisabled(const Module &M)
static bool hasUAVsAtEveryStage(const DXILResourceMap &DRM, const ModuleMetadataInfo &MMDI)
static bool checkDoubleExtensionOps(Intrinsic::ID IID)
static bool checkAdvancedTextureOps(const IntrinsicInst &II)
Texture load and sample operations accept "programmable offsets", i.e.
static bool checkWaveOps(Intrinsic::ID IID)
Module.h This file contains the declarations for the Module class.
uint64_t IntrinsicInst * II
#define INITIALIZE_PASS_DEPENDENCY(depName)
#define INITIALIZE_PASS_END(passName, arg, name, cfg, analysis)
#define INITIALIZE_PASS_BEGIN(passName, arg, name, cfg, analysis)
This builds on the llvm/ADT/GraphTraits.h file to find the strongly connected components (SCCs) of a ...
This file defines the SmallVector class.
PassT::Result & getResult(IRUnitT &IR, ExtraArgTs... ExtraArgs)
Get the result of an analysis pass for a given IR unit.
Represent the analysis usage information of a pass.
AnalysisUsage & addRequired()
void setPreservesAll()
Set by analyses that do not transform their input at all.
AnalysisUsage & addRequiredTransitive()
A node in the call graph for a module.
The basic data container for the call graph of a Module of IR.
iterator_range< iterator > uavs()
A wrapper class for inspecting calls to intrinsic functions.
A Module instance is used to store all the information related to an LLVM module.
AnalysisType & getAnalysis() const
getAnalysis<AnalysisType>() - This function is used by subclasses to get to the analysis information ...
A set of analyses that are preserved following a run of a transformation pass.
static PreservedAnalyses all()
Construct a special preserved set that preserves all passes.
void push_back(const T &Elt)
This is a 'vector' (really, a variable-sized array), optimized for the case when the array is small.
Represent a constant reference to a string, i.e.
Represents a version number in the form major[.minor[.subminor[.build]]].
const EntrySignature * get(const Function *F) const
LLVM_ABI bool isUAV() const
LLVM_ABI bool isSampler() const
LLVM_ABI bool isTyped() const
LLVM_ABI TypedInfo getTyped() const
dxil::ResourceKind getResourceKind() const
PreservedAnalyses run(Module &M, ModuleAnalysisManager &AM)
Wrapper pass for the legacy pass manager.
void getAnalysisUsage(AnalysisUsage &AU) const override
getAnalysisUsage - This function should be overriden by passes that need analysis information to do t...
bool runOnModule(Module &M) override
runOnModule - Virtual method overriden by subclasses to process the module being operated on.
ModuleShaderFlags run(Module &M, ModuleAnalysisManager &AM)
This class implements an extremely fast bulk output stream that can only output to a stream.
Enumerate the SCCs of a directed graph in reverse topological order of the SCC DAG.
bool isAtEnd() const
Direct loop termination test which is more efficient than comparison with end().
std::enable_if_t< detail::IsValidPointer< X, Y >::value, X * > extract_or_null(Y &&MD)
Extract a Value from Metadata, allowing null.
This is an optimization pass for GlobalISel generic memory operations.
decltype(auto) dyn_cast(const From &Val)
dyn_cast<X> - Return the argument parameter cast to the specified type.
scc_iterator< T > scc_begin(const T &G)
Construct the begin iterator for a deduced graph type T.
RelativeUniformCounterPtr ValuesPtrExpr VTableAddr Value
bool any_of(R &&range, UnaryPredicate P)
Provide wrappers to std::any_of which take ranges instead of having to pass begin/end explicitly.
auto formatv(bool Validate, const char *Fmt, Ts &&...Vals)
bool isa(const From &Val)
isa<X> - Return true if the parameter to the template is an instance of one of the template type argu...
LLVM_ATTRIBUTE_VISIBILITY_DEFAULT AnalysisKey InnerAnalysisManagerProxy< AnalysisManagerT, IRUnitT, ExtraArgTs... >::Key
DWARFExpression::Operation Op
decltype(auto) cast(const From &Val)
cast<X> - Return the argument parameter cast to the specified type.
AnalysisManager< Module > ModuleAnalysisManager
Convenience typedef for the Module analysis manager.
A special type used by analysis passes to provide an address that identifies that particular analysis...
void merge(const ComputedShaderFlags CSF)
void print(raw_ostream &OS=dbgs()) const
const ComputedShaderFlags & getFunctionFlags(const Function *) const
Return the shader flags mask of the specified function Func.
void initialize(Module &, DXILResourceTypeMap &DRTM, const DXILResourceMap &DRM, const ModuleMetadataInfo &MMDI, const ModuleSignatureInfo &Signatures)
Construct ModuleShaderFlags for module Module M.
const ComputedShaderFlags & getCombinedFlags() const