PiPPy icon indicating copy to clipboard operation
PiPPy copied to clipboard
verified publisher icon pytorch

[SPMD] Add support for convolution ops to DTensor sharding prop

Open anj-s opened this issue 2 years ago • 0 comments

Currently MNIST benchmark fails due to unsupported convolution ops in the DTensor registry. Error: NotImplementedError: Operator aten.convolution.default does not have a DistributedTensor rule registered.

anj-s avatar Feb 14 '23 17:02 anj-s