xla
b826def1 - [LTC] Reuse more upstream LazyGraphExecutor data structures (#4312)

Commit
2 years ago
[LTC] Reuse more upstream LazyGraphExecutor data structures (#4312) Summary: This patch reuses more upstream LazyGraphExecutor data structures, including: 1. TlsData. 2. DeviceLocker. 3. DeviceLockerArena. 4. DataCacheArena. And also reuses the following methods: 1. DeviceBarrier(). 2. GetDeviceData(). 3. IncTrimCounter(). P.S. It needs pytorch/pytorch#90598 to work. And it tries to re-land #4303. Test Plan: CI.
Author
Parents
Loading