c487e248b9
When the triple is spirv-unknown-unknown, SPIRVSubtarget::Env starts as Unknown and was set via const_cast in SPIRVCallLowering when the first entry point was lowered. This is too late: SPIRVLegalizerInfo has already been constructed with, for example, the wrong vector size limits, causing a crash (at best) or invalid SPIR-V generation. Resolve the environment early in SPIRVPrepareFunctions::runOnModule() by scanning the module for "hlsl.shader" attributes. Reinitialize the legalizer and extended instruction sets after resolution. Remove the const_cast lazy setting from SPIRVCallLowering. Fixes: https://github.com/llvm/llvm-project/issues/171898
54 lines
1.8 KiB
C++
54 lines
1.8 KiB
C++
//===-- SPIRVTargetMachine.h - Define TargetMachine for SPIR-V -*- 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file declares the SPIR-V specific subclass of TargetMachine.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_LIB_TARGET_SPIRV_SPIRVTARGETMACHINE_H
|
|
#define LLVM_LIB_TARGET_SPIRV_SPIRVTARGETMACHINE_H
|
|
|
|
#include "SPIRVSubtarget.h"
|
|
#include "llvm/CodeGen/CodeGenTargetMachineImpl.h"
|
|
#include <optional>
|
|
|
|
namespace llvm {
|
|
class SPIRVTargetMachine : public CodeGenTargetMachineImpl {
|
|
std::unique_ptr<TargetLoweringObjectFile> TLOF;
|
|
SPIRVSubtarget Subtarget;
|
|
|
|
public:
|
|
SPIRVTargetMachine(const Target &T, const Triple &TT, StringRef CPU,
|
|
StringRef FS, const TargetOptions &Options,
|
|
std::optional<Reloc::Model> RM,
|
|
std::optional<CodeModel::Model> CM, CodeGenOptLevel OL,
|
|
bool JIT);
|
|
|
|
const SPIRVSubtarget *getSubtargetImpl() const { return &Subtarget; }
|
|
|
|
const SPIRVSubtarget *getSubtargetImpl(const Function &) const override {
|
|
return &Subtarget;
|
|
}
|
|
|
|
SPIRVSubtarget *getMutableSubtargetImpl() { return &Subtarget; }
|
|
|
|
TargetTransformInfo getTargetTransformInfo(const Function &F) const override;
|
|
|
|
TargetPassConfig *createPassConfig(PassManagerBase &PM) override;
|
|
bool usesPhysRegsForValues() const override { return false; }
|
|
|
|
TargetLoweringObjectFile *getObjFileLowering() const override {
|
|
return TLOF.get();
|
|
}
|
|
|
|
void registerPassBuilderCallbacks(PassBuilder &PB) override;
|
|
};
|
|
} // namespace llvm
|
|
|
|
#endif // LLVM_LIB_TARGET_SPIRV_SPIRVTARGETMACHINE_H
|