Skip to content

Commit b2b0cbd

Browse files
committed
Run cuDF device stream harness in CI
Signed-off-by: Alexander Droste <alexander.droste@protonmail.com>
1 parent 4ab3b2e commit b2b0cbd

1 file changed

Lines changed: 20 additions & 17 deletions

File tree

vortex-test/e2e-cuda/src/bin/cudf_harness_runner.rs

Lines changed: 20 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@ const PRIMITIVE_DTYPES: &[&str] = &[
99
"u8", "u16", "u32", "u64", "i8", "i16", "i32", "i64", "f32", "f64",
1010
];
1111
const PRIMITIVE_DTYPE_ENV: &str = "VORTEX_CUDF_PRIMITIVE_DTYPE";
12+
const HARNESS_COMMANDS: &[&str] = &["check", "check-stream"];
1213

1314
fn main() -> ExitCode {
1415
let args = env::args().collect::<Vec<_>>();
@@ -21,25 +22,27 @@ fn main() -> ExitCode {
2122
};
2223

2324
for primitive_dtype in PRIMITIVE_DTYPES {
24-
eprintln!("\n== {PRIMITIVE_DTYPE_ENV}={primitive_dtype} :: check ==");
25+
for harness_command in HARNESS_COMMANDS {
26+
eprintln!("\n== {PRIMITIVE_DTYPE_ENV}={primitive_dtype} :: {harness_command} ==");
2527

26-
let status = Command::new("compute-sanitizer")
27-
.args(["--tool", "memcheck", "--error-exitcode", "1"])
28-
.arg(harness)
29-
.arg("check")
30-
.arg(library)
31-
.env(PRIMITIVE_DTYPE_ENV, primitive_dtype)
32-
.status();
28+
let status = Command::new("compute-sanitizer")
29+
.args(["--tool", "memcheck", "--error-exitcode", "1"])
30+
.arg(harness)
31+
.arg(harness_command)
32+
.arg(library)
33+
.env(PRIMITIVE_DTYPE_ENV, primitive_dtype)
34+
.status();
3335

34-
match status {
35-
Ok(status) if status.success() => {}
36-
Ok(status) => {
37-
eprintln!("cudf-test-harness failed with {status}");
38-
return ExitCode::from(1);
39-
}
40-
Err(err) => {
41-
eprintln!("failed to run cudf-test-harness: {err}");
42-
return ExitCode::from(1);
36+
match status {
37+
Ok(status) if status.success() => {}
38+
Ok(status) => {
39+
eprintln!("cudf-test-harness {harness_command} failed with {status}");
40+
return ExitCode::from(1);
41+
}
42+
Err(err) => {
43+
eprintln!("failed to run cudf-test-harness {harness_command}: {err}");
44+
return ExitCode::from(1);
45+
}
4346
}
4447
}
4548
}

0 commit comments

Comments
 (0)