pytorch
a52d4ddc - [LT] Fix a bug in lazy_model.mark_step()

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...
Author
Jiewen Tan
Committer
Parents
Loading