File tree 2 files changed +7
-1
lines changed
torch/csrc/jit/tensorexpr 2 files changed +7
-1
lines changed Original file line number Diff line number Diff line change @@ -739,7 +739,9 @@ void LLVMCodeGenImpl::emitKernel(
739
739
PM,
740
740
asmStream,
741
741
nullptr ,
742
- #if LLVM_VERSION_MAJOR >= 10
742
+ #if LLVM_VERSION_MAJOR >= 18
743
+ llvm::CodeGenFileType::AssemblyFile);
744
+ #elif LLVM_VERSION_MAJOR >= 10
743
745
llvm::CodeGenFileType::CGFT_AssemblyFile);
744
746
#else
745
747
llvm::TargetMachine::CodeGenFileType::CGFT_AssemblyFile);
Original file line number Diff line number Diff line change @@ -101,7 +101,11 @@ static llvm::orc::JITTargetMachineBuilder makeTargetMachineBuilder(
101
101
c10::optional<std::string> attrs) {
102
102
auto JTMB = triple ? makeJTMBFromTriple (*triple, cpu, attrs)
103
103
: makeJTMBFromHost (cpu, attrs);
104
+ #if LLVM_VERSION_MAJOR >= 18
105
+ JTMB.setCodeGenOptLevel (llvm::CodeGenOptLevel::Default);
106
+ #else
104
107
JTMB.setCodeGenOptLevel (llvm::CodeGenOpt::Default);
108
+ #endif
105
109
JTMB.getOptions ().AllowFPOpFusion = llvm::FPOpFusion::Fast;
106
110
return JTMB;
107
111
}
You can’t perform that action at this time.
0 commit comments