xla
model should be on xla device for Dynamo torchxla_trace_once backend
#4205

Merged

model should be on xla device for Dynamo torchxla_trace_once backend #4205

JackCaoG merged 4 commits into master from jackcao/dynamo_test_on_xla_device

JackCaoG requested a review from

shunting314 3 years ago

JackCaoG added dynamo

shunting314 approved these changes on 2022-11-16

model should be on xla device for Dynamo torchxla_trace_once backend

29d008e1

torch pin

da4b5c84

Add _clear_pending_irs API and tests

50855d7b

JackCaoG force pushed from 09e7bd90 to 50855d7b 3 years ago

JackCaoG added REMOVE_TORCH_PIN

Delete .torch_pin

6f2475b2

JackCaoG merged 36e2acab into master 3 years ago

Reviewers

shunting314

Assignees

No one assigned

Labels

REMOVE_TORCH_PIN dynamo

Milestone

No milestone