wip

Author: ZuseZ4

src/bootstrap/src/core/build_steps/compile.rs

@@ -2216,6 +2216,42 @@ impl Step for Assemble {
             }
         }
 
+        if builder.config.llvm_offload && !builder.config.dry_run() {
+            let offload_install = builder.ensure(llvm::OmpOffload { target: build_compiler.host });
+            if let Some(llvm_config) = builder.llvm_config(builder.config.host_target) {
+                let src_dir = offload_install.join("lib");
+                let target_libdir =
+                    builder.sysroot_target_libdir(target_compiler, target_compiler.host);
+                let llvm = llvm::get_llvm_version(builder, &llvm_config);
+                let llvm_version = llvm.split('.').take(2).collect::<Vec<_>>().join(".");
+                let rust_version = builder.rust_version();
+                // e.g. libLLVMOffload.so.21.1
+                let lib_ext = std::env::consts::DLL_EXTENSION;
+                let libenzyme = format!("libLLVMOffload.{lib_ext}.{llvm_version}");
+                dbg!(&libenzyme);
+                let dst_lib = libdir.join(&libenzyme).with_extension(lib_ext);
+                let target_dst_lib = target_libdir.join(&libenzyme).with_extension(lib_ext);
+                let src_lib = src_dir.join(&libenzyme);
+                builder.copy_link(&src_lib, &dst_lib, FileType::NativeLibrary);
+                builder.copy_link(&src_lib, &target_dst_lib, FileType::NativeLibrary);
+
+                // libomp.so, libomptarget.so
+                let omp = src_dir.join("libomp").with_extension(lib_ext);
+                let tgt = src_dir.join("libomptarget").with_extension(lib_ext);
+                builder.copy_link(&omp, &dst_lib, FileType::NativeLibrary);
+                builder.copy_link(&omp, &dst_lib, FileType::NativeLibrary);
+                builder.copy_link(&tgt, &dst_lib, FileType::NativeLibrary);
+                builder.copy_link(&tgt, &dst_lib, FileType::NativeLibrary);
+                //[\u@\h:\W]$ ls ../offload-outdir/lib
+                // amdgcn-amd-amdhsa  libarcher.so        libgomp.so    libiomp5.so	libLLVMOffload.so.21.1	libomp.so	 libomptarget.so.21.1
+                // cmake		   libarcher_static.a  libgomp.so.1  libLLVMOffload.so	libompd.so		libomptarget.so  nvptx64-nvidia-cuda
+                // [\u@\h:\W]$ ls ../offload-outdir/lib/amdgcn-amd-amdhsa
+                // libompdevice.a	libomptarget-amdgpu.bc
+                // [\u@\h:\W]$ ls nv
+                // [\u@\h:\W]$ ls ../offload-outdir/lib/nvptx64-nvidia-cuda
+            }
+        }
+
         // Build the libraries for this compiler to link to (i.e., the libraries
         // it uses at runtime).
         debug!(