"vscode:/vscode.git/clone" did not exist on "40f16872c749d5b5cbf26218b3bd33c6a6788582"
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(
use_mask = mask is not 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:
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