xla
[LTC] Inherits LazyGraphExecutor::DeviceContextArena
#4307

Merged

[LTC] Inherits LazyGraphExecutor::DeviceContextArena #4307

alanwaketan merged 8 commits into master from alanwaketan/lazy_graph_3

alanwaketan added tracing

alanwaketan requested a review from

wonjoo-wj 2 years ago

alanwaketan requested a review from

JackCaoG 2 years ago

alanwaketan assigned

alanwaketan 2 years ago

alanwaketan force pushed from ef8b2fdc to 30cb3060 2 years ago

JackCaoG approved these changes on 2022-12-10

alanwaketan force pushed from 30cb3060 to a132b4c8 2 years ago

Use torch::lazy::Data inDeviceContext

cfff62bf

Make DeviceContextArena proctected

8918a3d8

Inherits torch::lazy::LazyGraphExecutor::DeviceContextArena

12f6c4be

Mark some overrided methods final

3e48ad6b

Fix linters

f4a550b6

Add .torch_pin

3db6a0cf

Add back ssome XLA counters

ad8696a0

alanwaketan force pushed from a132b4c8 to ad8696a0 2 years ago

wonjoo-wj approved these changes on 2022-12-12

Remove .torch_pin

cfefbf18

alanwaketan merged c570f7f7 into master 2 years ago

Reviewers

wonjoo-wj

JackCaoG

Assignees

alanwaketan

Labels

tracing

Milestone

No milestone