xla
Update the test to use the more standard torch.compile api
#4634

Merged

Update the test to use the more standard torch.compile api #4634

JackCaoG merged 3 commits into master from JackCaoG/test_torch_compile_usage

Move to the new torch.compile api

d17af75b

use torch.compile for inference test

26791dc2

Use torch.compile for training too

7cbf5080

JackCaoG added dynamo

JackCaoG requested a review from

alanwaketan 2 years ago

JackCaoG requested a review from

wonjoo-wj 2 years ago

JackCaoG requested a review from

shunting314 2 years ago

JackCaoG changed the title ~~Jack cao g/test torch compile usage~~ Update the test to use the more standard torch.compile api 2 years ago

wonjoo-wj approved these changes on 2023-02-16

alanwaketan approved these changes on 2023-02-16

JackCaoG merged b862368c into master 2 years ago

Reviewers

alanwaketan

wonjoo-wj

shunting314

Assignees

No one assigned

Labels

dynamo

Milestone

No milestone