[fbsync] Fix segfault in DeformConv2d when mask is None (#7632)

Reviewed By: vmoens Differential Revision: D46314043 fbshipit-source-id: db228b42bca16c802989df152f89321d6047ce76
pytorch · May 31, 2023 · 2da5ebd · 2da5ebd
1 parent b8ec1ad
commit 2da5ebd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchvision/ops/deform_conv.py b/torchvision/ops/deform_conv.py
@@ -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)