From e32db8dd7b3da937db7c74f08e31198b6d7a7910 Mon Sep 17 00:00:00 2001 From: Andy Grove Date: Fri, 20 Dec 2024 16:58:43 -0700 Subject: [PATCH] fix --- native/core/benches/batch_serde.rs | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/native/core/benches/batch_serde.rs b/native/core/benches/batch_serde.rs index 436c30435..063d7b4f0 100644 --- a/native/core/benches/batch_serde.rs +++ b/native/core/benches/batch_serde.rs @@ -17,9 +17,7 @@ use arrow_array::{builder::StringBuilder, ArrayRef, RecordBatch}; use arrow_schema::{DataType, Field, Schema}; -use comet::execution::shuffle::batch_serde::{ - read_batch_fast, read_batch_ipc, write_batch_fast, write_batch_ipc, -}; +use comet::execution::shuffle::batch_serde::{read_batch_fast, read_batch_ipc, write_batch_fast, write_batch_ipc, write_batch_ipc_data_only}; use criterion::{criterion_group, criterion_main, Criterion}; use std::sync::Arc; @@ -72,7 +70,7 @@ fn criterion_benchmark(c: &mut Criterion) { let mut buffer = vec![]; b.iter(|| { buffer.clear(); - criterion::black_box(write_batch_ipc(&batch, &mut buffer).unwrap()) + criterion::black_box(write_batch_ipc_data_only(&batch, &mut buffer).unwrap()) }); }); }