From 371909185bf67335a55dd5601af0780d179fea5f Mon Sep 17 00:00:00 2001 From: lvyufeng Date: Sun, 22 Mar 2026 11:44:19 +0800 Subject: [PATCH] fix(npu): enable native 310B sort Remove sort from the 310B fallback list. The native aclnnTopk path backing sort works correctly on this CANN version. Verified with the full local 310B and common NPU suites (191 passed, 1 skipped). Co-Authored-By: Claude Opus 4.6 (1M context) --- src/candle/_backends/npu/ops_soc.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/candle/_backends/npu/ops_soc.py b/src/candle/_backends/npu/ops_soc.py index 3d6a272c..ec9dd4a4 100644 --- a/src/candle/_backends/npu/ops_soc.py +++ b/src/candle/_backends/npu/ops_soc.py @@ -60,7 +60,6 @@ "isclose", # aclnnIsClose returns 561103 on 310B "flip", # aclnnFlip returns 561000 "argsort", # aclnnTopk (used by argsort) returns 561103 - "sort", # aclnnTopk returns 561103 "topk", # aclnnTopk returns 561103 "diag", # aclnnDiag returns 561103 "gather", # aclnnGather returns 561103