Closed feifei-111 closed 1 year ago
Others
support autodiff to accept modules from different devices. user should init their module on different devices by themselves
PR types
Others
PR changes
Others
Description
support autodiff to accept modules from different devices. user should init their module on different devices by themselves