From 942a1e4224065cc7cbab0ef7322fc1cb6d70e383 Mon Sep 17 00:00:00 2001 From: lvyufeng Date: Sun, 22 Mar 2026 11:20:30 +0800 Subject: [PATCH] fix(npu): enable native 310B argsort Remove argsort from 310B fallback list. The native aclnnTopk path (used internally by argsort) now 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..783e68f2 100644 --- a/src/candle/_backends/npu/ops_soc.py +++ b/src/candle/_backends/npu/ops_soc.py @@ -59,7 +59,6 @@ "softplus", # aclnnSoftplus returns 561103; mish composite depends on this "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