1 files changed, 21 insertions, 7 deletions
diff --git a/zluda_inject/tests/inject.rs b/zluda_inject/tests/inject.rs
index 15e5e04..3e6ae97 100644
--- a/zluda_inject/tests/inject.rs
+++ b/zluda_inject/tests/inject.rs
@@ -6,18 +6,28 @@ fn direct_cuinit() -> io::Result<()> {
 }
 
 #[test]
-fn indirect_cuinit() -> io::Result<()> {
-    run_process_and_check_for_zluda_dump("indirect_cuinit")
+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() -> io::Result<()> {
-    run_process_and_check_for_zluda_dump("do_cuinit_main")
+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 do_cuinit_clr() -> io::Result<()> {
-    run_process_and_check_for_zluda_dump("do_cuinit_main_clr")
+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<()> {
@@ -27,7 +37,11 @@ fn run_process_and_check_for_zluda_dump(name: &'static str) -> io::Result<()> {
     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(&zluda_dump_dll).arg("--").arg(&exe_under_test);
+    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();