Skip to content

Conversation

@DN6
Copy link
Collaborator

@DN6 DN6 commented Jun 26, 2024

What does this PR do?

Compile tests originally only ran on Python 3.9, but since our default is 3.10, they're running inside the other test runner jobs in addition to their dedicated runner.

This PR introduces an is_torch_compile decorator similar to slow for the compile tests. The compile tests will only run when a RUN_COMPILE env variable is set. This ensures that the compile tests run in their own module.

Also removes the requires_python39 higher check as it's no longer needed.

Fixes # (issue)

Before submitting

Who can review?

Anyone in the community is free to review the PR once the tests have passed. Feel free to tag
members/contributors who may be interested in your PR.

@DN6 DN6 requested a review from sayakpaul June 26, 2024 05:13
@HuggingFaceDocBuilderDev

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

Copy link
Member

@sayakpaul sayakpaul left a comment

Choose a reason for hiding this comment

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

Thanks! Could you also confirm if the compile tests are running successfully on our T4 CI runner? Related: pytorch/pytorch#129351.

@DN6
Copy link
Collaborator Author

DN6 commented Jun 26, 2024

They're not unfortunately. Will look into that issue separately.

@DN6 DN6 merged commit 0f0b531 into main Jun 26, 2024
@sayakpaul sayakpaul deleted the compile-tests branch June 26, 2024 05:57
sayakpaul added a commit that referenced this pull request Dec 23, 2024
* update

* update

---------

Co-authored-by: Sayak Paul <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants