Unverified Commit 81d12c3d authored by Nicolas Hug's avatar Nicolas Hug Committed by GitHub
Browse files

Fix segfault in DeformConv2d when `mask` is None (#7632)

parent 4a51822c
...@@ -68,7 +68,7 @@ def deform_conv2d( ...@@ -68,7 +68,7 @@ def deform_conv2d(
use_mask = mask is not None use_mask = mask is not None
if mask is None: if mask is None:
mask = torch.zeros((input.shape[0], 0), device=input.device, dtype=input.dtype) mask = torch.zeros((input.shape[0], 1), device=input.device, dtype=input.dtype)
if bias is None: if bias is None:
bias = torch.zeros(out_channels, device=input.device, dtype=input.dtype) bias = torch.zeros(out_channels, device=input.device, dtype=input.dtype)
......
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