Unverified Commit e6f434c4 authored by zcc9707's avatar zcc9707 Committed by GitHub
Browse files

[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: default avatarZaida Zhou <58739961+zhouzaida@users.noreply.github.com>
parent 1de3aeff
...@@ -46,6 +46,8 @@ class MaskedConv2dFunction(Function): ...@@ -46,6 +46,8 @@ class MaskedConv2dFunction(Function):
out_channel, in_channel, kernel_h, kernel_w = weight.size() out_channel, in_channel, kernel_h, kernel_w = weight.size()
if features.device.type == 'npu': 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 import torch_npu
output = torch_npu.npu_conv2d( output = torch_npu.npu_conv2d(
features, features,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment