fix lint

open-mmlab · Jun 8, 2023 · 8011be9 · 8011be9
1 parent ecdb7e1
commit 8011be9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mmcv/ops/csrc/pytorch/npu/roi_pool_npu.cpp b/mmcv/ops/csrc/pytorch/npu/roi_pool_npu.cpp
@@ -25,6 +25,7 @@ void roi_pool_forward_npu(Tensor input, Tensor rois, Tensor output,
         .Attr("spatial_scale_w", spatial_scale)
         .Attr("pool_channel", pooled_channel)
         .Run();
+
   } else {
     OpCommand cmd;
     cmd.Name("RoiPoolingWithArgMax")
@@ -38,7 +39,7 @@ void roi_pool_forward_npu(Tensor input, Tensor rois, Tensor output,
         .Attr("spatial_scale_h", spatial_scale)
         .Attr("spatial_scale_w", spatial_scale)
         .Attr("pool_channel", pooled_channel)
-        .Attr("_exclude_engines", (string)"AiCore")
+        .Attr("_exclude_engines", (string) "AiCore")
         .Run();
   }