PR #4294 [LTC] Reuse some methods from LazyTensor

[LTC] Reuse some methods from LazyTensor #4294

alanwaketan merged 14 commits into master from alanwaketan/xla_tensor

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

wonjoo-wj commented on 2022-12-07

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

JackCaoG commented on 2022-12-07

JackCaoG approved these changes on 2022-12-07

Rename Data.xla_data to handle

0ffae4a5

Let XLATensor::Data inherits torch::lazy::LazyTensor::Data

cf872d94

Remove somee duplicated methods as the base class

af146d77

Make data() virtual

fe5e4e22

Rename CurrentXlaData to CurrentDataHandle

95ef5069

Change data() to return shared_ptr

d6feb445

Replace data_ptr() with data()

bed594ba

Remove GetNextTensorId

50707603

Add .torch_pin

d5afdb79

Fix a typo

15d98711

Replace TORCH_CHECK with XLA_CHECK

864b2e70

Fix the build

ccf493e2

Fix the build

b55aec84

alanwaketan force pushed from 7729ed0e to b55aec84 2 years ago

Remove .torch_pin

335df211

alanwaketan merged 16607266 into master 2 years ago

Reviewers

JackCaoG

wonjoo-wj

Assignees

alanwaketan

Labels

tracing

Milestone

No milestone

xla [LTC] Reuse some methods from LazyTensor #4294 Merged

[LTC] Reuse some methods from LazyTensor #4294

xla
[LTC] Reuse some methods from LazyTensor
#4294

Merged