From e1aab12f9be103e228b1c36203ba86082164f254 Mon Sep 17 00:00:00 2001 From: huangyuan64 Date: Mon, 18 Nov 2024 13:41:44 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0knn=E5=92=8Cthree=5Fnn?= =?UTF-8?q?=E7=9A=84NPU=E9=80=82=E9=85=8D=E4=BB=A3=E7=A0=81=20(#3194)?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- mmcv/ops/csrc/pytorch/npu/knn_npu.cpp | 2 +- mmcv/ops/csrc/pytorch/npu/three_nn_npu.cpp | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/mmcv/ops/csrc/pytorch/npu/knn_npu.cpp b/mmcv/ops/csrc/pytorch/npu/knn_npu.cpp index f25f9cf623..bfbf45d67e 100644 --- a/mmcv/ops/csrc/pytorch/npu/knn_npu.cpp +++ b/mmcv/ops/csrc/pytorch/npu/knn_npu.cpp @@ -12,7 +12,7 @@ void knn_forward_npu(int b, int n, int m, int nsample, const Tensor xyz, at::Tensor target = new_xyz.contiguous(); bool is_from_knn = true; - EXEC_NPU_CMD(aclnnKnn, source, target, nsample, is_from_knn, idx, dist2); + EXEC_NPU_CMD(aclnnKnn, source, target, is_from_knn, nsample, dist2, idx); } void knn_forward_impl(int b, int n, int m, int nsample, const Tensor xyz, diff --git a/mmcv/ops/csrc/pytorch/npu/three_nn_npu.cpp b/mmcv/ops/csrc/pytorch/npu/three_nn_npu.cpp index 9766816f6c..60e80544df 100644 --- a/mmcv/ops/csrc/pytorch/npu/three_nn_npu.cpp +++ b/mmcv/ops/csrc/pytorch/npu/three_nn_npu.cpp @@ -18,7 +18,7 @@ void three_nn_forward_npu(int b, int n, int m, const Tensor unknown, bool is_from_knn = false; uint32_t nsample = 3; - EXEC_NPU_CMD(aclnnKnn, source, target, nsample, is_from_knn, idx, dist2); + EXEC_NPU_CMD(aclnnKnn, source, target, is_from_knn, nsample, dist2, idx); if (originDtype == at::kHalf) { dist2 = dist2.to(at::kHalf); }