pytorch
a52d4ddc
- [LT] Fix a bug in lazy_model.mark_step()
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
[LT] Fix a bug in lazy_model.mark_step() Summary: lazy_tensor_core._LAZYC._ltc_step_marker accepts a device string instead of torch.device. Fix that by wrapping torch.device with str(). Test Plan: Manual test...
References
#73000 - [LT] Fix a bug in lazy_model.mark_step()
Author
Jiewen Tan
Committer
desertfire
Parents
8aabaa5e
Loading