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

Merged

Commits

Move to the new torch.compile api

JackCaoG committed 3 years ago
use torch.compile for inference test

JackCaoG committed 3 years ago
Use torch.compile for training too

JackCaoG committed 3 years ago

Loading