[Enhance] Add a comment for the use of masked_conv on Ascend device (#2689)

* Update masked_conv.py

* Update mmcv/ops/masked_conv.py

---------

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
pull/2692/merge
zcc9707 2023-03-22 20:37:15 +08:00 committed by GitHub
parent 1de3aeffd7
commit e6f434c440
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions

View File

@ -46,6 +46,8 @@ class MaskedConv2dFunction(Function):
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,