Skip to content

[tests] Changes to the torch.compile() CI and tests #11508

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Open
wants to merge 10 commits into
base: main
Choose a base branch
from
Open

[tests] Changes to the torch.compile() CI and tests #11508

wants to merge 10 commits into from

Conversation

sayakpaul
Copy link
Member

What does this PR do?

  • Removes the CUDA compile docker container
  • Consistency in how compilation cache is cleared and ensuring we start a compilation run with fresh cache
  • Removes unnecessary compilation tests from pipelines

Regarding the final point, I think we already test torch.compile() support for popular models and doing it at the pipeline-level makes little sense to me. But no strong opinions.

@sayakpaul sayakpaul requested a review from DN6 May 6, 2025 10:06
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@DN6 DN6 Awaiting requested review from DN6

At least 1 approving review is required to merge this pull request.

Assignees
No one assigned
Labels
torch.compile
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
1 participant
@sayakpaul