File tree Expand file tree Collapse file tree 2 files changed +6
-15
lines changed
Expand file tree Collapse file tree 2 files changed +6
-15
lines changed Original file line number Diff line number Diff line change 66use std:: fs:: File ;
77use std:: mem;
88use std:: sync:: Arc ;
9+ use std:: thread:: sleep;
10+ use std:: time:: Duration ;
911
1012use criterion:: { Criterion , Throughput , criterion_group, criterion_main} ;
1113use cudarc:: cufile:: Cufile ;
@@ -92,6 +94,8 @@ fn benchmark_gpu_scan(c: &mut Criterion) {
9294 let file_device_slice = read_file_to_device ( & cuda_ctx. default_stream ( ) , file) ;
9395 let slice_view = & file_device_slice;
9496
97+ sleep ( Duration :: from_millis ( 60 * 1000 ) ) ;
98+
9599 group. throughput ( Throughput :: Bytes ( ( len * size_of :: < u32 > ( ) * 2 ) as u64 ) ) ;
96100 group. bench_function ( * label, |b| {
97101 b. to_async ( & runtime) . iter_with_large_drop ( async || {
@@ -103,7 +107,7 @@ fn benchmark_gpu_scan(c: &mut Criterion) {
103107 . open ( bench_file_name)
104108 . await
105109 . vortex_unwrap ( ) ;
106- let stream = vx_file
110+ vx_file
107111 . gpu_scan (
108112 cuda_ctx. clone ( ) ,
109113 Arc :: new ( FileGpuSegmentSource :: new (
@@ -117,19 +121,6 @@ fn benchmark_gpu_scan(c: &mut Criterion) {
117121 . try_collect :: < Vec < _ > > ( )
118122 . await
119123 . vortex_unwrap ( ) ;
120- stream
121-
122- // VortexOpenOptions::new()
123- // .open(bench_file_name)
124- // .await
125- // .vortex_unwrap()
126- // .gpu_scan(ctx.clone())
127- // .vortex_unwrap()
128- // .into_array_stream()
129- // .vortex_unwrap()
130- // .try_collect::<Vec<_>>()
131- // .await
132- // .vortex_unwrap()
133124 } ) ;
134125 } ) ;
135126 }
Original file line number Diff line number Diff line change @@ -62,7 +62,7 @@ pub fn create_run_jit_kernel(
6262 let mut out = unsafe {
6363 stream
6464 . alloc:: <P >( len)
65- . map_err( |e| vortex_err!( "failed to alloc zeros {e}" ) ) ?
65+ . map_err( |e| vortex_err!( "failed to alloc output {e}" ) ) ?
6666 } ;
6767 launch_builder. arg( & mut out) ;
6868 let start = stream
You can’t perform that action at this time.
0 commit comments