mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix lint

pull/3266/head
hust17yixuan 2025-03-24 15:08:18 +08:00
parent e5633468b6
commit 6d6db50549
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mmcv/ops/fused_bias_leakyrelu.py
View File

@ -258,7 +258,8 @@ def fused_bias_leakyrelu(input: torch.Tensor,
torch.Tensor: Feature map after non-linear activation.
"""
if (not input.is_cuda) and (not input.is_musa) and input.device.type != 'npu':
if (not input.is_cuda) and (not input.is_musa) and \
input.device.type != 'npu':
return bias_leakyrelu_ref(input, bias, negative_slope, scale)
return FusedBiasLeakyReLUFunction.apply(input, bias.to(input.dtype),