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