f2650c54c9
- Set the shader flag `DisableOptimizations` based on `optnone` attribute of shader entry functions. - Add DXIL Metadata Analysis pass as pre-requisite for Shader Flags pass to obtain entry function information collected therein. - Named module metadata `dx.disable_optimizations` is intended to indicate disabling optimizations (`-O0`) via commandline flag. However, its intent is fulfilled by `optnone` attribute of shader entry functions as implemented in a recent change, and thus not needed. Delete generation of named metadata and corresponding test file `disable_opt.ll`. - Add tests to verify correctness of setting shader flag. Closes #112263
148 lines
4.8 KiB
C++
148 lines
4.8 KiB
C++
//===- DXILShaderFlags.h - DXIL Shader Flags helper objects ---------------===//
|
|
//
|
|
// 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 helper objects and APIs for working with DXIL
|
|
/// Shader Flags.
|
|
///
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_TARGET_DIRECTX_DXILSHADERFLAGS_H
|
|
#define LLVM_TARGET_DIRECTX_DXILSHADERFLAGS_H
|
|
|
|
#include "llvm/Analysis/DXILMetadataAnalysis.h"
|
|
#include "llvm/IR/Function.h"
|
|
#include "llvm/IR/PassManager.h"
|
|
#include "llvm/Pass.h"
|
|
#include "llvm/Support/Compiler.h"
|
|
#include "llvm/Support/Debug.h"
|
|
#include "llvm/Support/raw_ostream.h"
|
|
#include <cstdint>
|
|
#include <memory>
|
|
|
|
namespace llvm {
|
|
class Module;
|
|
class GlobalVariable;
|
|
class DXILResourceTypeMap;
|
|
|
|
namespace dxil {
|
|
|
|
struct ComputedShaderFlags {
|
|
#define SHADER_FEATURE_FLAG(FeatureBit, DxilModuleBit, FlagName, Str) \
|
|
bool FlagName : 1;
|
|
#define DXIL_MODULE_FLAG(DxilModuleBit, FlagName, Str) bool FlagName : 1;
|
|
#include "llvm/BinaryFormat/DXContainerConstants.def"
|
|
|
|
#define SHADER_FEATURE_FLAG(FeatureBit, DxilModuleBit, FlagName, Str) \
|
|
FlagName = false;
|
|
#define DXIL_MODULE_FLAG(DxilModuleBit, FlagName, Str) FlagName = false;
|
|
ComputedShaderFlags() {
|
|
#include "llvm/BinaryFormat/DXContainerConstants.def"
|
|
}
|
|
|
|
constexpr uint64_t getMask(int Bit) const {
|
|
return Bit != -1 ? 1ull << Bit : 0;
|
|
}
|
|
|
|
uint64_t getModuleFlags() const {
|
|
uint64_t ModuleFlags = 0;
|
|
#define DXIL_MODULE_FLAG(DxilModuleBit, FlagName, Str) \
|
|
ModuleFlags |= FlagName ? getMask(DxilModuleBit) : 0ull;
|
|
#include "llvm/BinaryFormat/DXContainerConstants.def"
|
|
return ModuleFlags;
|
|
}
|
|
|
|
operator uint64_t() const {
|
|
uint64_t FlagValue = getModuleFlags();
|
|
#define SHADER_FEATURE_FLAG(FeatureBit, DxilModuleBit, FlagName, Str) \
|
|
FlagValue |= FlagName ? getMask(DxilModuleBit) : 0ull;
|
|
#include "llvm/BinaryFormat/DXContainerConstants.def"
|
|
return FlagValue;
|
|
}
|
|
|
|
uint64_t getFeatureFlags() const {
|
|
uint64_t FeatureFlags = 0;
|
|
#define SHADER_FEATURE_FLAG(FeatureBit, DxilModuleBit, FlagName, Str) \
|
|
FeatureFlags |= FlagName ? getMask(FeatureBit) : 0ull;
|
|
#include "llvm/BinaryFormat/DXContainerConstants.def"
|
|
return FeatureFlags;
|
|
}
|
|
|
|
void merge(const ComputedShaderFlags CSF) {
|
|
#define SHADER_FEATURE_FLAG(FeatureBit, DxilModuleBit, FlagName, Str) \
|
|
FlagName |= CSF.FlagName;
|
|
#define DXIL_MODULE_FLAG(DxilModuleBit, FlagName, Str) FlagName |= CSF.FlagName;
|
|
#include "llvm/BinaryFormat/DXContainerConstants.def"
|
|
}
|
|
|
|
void print(raw_ostream &OS = dbgs()) const;
|
|
LLVM_DUMP_METHOD void dump() const { print(); }
|
|
};
|
|
|
|
struct ModuleShaderFlags {
|
|
void initialize(Module &, DXILResourceTypeMap &DRTM,
|
|
const ModuleMetadataInfo &MMDI);
|
|
const ComputedShaderFlags &getFunctionFlags(const Function *) const;
|
|
const ComputedShaderFlags &getCombinedFlags() const { return CombinedSFMask; }
|
|
|
|
private:
|
|
/// Map of Function-Shader Flag Mask pairs representing properties of each of
|
|
/// the functions in the module. Shader Flags of each function represent both
|
|
/// module-level and function-level flags
|
|
DenseMap<const Function *, ComputedShaderFlags> FunctionFlags;
|
|
/// Combined Shader Flag Mask of all functions of the module
|
|
ComputedShaderFlags CombinedSFMask{};
|
|
void updateFunctionFlags(ComputedShaderFlags &, const Instruction &,
|
|
DXILResourceTypeMap &);
|
|
};
|
|
|
|
class ShaderFlagsAnalysis : public AnalysisInfoMixin<ShaderFlagsAnalysis> {
|
|
friend AnalysisInfoMixin<ShaderFlagsAnalysis>;
|
|
static AnalysisKey Key;
|
|
|
|
public:
|
|
ShaderFlagsAnalysis() = default;
|
|
|
|
using Result = ModuleShaderFlags;
|
|
|
|
ModuleShaderFlags run(Module &M, ModuleAnalysisManager &AM);
|
|
};
|
|
|
|
/// Printer pass for ShaderFlagsAnalysis results.
|
|
class ShaderFlagsAnalysisPrinter
|
|
: public PassInfoMixin<ShaderFlagsAnalysisPrinter> {
|
|
raw_ostream &OS;
|
|
|
|
public:
|
|
explicit ShaderFlagsAnalysisPrinter(raw_ostream &OS) : OS(OS) {}
|
|
PreservedAnalyses run(Module &M, ModuleAnalysisManager &AM);
|
|
};
|
|
|
|
/// Wrapper pass for the legacy pass manager.
|
|
///
|
|
/// This is required because the passes that will depend on this are codegen
|
|
/// passes which run through the legacy pass manager.
|
|
class ShaderFlagsAnalysisWrapper : public ModulePass {
|
|
ModuleShaderFlags MSFI;
|
|
|
|
public:
|
|
static char ID;
|
|
|
|
ShaderFlagsAnalysisWrapper() : ModulePass(ID) {}
|
|
|
|
const ModuleShaderFlags &getShaderFlags() { return MSFI; }
|
|
|
|
bool runOnModule(Module &M) override;
|
|
|
|
void getAnalysisUsage(AnalysisUsage &AU) const override;
|
|
};
|
|
|
|
} // namespace dxil
|
|
} // namespace llvm
|
|
|
|
#endif // LLVM_TARGET_DIRECTX_DXILSHADERFLAGS_H
|