From 3a4109e79e407048115541fc642a606f9db5a0f5 Mon Sep 17 00:00:00 2001 From: lvyufeng Date: Sun, 22 Mar 2026 12:28:19 +0800 Subject: [PATCH] fix(npu): enable native 310B take_along_dim Remove take_along_dim from the 310B fallback list. The native aclnnGather path 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..e2c418a7 100644 --- a/src/candle/_backends/npu/ops_soc.py +++ b/src/candle/_backends/npu/ops_soc.py @@ -64,7 +64,6 @@ "topk", # aclnnTopk returns 561103 "diag", # aclnnDiag returns 561103 "gather", # aclnnGather returns 561103 - "take_along_dim", # aclnnGather returns 561103 "layer_norm", # aclnnLayerNorm returns 561103 for float32 (float16 works) "mish", # aclnnMish returns 561103 "batch_norm", # aclnnBatchNorm returns 161002