[LLVM][RUNTIME] Fix CodeModel propagation to ORCJIT runtime executor

apache · Sep 8, 2024 · 5836db4 · 5836db4
1 parent e468426
commit 5836db4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/target/llvm/llvm_instance.h b/src/target/llvm/llvm_instance.h
@@ -215,6 +215,11 @@ class LLVMTargetInfo {
    * \return `llvm::TargetOptions` object for this target
    */
   const llvm::TargetOptions& GetTargetOptions() const { return target_options_; }
+  /*!
+   * \brief Get the LLVM target code model
+   * \return `llvm::CodeModel` object for this target
+   */
+  const llvm::CodeModel::Model& GetTargetCodeModel() const { return code_model_; }
   /*!
    * \brief Get fast math flags
    * \return `llvm::FastMathFlags` for this target

diff --git a/src/target/llvm/llvm_module.cc b/src/target/llvm/llvm_module.cc
@@ -476,6 +476,7 @@ void LLVMModuleNode::InitORCJIT() {
   tm_builder.setCPU(llvm_target->GetCPU());
   tm_builder.setFeatures(llvm_target->GetTargetFeatureString());
   tm_builder.setOptions(llvm_target->GetTargetOptions());
+  tm_builder.setCodeModel(llvm_target->GetTargetCodeModel());
 #if TVM_LLVM_VERSION <= 170
   tm_builder.setCodeGenOptLevel(llvm::CodeGenOpt::Aggressive);
 #else