Fix TODO parallel comment

tracel-ai · Mar 18, 2024 · 12c0f67 · 12c0f67
1 parent 732c47e
commit 12c0f67
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/crates/burn-tensor/src/tensor/api/sort.rs b/crates/burn-tensor/src/tensor/api/sort.rs
@@ -302,8 +302,9 @@ fn sort_slice<B: Backend, const D: usize, K: BasicOps<B>>(
         .map(|(_, d)| d)
         .product();
 
-    // TODO:     run_par!(|| {
-    // iter_par!(output.axis_iter_mut(Axis(0)))
+    // TODO: run each sort in parallel
+    // run_par!(|| {
+    //     iter_range_par!(0, num_sorts).for_each(|id| {...})
     for id in 0..num_sorts {
         let mut index_offset = 0;
         let mut stride_dim = 0;