From 12c0f67d0d2d7f653eff966d9ad018c3eb468144 Mon Sep 17 00:00:00 2001
From: Guillaume Lagrange <lagrange.guillaume.1@gmail.com>
Date: Mon, 18 Mar 2024 13:43:11 -0400
Subject: [PATCH] Fix TODO parallel comment

---
 crates/burn-tensor/src/tensor/api/sort.rs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/crates/burn-tensor/src/tensor/api/sort.rs b/crates/burn-tensor/src/tensor/api/sort.rs
index 08306c10b3..115240e89e 100644
--- 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;