1b9ba2b233
Too many changes to list, but broadly: * Remove Intel GPU support from the compiler * Add AMD GPU support to the compiler * Remove Intel GPU host code * Add AMD GPU host code * More device instructions. From 40 to 68 * More host functions. From 48 to 184 * Add proof of concept implementation of OptiX framework * Add minimal support of cuDNN, cuBLAS, cuSPARSE, cuFFT, NCCL, NVML * Improve ZLUDA launcher for Windows
97 lines
3.4 KiB
Rust
97 lines
3.4 KiB
Rust
use embed_manifest::{embed_manifest, new_manifest};
|
|
use std::{
|
|
env::{self, VarError},
|
|
fs::{self, DirEntry},
|
|
io,
|
|
path::{self, PathBuf},
|
|
process::Command,
|
|
};
|
|
|
|
fn main() -> Result<(), VarError> {
|
|
if std::env::var_os("CARGO_CFG_WINDOWS").is_some() {
|
|
embed_manifest(new_manifest("zluda_with")).expect("unable to embed manifest file");
|
|
}
|
|
println!("cargo:rerun-if-changed=build.rs");
|
|
if env::var("PROFILE")? != "debug" {
|
|
return Ok(());
|
|
}
|
|
if env::var("CARGO_CFG_TARGET_OS")? != "windows" {
|
|
return Ok(());
|
|
}
|
|
let rustc_exe = env::var("RUSTC")?;
|
|
let out_dir = env::var("OUT_DIR")?;
|
|
let target = env::var("TARGET")?;
|
|
let is_msvc = env::var("CARGO_CFG_TARGET_ENV")? == "msvc";
|
|
let opt_level = env::var("OPT_LEVEL")?;
|
|
let debug = str::parse::<bool>(env::var("DEBUG")?.as_str()).unwrap();
|
|
let mut helpers_dir = PathBuf::from(env::var("CARGO_MANIFEST_DIR")?);
|
|
helpers_dir.push("tests");
|
|
helpers_dir.push("helpers");
|
|
let helpers_dir_as_string = helpers_dir.to_string_lossy();
|
|
println!("cargo:rerun-if-changed={}", helpers_dir_as_string);
|
|
for rust_file in fs::read_dir(&helpers_dir).unwrap().filter_map(rust_file) {
|
|
let full_file_path = format!(
|
|
"{}{}{}",
|
|
helpers_dir_as_string,
|
|
path::MAIN_SEPARATOR,
|
|
rust_file
|
|
);
|
|
let mut rustc_cmd = Command::new(&*rustc_exe);
|
|
if debug {
|
|
rustc_cmd.arg("-g");
|
|
}
|
|
rustc_cmd.arg(format!("-Lnative={}", helpers_dir_as_string));
|
|
if !is_msvc {
|
|
// HACK ALERT
|
|
// I have no idea why the extra library below has to be linked
|
|
rustc_cmd.arg(r"-lucrt");
|
|
} else {
|
|
// For some reason rustc emits foobar.dll.lib and then expects foobar.lib
|
|
let mut implib_path = PathBuf::from(&out_dir);
|
|
let implib = PathBuf::from(rust_file);
|
|
implib_path.push(format!(
|
|
"{}.lib",
|
|
implib.file_stem().unwrap().to_string_lossy()
|
|
));
|
|
let link_args = format!("link-args=/IMPLIB:{}", implib_path.as_path().display());
|
|
rustc_cmd.args(["-C", link_args.as_str()]);
|
|
}
|
|
rustc_cmd
|
|
.arg("-ldylib=nvcuda")
|
|
.arg("-C")
|
|
.arg(format!("opt-level={}", opt_level))
|
|
.arg("-L")
|
|
.arg(format!("{}", out_dir))
|
|
.arg("--out-dir")
|
|
.arg(format!("{}", out_dir))
|
|
.arg("--target")
|
|
.arg(format!("{}", target))
|
|
.arg(full_file_path);
|
|
assert!(rustc_cmd.status().unwrap().success());
|
|
}
|
|
std::fs::copy(
|
|
format!(
|
|
"{}{}do_cuinit_late_clr.exe",
|
|
helpers_dir_as_string,
|
|
path::MAIN_SEPARATOR
|
|
),
|
|
format!("{}{}do_cuinit_late_clr.exe", out_dir, path::MAIN_SEPARATOR),
|
|
)
|
|
.unwrap();
|
|
println!("cargo:rustc-env=HELPERS_OUT_DIR={}", &out_dir);
|
|
Ok(())
|
|
}
|
|
|
|
fn rust_file(entry: io::Result<DirEntry>) -> Option<String> {
|
|
entry.ok().and_then(|e| {
|
|
let os_file_name = e.file_name();
|
|
let file_name = os_file_name.to_string_lossy();
|
|
let is_file = e.file_type().ok().map(|t| t.is_file()).unwrap_or(false);
|
|
if is_file && file_name.ends_with(".rs") {
|
|
Some(file_name.to_string())
|
|
} else {
|
|
None
|
|
}
|
|
})
|
|
}
|