vision icon indicating copy to clipboard operation
vision copied to clipboard

deform_conv2d on MPS.

Open bukhalmae145 opened this issue 8 months ago • 1 comments

🚀 The feature

I want the deform_conv2d to be able to run on MPS(Apple Silicon) natively not falling back to CPU.

Motivation, pitch

I want to use this on MAC:

return torchvision.ops.deform_conv2d(x, offset, self.weight, self.bias, self.stride, self.padding, self.dilation, mask)

Alternatives

No response

Additional context

No response

bukhalmae145 avatar Mar 13 '25 06:03 bukhalmae145

Duplicate of deform_conv2d for mps #7490

mcioffi avatar Mar 27 '25 12:03 mcioffi