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

[SPMD] Missing DT support NotImplementedError: Operator aten.amax.default does not have a DistributedTensor rule registered.

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

This error is thrown when using torch.nn.CrossEntropyLoss() with SPMD API.

anj-s avatar Feb 16 '23 19:02 anj-s