PiPPy icon indicating copy to clipboard operation
PiPPy copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

[SPMD][Fusion] Add unit tests for fusion

Open lessw2020 opened this issue 3 years ago • 0 comments

After PR https://github.com/pytorch/tau/pull/631 lands, add unit testing.

Simple tests would involved fusion based on a set policy and verifying output gradients and inspecting the graph.

lessw2020 avatar Nov 29 '22 01:11 lessw2020