xla

b862368c - Update the test to use the more standard torch.compile api (#4634)

Update the test to use the more standard torch.compile api (#4634) * Move to the new torch.compile api * use torch.compile for inference test * Use torch.compile for training too