Skip to content

support eager mode with torch.compile#7256

Merged
JackCaoG merged 4 commits intomasterfrom
JackCaoG/eager_dynamo
Jun 13, 2024
Merged

support eager mode with torch.compile#7256
JackCaoG merged 4 commits intomasterfrom
JackCaoG/eager_dynamo

Conversation

@JackCaoG
Copy link
Copy Markdown
Collaborator

No description provided.

@JackCaoG JackCaoG requested review from lsy323, qihqi and wonjoo-wj June 12, 2024 18:52
@JackCaoG JackCaoG merged commit 192dec6 into master Jun 13, 2024
@JackCaoG JackCaoG deleted the JackCaoG/eager_dynamo branch June 13, 2024 18:08
Comment thread test/run_tests.sh
run_save_tensor_hlo "$CDIR/spmd/test_spmd_graph_dump.py"
}

# DO NOT MODIFY
Copy link
Copy Markdown
Collaborator

@vanbasten23 vanbasten23 Jun 13, 2024

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think Yeounoh added the comment because GPU test run_xla_op_tests2 is running so long that all the new tests should be added to run_xla_op_tests3 or later test. This is to reduce the overall CPU/GPU tests running time.

Copy link
Copy Markdown
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

right, but I think you fixed the AMP taking too long issue, now this test suite actually finished very quickly

@JackCaoG JackCaoG added the eager PyTorch/XLA eager-mode label Jun 20, 2024
JackCaoG added a commit that referenced this pull request Jul 9, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

eager PyTorch/XLA eager-mode

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants