@@ -9,6 +9,7 @@ const PRIMITIVE_DTYPES: &[&str] = &[
99 "u8" , "u16" , "u32" , "u64" , "i8" , "i16" , "i32" , "i64" , "f32" , "f64" ,
1010] ;
1111const PRIMITIVE_DTYPE_ENV : & str = "VORTEX_CUDF_PRIMITIVE_DTYPE" ;
12+ const HARNESS_COMMANDS : & [ & str ] = & [ "check" , "check-stream" ] ;
1213
1314fn 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