lint1

open-mmlab · Dec 2, 2024 · 9a6be87 · 9a6be87
1 parent 007ba98
commit 9a6be87
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/mmcv/ops/csrc/pytorch/npu/assign_score_withk_npu.cpp b/mmcv/ops/csrc/pytorch/npu/assign_score_withk_npu.cpp
@@ -7,10 +7,10 @@ void assign_score_withk_forward_npu(int B, int N0, int N1, int M, int K, int O,
                                     int aggregate, const Tensor& points,
                                     const Tensor& centers, const Tensor& scores,
                                     const Tensor& knn_idx, Tensor& output) {
-at::Tensor points_trans = points.permute({0, 3, 1, 2});
-at::Tensor centers_trans = centers.permute({0, 3, 1, 2});
-EXEC_NPU_CMD(aclnnAssignScoreWithk, points_trans, centers_trans, scores,
-            knn_idx, B, N0, N1, M, K, O, aggregate, output);
+  at::Tensor points_trans = points.permute({0, 3, 1, 2});
+  at::Tensor centers_trans = centers.permute({0, 3, 1, 2});
+  EXEC_NPU_CMD(aclnnAssignScoreWithk, points_trans, centers_trans, scores,
+               knn_idx, B, N0, N1, M, K, O, aggregate, output);
 }
 
 void assign_score_withk_forward_impl(int B, int N0, int N1, int M, int K, int O,
@@ -20,4 +20,4 @@ void assign_score_withk_forward_impl(int B, int N0, int N1, int M, int K, int O,
                                      const Tensor& knn_idx, Tensor& output);
 
 REGISTER_NPU_IMPL(assign_score_withk_forward_impl,
-                assign_score_withk_forward_npu);
+                  assign_score_withk_forward_npu);