add+use timer() method on TimedLaunchStrategy

Signed-off-by: Andrew Duffy <andrew@a10y.dev>

Author: a10y

vortex-cuda/benches/alp_cuda.rs

@@ -105,7 +105,7 @@ where
             |b, array| {
                 b.iter_custom(|iters| {
                     let timed = TimedLaunchStrategy::default();
-                    let timer = Arc::clone(&timed.total_time_ns);
+                    let timer = timed.timer();
 
                     let mut cuda_ctx = CudaSession::create_execution_ctx(&VortexSession::empty())
                         .vortex_expect("failed to create execution context")

vortex-cuda/benches/bitpacked_cuda.rs

@@ -119,7 +119,7 @@ where
         |b, array| {
             b.iter_custom(|iters| {
                 let timed = TimedLaunchStrategy::default();
-                let timer = Arc::clone(&timed.total_time_ns);
+                let timer = timed.timer();
 
                 let mut cuda_ctx = CudaSession::create_execution_ctx(&VortexSession::empty())
                     .vortex_expect("failed to create execution context")
@@ -164,7 +164,7 @@ where
             |b, array| {
                 b.iter_custom(|iters| {
                     let timed = TimedLaunchStrategy::default();
-                    let timer = Arc::clone(&timed.total_time_ns);
+                    let timer = timed.timer();
 
                     let mut cuda_ctx = CudaSession::create_execution_ctx(&VortexSession::empty())
                         .vortex_expect("failed to create execution context")

vortex-cuda/benches/common/mod.rs

@@ -16,6 +16,13 @@ pub struct TimedLaunchStrategy {
     pub total_time_ns: Arc<AtomicU64>,
 }
 
+impl TimedLaunchStrategy {
+    /// Returns a shared handle to the accumulated kernel time, for reading after launches complete.
+    pub fn timer(&self) -> Arc<AtomicU64> {
+        Arc::clone(&self.total_time_ns)
+    }
+}
+
 impl LaunchStrategy for TimedLaunchStrategy {
     fn event_flags(&self) -> CUevent_flags {
         // using blocking_sync to make sure all events flush before we complete.

vortex-cuda/benches/date_time_parts_cuda.rs

@@ -64,7 +64,7 @@ fn benchmark_datetimeparts(c: &mut Criterion) {
             |b, dtp_array| {
                 b.iter_custom(|iters| {
                     let timed = TimedLaunchStrategy::default();
-                    let timer = Arc::clone(&timed.total_time_ns);
+                    let timer = timed.timer();
 
                     let mut cuda_ctx = CudaSession::create_execution_ctx(&VortexSession::empty())
                         .vortex_expect("failed to create execution context")

vortex-cuda/benches/dict_cuda.rs

@@ -92,7 +92,7 @@ where
             |b, dict_array| {
                 b.iter_custom(|iters| {
                     let timed = TimedLaunchStrategy::default();
-                    let timer = Arc::clone(&timed.total_time_ns);
+                    let timer = timed.timer();
 
                     let mut cuda_ctx = CudaSession::create_execution_ctx(&VortexSession::empty())
                         .vortex_expect("failed to create execution context")

vortex-cuda/benches/for_cuda.rs

@@ -83,7 +83,7 @@ where
             |b, for_array| {
                 b.iter_custom(|iters| {
                     let timed = TimedLaunchStrategy::default();
-                    let timer = Arc::clone(&timed.total_time_ns);
+                    let timer = timed.timer();
 
                     let mut cuda_ctx = CudaSession::create_execution_ctx(&VortexSession::empty())
                         .vortex_expect("failed to create execution context")
@@ -121,7 +121,7 @@ where
             |b, for_array| {
                 b.iter_custom(|iters| {
                     let timed = TimedLaunchStrategy::default();
-                    let timer = Arc::clone(&timed.total_time_ns);
+                    let timer = timed.timer();
 
                     let mut cuda_ctx = CudaSession::create_execution_ctx(&VortexSession::empty())
                         .vortex_expect("failed to create execution context")

vortex-cuda/benches/runend_cuda.rs

@@ -77,7 +77,7 @@ where
                 |b, runend_array| {
                     b.iter_custom(|iters| {
                         let timed = TimedLaunchStrategy::default();
-                        let timer = Arc::clone(&timed.total_time_ns);
+                        let timer = timed.timer();
 
                         let mut cuda_ctx =
                             CudaSession::create_execution_ctx(&VortexSession::empty())