99 lines
3.0 KiB
C++
99 lines
3.0 KiB
C++
//=- DXILMetadataAnalysis.h - Representation of Module metadata --*- C++ -*-=//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_ANALYSIS_DXILMETADATA_H
|
|
#define LLVM_ANALYSIS_DXILMETADATA_H
|
|
|
|
#include "llvm/ADT/SmallVector.h"
|
|
#include "llvm/IR/PassManager.h"
|
|
#include "llvm/MC/DXContainerInfo.h"
|
|
#include "llvm/Pass.h"
|
|
#include "llvm/Support/VersionTuple.h"
|
|
#include "llvm/TargetParser/Triple.h"
|
|
|
|
namespace llvm {
|
|
|
|
class Function;
|
|
namespace dxil {
|
|
|
|
struct EntryProperties {
|
|
const Function *Entry{nullptr};
|
|
// Specific target shader stage may be specified for entry functions
|
|
Triple::EnvironmentType ShaderStage{Triple::UnknownEnvironment};
|
|
unsigned NumThreadsX{0}; // X component
|
|
unsigned NumThreadsY{0}; // Y component
|
|
unsigned NumThreadsZ{0}; // Z component
|
|
unsigned WaveSizeMin{0}; // Minimum component
|
|
unsigned WaveSizeMax{0}; // Maximum component
|
|
unsigned WaveSizePref{0}; // Preferred component
|
|
|
|
EntryProperties(const Function *Fn = nullptr) : Entry(Fn) {};
|
|
};
|
|
|
|
struct ModuleMetadataInfo {
|
|
VersionTuple DXILVersion{};
|
|
VersionTuple ShaderModelVersion{};
|
|
Triple::EnvironmentType ShaderProfile{Triple::UnknownEnvironment};
|
|
VersionTuple ValidatorVersion{};
|
|
SmallVector<EntryProperties> EntryPropertyVec{};
|
|
std::optional<mcdxbc::SourceInfoBuilder> SourceInfo;
|
|
LLVM_ABI void print(raw_ostream &OS) const;
|
|
};
|
|
|
|
} // namespace dxil
|
|
|
|
// Module metadata analysis pass for new pass manager
|
|
class DXILMetadataAnalysis : public AnalysisInfoMixin<DXILMetadataAnalysis> {
|
|
friend AnalysisInfoMixin<DXILMetadataAnalysis>;
|
|
|
|
static AnalysisKey Key;
|
|
|
|
public:
|
|
using Result = dxil::ModuleMetadataInfo;
|
|
/// Gather module metadata info for the module \c M.
|
|
LLVM_ABI dxil::ModuleMetadataInfo run(Module &M, ModuleAnalysisManager &AM);
|
|
};
|
|
|
|
/// Printer pass for the \c DXILMetadataAnalysis results.
|
|
class DXILMetadataAnalysisPrinterPass
|
|
: public RequiredPassInfoMixin<DXILMetadataAnalysisPrinterPass> {
|
|
raw_ostream &OS;
|
|
|
|
public:
|
|
explicit DXILMetadataAnalysisPrinterPass(raw_ostream &OS) : OS(OS) {}
|
|
|
|
LLVM_ABI PreservedAnalyses run(Module &M, ModuleAnalysisManager &AM);
|
|
};
|
|
|
|
/// Legacy pass
|
|
class LLVM_ABI DXILMetadataAnalysisWrapperPass : public ModulePass {
|
|
std::unique_ptr<dxil::ModuleMetadataInfo> MetadataInfo;
|
|
|
|
public:
|
|
static char ID; // Class identification, replacement for typeinfo
|
|
|
|
DXILMetadataAnalysisWrapperPass();
|
|
~DXILMetadataAnalysisWrapperPass() override;
|
|
|
|
const dxil::ModuleMetadataInfo &getModuleMetadata() const {
|
|
return *MetadataInfo;
|
|
}
|
|
dxil::ModuleMetadataInfo &getModuleMetadata() { return *MetadataInfo; }
|
|
|
|
void getAnalysisUsage(AnalysisUsage &AU) const override;
|
|
bool runOnModule(Module &M) override;
|
|
void releaseMemory() override;
|
|
|
|
void print(raw_ostream &OS, const Module *M) const override;
|
|
void dump() const;
|
|
};
|
|
|
|
} // namespace llvm
|
|
|
|
#endif // LLVM_ANALYSIS_DXILMETADATA_H
|