File tree Expand file tree Collapse file tree 2 files changed +4
-3
lines changed
rust/sedona-spatial-join-gpu Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Original file line number Diff line number Diff line change @@ -253,8 +253,6 @@ impl GpuSpatialJoinStream {
253253 // Execute GPU join with build data
254254 match self . execute_gpu_join_with_build_data ( & build_data) {
255255 Ok ( ( ) ) => {
256- let total_result_rows: usize =
257- self . result_batches . iter ( ) . map ( |b| b. num_rows ( ) ) . sum ( ) ;
258256 log:: info!(
259257 "GPU join completed, produced {} result batches" ,
260258 self . result_batches. len( )
@@ -342,7 +340,6 @@ impl GpuSpatialJoinStream {
342340
343341 // Concatenate all right batches into one batch
344342 let _concat_timer = self . join_metrics . concat_time . timer ( ) ;
345- let concat_start = Instant :: now ( ) ;
346343 let right_batch = if self . right_batches . len ( ) == 1 {
347344 self . right_batches [ 0 ] . clone ( )
348345 } else {
Original file line number Diff line number Diff line change @@ -118,6 +118,7 @@ impl RecordBatchStream for MockStream {
118118 }
119119}
120120
121+ #[ cfg( feature = "gpu" ) ]
121122#[ tokio:: test]
122123async fn test_gpu_join_exec_creation ( ) {
123124 // Create simple mock execution plans as children
@@ -151,6 +152,7 @@ async fn test_gpu_join_exec_creation() {
151152 assert_eq ! ( gpu_join. children( ) . len( ) , 2 ) ;
152153}
153154
155+ #[ cfg( feature = "gpu" ) ]
154156#[ tokio:: test]
155157async fn test_gpu_join_exec_display ( ) {
156158 let left_plan = Arc :: new ( MockExec :: new ( ) ) as Arc < dyn ExecutionPlan > ;
@@ -181,6 +183,7 @@ async fn test_gpu_join_exec_display() {
181183 assert ! ( display_str. contains( "Inner" ) ) ;
182184}
183185
186+ #[ cfg( feature = "gpu" ) ]
184187#[ tokio:: test]
185188async fn test_gpu_join_execution_with_fallback ( ) {
186189 // This test should handle GPU not being available and fallback to CPU error
@@ -250,6 +253,7 @@ async fn test_gpu_join_execution_with_fallback() {
250253 ) ;
251254}
252255
256+ #[ cfg( feature = "gpu" ) ]
253257#[ tokio:: test]
254258async fn test_gpu_join_with_empty_input ( ) {
255259 // Test with empty batches (MockExec returns empty stream)
You can’t perform that action at this time.
0 commit comments