⚡️Tried to not split columns

carefree0910 · Oct 25, 2024 · c13aeee · c13aeee
1 parent 6d1c4a1
commit c13aeee
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cfpyo3_rs_core/src/io/temporal/mem.rs b/cfpyo3_rs_core/src/io/temporal/mem.rs
@@ -1174,7 +1174,8 @@ pub fn redis_grouped_column_contiguous<'a, T: AFloat>(
     let num_data_per_batch = num_columns * datetime_len as usize * nc as usize;
     let mut flattened = vec![T::zero(); bz * num_data_per_batch];
     let flattened_slice = flattened.as_mut_slice();
-    let num_columns_per_task = (num_columns / 4).max(10.min(num_columns));
+    // let num_columns_per_task = (num_columns / 4).max(10.min(num_columns));
+    let num_columns_per_task = num_columns;
     let num_columns_task = num_columns / num_columns_per_task;
     let num_tasks = bz * n_groups * num_columns_task;
     let num_threads = num_threads.min(num_tasks);