mirror of https://github.com/open-mmlab/mmcv.git
[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
parent
1de3aeffd7
commit
e6f434c440
|
@ -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,
|
||||
|
|
Loading…
Reference in New Issue