implement XLAHooks
and register it to PyTorch when loaded.
#15060
Job | Run time |
---|---|
15s | |
1m 30s | |
1m 45s |
XLAHooks
and register it to PyTorch when loaded.
#15060
Job | Run time |
---|---|
15s | |
1m 30s | |
1m 45s |