PiPPy
PiPPy copied to clipboard
[spmd] Add support for testing backward() for dtensor ops
Add support for backward() in test_dtensor_ops.py since that will cover FW + BW.