From af2479568820f1d61407d6ba7d71d56015d65633 Mon Sep 17 00:00:00 2001 From: Guoqing Bao Date: Mon, 27 Jan 2025 10:11:20 +0000 Subject: [PATCH] Do not perform inplace sort --- candle-core/src/sort.rs | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/candle-core/src/sort.rs b/candle-core/src/sort.rs index df970c4592..73a2e10418 100644 --- a/candle-core/src/sort.rs +++ b/candle-core/src/sort.rs @@ -328,9 +328,8 @@ impl Tensor { op: "sort_last_dim", }); } - let asort = self.arg_sort_last_dim(asc)?; - // let sorted = self.gather(&asort, crate::D::Minus1)?; - let sorted = self.clone(); + let sorted = self.copy()?; + let asort = sorted.arg_sort_last_dim(asc)?; Ok((sorted, asort)) } }