pytorch
93aa6e3e - [LTC] Make some LazyGraphExecutor private data structures protected (#90457)

Commit
2 years ago
[LTC] Make some LazyGraphExecutor private data structures protected (#90457) Summary: This pull request makes some LazyGraphExecutor private data structures protected such that XLAGraphExecutor can reuse them. Here is the list: 1. DeviceLocker. 2. DeviceLockerArena. 3. DataCacheArena. In addition, it also introduces LazyGraphExecutor::ResetTrimCounter() such that XLAGraphExecutor can reuse the trim counter. Test Plan: CI. Pull Request resolved: https://github.com/pytorch/pytorch/pull/90457 Approved by: https://github.com/JackCaoG
Author
Committer
Parents
Loading