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
68 lines
1.9 KiB
Rust
68 lines
1.9 KiB
Rust
use std::{
|
|
env, io,
|
|
path::PathBuf,
|
|
process::{Command, Output},
|
|
};
|
|
|
|
#[test]
|
|
fn direct_cuinit() -> io::Result<()> {
|
|
run_process_and_check_for_zluda_dump("direct_cuinit")
|
|
}
|
|
|
|
#[test]
|
|
fn do_cuinit_early() -> io::Result<()> {
|
|
run_process_and_check_for_zluda_dump("do_cuinit_early")
|
|
}
|
|
|
|
#[test]
|
|
fn do_cuinit_late() -> io::Result<()> {
|
|
run_process_and_check_for_zluda_dump("do_cuinit_late")
|
|
}
|
|
|
|
#[test]
|
|
fn do_cuinit_late_clr() -> io::Result<()> {
|
|
run_process_and_check_for_zluda_dump("do_cuinit_late_clr")
|
|
}
|
|
|
|
#[test]
|
|
fn indirect_cuinit() -> io::Result<()> {
|
|
run_process_and_check_for_zluda_dump("indirect_cuinit")
|
|
}
|
|
|
|
#[test]
|
|
fn subprocess() -> io::Result<()> {
|
|
run_process_and_check_for_zluda_dump("subprocess")
|
|
}
|
|
|
|
#[test]
|
|
fn query_exe() -> io::Result<()> {
|
|
let process_output = run_process("query_exe")?;
|
|
let stdout_text = String::from_utf8(process_output.stdout).unwrap();
|
|
assert!(stdout_text.contains("SESSIONNAME"));
|
|
Ok(())
|
|
}
|
|
|
|
fn run_process_and_check_for_zluda_dump(name: &'static str) -> io::Result<()> {
|
|
let process_output = run_process(name)?;
|
|
let stderr_text = String::from_utf8(process_output.stderr).unwrap();
|
|
assert!(stderr_text.contains("ZLUDA_DUMP"));
|
|
Ok(())
|
|
}
|
|
|
|
fn run_process(name: &'static str) -> io::Result<Output> {
|
|
let zluda_with_exe = PathBuf::from(env!("CARGO_BIN_EXE_zluda_with"));
|
|
let mut zluda_dump_dll = zluda_with_exe.parent().unwrap().to_path_buf();
|
|
zluda_dump_dll.push("zluda_dump.dll");
|
|
let helpers_dir = env!("HELPERS_OUT_DIR");
|
|
let exe_under_test = format!("{}{}{}.exe", helpers_dir, std::path::MAIN_SEPARATOR, name);
|
|
let mut test_cmd = Command::new(&zluda_with_exe);
|
|
let test_cmd = test_cmd
|
|
.arg("--nvcuda")
|
|
.arg(&zluda_dump_dll)
|
|
.arg("--")
|
|
.arg(&exe_under_test);
|
|
let test_output = test_cmd.output()?;
|
|
assert!(test_output.status.success());
|
|
Ok(test_output)
|
|
}
|