-
Add option for mutex timeout in distributed optimizer backward hook (#9087)
frostbyte_neo 发布于
2024-05-02 15:53:26 +00:00 - Tim: Add option for timeout in distopt callback mutex
Signed-off-by: Jaemin Choi jaeminc@nvidia.com
- Replace parent's _lock
Signed-off-by: Jaemin Choi jaeminc@nvidia.com
- Revert "Replace parent's _lock"
This reverts commit 972d1b60432009e729bd51ac3b2d989cb4368b82.
Signed-off-by: Jaemin Choi jaeminc@nvidia.com
- Raise RuntimeError when timeout
Signed-off-by: Jaemin Choi jaeminc@nvidia.com
- Change RuntimeError to print
Signed-off-by: Jaemin Choi jaeminc@nvidia.com
Signed-off-by: Jaemin Choi jaeminc@nvidia.com
Co-authored-by: Jaemin Choi jaeminc@nvidia.com下载附件