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_start_vertex_location:
264 case Intrinsic::dx_start_instance_location:
265 CSF.ExtendedCommandInfo =
true;
267 case Intrinsic::dx_view_id:
270 case Intrinsic::dx_resource_handlefrombinding: {
275 if (!CSF.ResMayNotAlias && CanSetResMayNotAlias &&
277 CSF.ResMayNotAlias =
true;
282 CSF.EnableRawAndStructuredBuffers =
true;
289 case Intrinsic::dx_resource_handlefromheap: {
292 CSF.SamplerDescriptorHeapIndexing |= IsSamplerHeap;
293 CSF.ResourceDescriptorHeapIndexing |= !IsSamplerHeap;
295 if (!CSF.ResMayNotAlias && CanSetResMayNotAlias && RTI.
isUAV() &&
297 CSF.ResMayNotAlias =
true;
301 case Intrinsic::dx_resource_load_level:
302 case Intrinsic::dx_resource_load_typedbuffer: {
303 dxil::ResourceTypeInfo &RTI =
307 if (
II->getIntrinsicID() == Intrinsic::dx_resource_load_typedbuffer &&
309 CSF.TiledResources =
true;
312 case Intrinsic::dx_resource_load_rawbuffer: {
314 CSF.TiledResources =
true;
317 case Intrinsic::dx_resource_atomic_binop:
318 case Intrinsic::dx_resource_atomic_compare_exchange: {
319 if (
II->getType()->isIntegerTy(64)) {
320 dxil::ResourceTypeInfo &RTI =
323 CSF.AtomicInt64OnTypedResource =
true;
333 if (ARMW->getValOperand()->getType()->isIntegerTy(64) &&
334 ARMW->getPointerAddressSpace() == 3)
335 CSF.AtomicInt64OnGroupShared =
true;
337 if (AXCG->getNewValOperand()->getType()->isIntegerTy(64) &&
338 AXCG->getPointerAddressSpace() == 3)
339 CSF.AtomicInt64OnGroupShared =
true;
343 const Function *CF = CI->getCalledFunction();
345 if (FunctionFlags.contains(CF))
346 CSF.
merge(FunctionFlags[CF]);
348 CSF.DX11_1_DoubleExtensions |=
356ModuleShaderFlags::gatherGlobalModuleFlags(
const Module &M,
357 const DXILResourceMap &DRM,
360 ComputedShaderFlags CSF;
367 uint32_t NumUAVs = 0;
370 if (!
UAV.hasBinding())
376 uint32_t NewNum = NumUAVs + (
Size == 0 ? ~0
U :
Size);
377 if (NewNum < NumUAVs)
383 CSF.Max64UAVs =
true;
393 M.getModuleFlag(
"dx.nativelowprec")))
395 CSF.NativeLowPrecisionMode = NativeLowPrec->getValue().getBoolValue();
400 CSF.ResMayNotAlias = !DRM.
uavs().
empty();
405 M.getModuleFlag(
"dx.allresourcesbound")))
406 if (AllResourcesBound->getValue().getBoolValue())
407 CSF.AllResourcesBound =
true;
422 M.getModuleFlag(
"dx.resmayalias")))
423 if (ResMayAlias->getValue().getBoolValue())
424 CanSetResMayNotAlias =
false;
434 const std::vector<CallGraphNode *> &CurSCC = *SCCI;
446 if (
F->isDeclaration()) {
447 assert(!
F->getName().starts_with(
"dx.op.") &&
448 "DXIL Shader Flag analysis should not be run post-lowering.");
453 for (
const auto &BB : *
F)
454 for (
const auto &
I : BB)
455 updateFunctionFlags(CSF,
I, DRTM, MMDI);
459 CSF.Barycentrics |=
llvm::any_of(Sig->Inputs, [](
const auto &E) {
460 return E.SemanticKind == dxbc::PSV::SemanticKind::Barycentrics;
464 auto HasLowPrecision = [](
const auto &Elements) {
471 if (HasLowPrecision(Sig->Inputs) || HasLowPrecision(Sig->Outputs)) {
472 CSF.LowPrecisionPresent =
true;
473 if (Sig->UseNative16Bit)
474 CSF.NativeLowPrecision =
true;
476 CSF.MinimumPrecision =
true;
480 auto HasShadingRate = [](
const auto &Elements) {
482 return E.SemanticKind == dxbc::PSV::SemanticKind::ShadingRate;
485 if (HasShadingRate(Sig->Inputs) || HasShadingRate(Sig->Outputs))
486 CSF.ShadingRate =
true;
495 CombinedSFMask.merge(SCCSF);
503 FunctionFlags[
F].merge(SCCSF);
508 uint64_t FlagVal = (uint64_t) *
this;
509 OS <<
formatv(
"; Shader Flags Value: {0:x8}\n;\n", FlagVal);
512 OS <<
"; Note: shader requires additional functionality:\n";
513#define SHADER_FEATURE_FLAG(FeatureBit, DxilModuleNum, FlagName, Str) \
515 (OS << ";").indent(7) << Str << "\n";
516#include "llvm/BinaryFormat/DXContainerConstants.def"
517 OS <<
"; Note: extra DXIL module flags:\n";
518#define DXIL_MODULE_FLAG(DxilModuleBit, FlagName, Str) \
520 (OS << ";").indent(7) << Str << "\n";
521#include "llvm/BinaryFormat/DXContainerConstants.def"
528 auto Iter = FunctionFlags.find(Func);
529 assert((Iter != FunctionFlags.end() && Iter->first == Func) &&
530 "Get Shader Flags : No Shader Flags Mask exists for function");
556 OS <<
"; Combined Shader Flags for Module\n";
559 OS <<
"; Shader Flags for Module Functions\n";
560 for (
const auto &
F : M.getFunctionList()) {
561 if (
F.isDeclaration())
564 OS <<
formatv(
"; Function {0} : {1:x8}\n;\n",
F.getName(),
582 MSFI.initialize(M, DRTM, DRM, MMDI,
598 "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