target triple

Author: ZuseZ4

compiler/rustc_codegen_llvm/src/back/lto.rs

@@ -660,7 +660,6 @@ pub(crate) fn run_pass_manager(
     // in the enzyme differentiation pass.
     let enable_ad = config.autodiff.contains(&config::AutoDiff::Enable);
     let enable_gpu = config.offload.contains(&config::Offload::Enable);
-    dbg!(&enable_gpu);
     let stage = if thin {
         write::AutodiffStage::PreAD
     } else {
@@ -676,6 +675,7 @@ pub(crate) fn run_pass_manager(
     }
 
     if cfg!(llvm_enzyme) && enable_gpu && !thin {
+        dbg!(&enable_gpu);
         let cx =
             SimpleCx::new(module.module_llvm.llmod(), &module.module_llvm.llcx, cgcx.pointer_size);
         crate::builder::gpu_offload::handle_gpu_code(cgcx, &cx);

compiler/rustc_codegen_llvm/src/builder/gpu_offload.rs

@@ -1,4 +1,4 @@
-use std::ffi::CString;
+use std::ffi::{CStr, CString};
 
 use crate::builder::{SBuilder, UNNAMED};
 use crate::common::AsCCharPtr;
@@ -29,6 +29,9 @@ pub(crate) fn gen_image_wrapper_module<'ll>(
         let ti8 = cx.type_i8();
         let dl_cstr = llvm::LLVMGetDataLayoutStr(old_cx.llmod);
         llvm::LLVMSetDataLayout(llmod, dl_cstr);
+        let target_cstr = llvm::LLVMGetTarget(old_cx.llmod);
+        let target = CStr::from_ptr(target_cstr).to_string_lossy().into_owned();
+        llvm::LLVMSetTarget(llmod, target_cstr);
         //  target triple = "x86_64-unknown-linux-gnu"
 
         let mut entry_fields = [ti64, ti16, ti16, ti32, tptr, tptr, ti64, ti64, tptr];

compiler/rustc_codegen_llvm/src/llvm/ffi.rs

@@ -1016,6 +1016,9 @@ unsafe extern "C" {
     ) -> &Module;
     pub(crate) fn LLVMCloneModule(M: &Module) -> &Module;
 
+    pub(crate) fn LLVMGetTarget(M: &Module) -> *const c_char;
+    pub(crate) fn LLVMSetTarget(M: &Module, Name: *const c_char);
+
     /// Data layout. See Module::getDataLayout.
     pub(crate) fn LLVMGetDataLayoutStr(M: &Module) -> *const c_char;
     pub(crate) fn LLVMSetDataLayout(M: &Module, Triple: *const c_char);