zluda_inject/tests/inject.rs


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50

use std::{env, io, path::PathBuf, process::Command};

#[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")
}

fn run_process_and_check_for_zluda_dump(name: &'static str) -> io::Result<()> {
    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());
    let stderr_text = String::from_utf8(test_output.stderr).unwrap();
    assert!(stderr_text.contains("ZLUDA_DUMP"));
    Ok(())
}