[Enhance] Add a comment for the use of masked_conv on Ascend device (o…

…pen-mmlab#2689) * Update masked_conv.py * Update mmcv/ops/masked_conv.py --------- Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
CokeDong · Apr 6, 2023 · 8304b43 · 8304b43
1 parent 5ee8aad
commit 8304b43
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mmcv/ops/masked_conv.py b/mmcv/ops/masked_conv.py
@@ -46,6 +46,8 @@ def forward(ctx,
         out_channel, in_channel, kernel_h, kernel_w = weight.size()
 
         if features.device.type == 'npu':
+            # For the NPU implementation of the masked_conv operator, it is
+            # not recommended to use the scenario where bias=false.
             import torch_npu
             output = torch_npu.npu_conv2d(
                 features,