fix lint

momo609 · Sep 14, 2023 · 4ad91ce · 4ad91ce
1 parent c4ca692
commit 4ad91ce
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/mmcv/ops/csrc/pytorch/npu/gather_points_npu.cpp b/mmcv/ops/csrc/pytorch/npu/gather_points_npu.cpp
@@ -22,8 +22,8 @@ void gather_points_forward_npu(int b, int c, int n, int npoints,
       .Run();
 }
 void gather_points_backward_npu(int b, int c, int n, int npoints,
-                                 const Tensor grad_out, const Tensor idx,
-                                 Tensor grad_points) {
+                                const Tensor grad_out, const Tensor idx,
+                                Tensor grad_points) {
   at::Tensor indices = idx;
   if (idx.scalar_type() != at::ScalarType::Int) {
     indices = idx.to(at::kInt);
@@ -35,10 +35,12 @@ void gather_points_backward_npu(int b, int c, int n, int npoints,
   at::SmallVector<int64_t, N> pad_size = array_to_small_vector(idx.sizes());
   at::Tensor trans_grad_points = grad_points.transpose(1, 2).contiguous();
   at::Tensor grad_points_view = trans_grad_points.view(
-    {trans_grad_points.sizes()[0] * trans_grad_points.sizes()[1], trans_grad_points.sizes()[2]});
+      {trans_grad_points.sizes()[0] * trans_grad_points.sizes()[1],
+       trans_grad_points.sizes()[2]});
   at::Tensor trans_grad_out = grad_out.transpose(1, 2).contiguous();
   trans_grad_out = trans_grad_out.view(
-    {trans_grad_out.sizes()[0] * trans_grad_out.sizes()[1], trans_grad_out.sizes()[2]});
+      {trans_grad_out.sizes()[0] * trans_grad_out.sizes()[1],
+       trans_grad_out.sizes()[2]});
   auto index = at::arange(0, b);
   index = index.to(grad_out.device());
   index = at::mul(index, n);
@@ -54,7 +56,8 @@ void gather_points_backward_npu(int b, int c, int n, int npoints,
       .Output(grad_points_view)
       .Attr("axis", dim)
       .Run();
-  at::Tensor grad_points_result = grad_points_view.view(trans_grad_points.sizes());
+  at::Tensor grad_points_result =
+      grad_points_view.view(trans_grad_points.sizes());
   grad_points_result = grad_points_result.transpose(1, 2);
   grad_points.copy_(grad_points_result);
 }