Skip to content
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.

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

Add support for model.compile in PyTorch 2.0 #3246

Merged
merged 15 commits into from
Mar 17, 2023
Merged

Add support for model.compile in PyTorch 2.0 #3246

merged 15 commits into from
Mar 17, 2023

Conversation

tgaddair
Copy link
Collaborator

trainer:
  compile: true

@tgaddair tgaddair added the release-0.7 Needs cherry-pick into 0.7 release branch label Mar 14, 2023
@github-actions
Copy link

github-actions bot commented Mar 14, 2023

Unit Test Results

    6 files  ±         0    6 suites  ±0   30m 57s ⏱️ - 6h 47m 54s
  62 tests  -   4 037  59 ✔️  -   3 996    3 💤  - 40  0  - 1 
102 runs   - 11 991  91 ✔️  - 11 896  11 💤  - 94  0  - 1 

Results for commit db6dc93. ± Comparison against base commit 0d1f4d8.

♻️ This comment has been updated with latest results.

Comment on lines 80 to 82
raise ConfigValidationError(
"ECD model training requires PyTorch 2.0.0 or higher. Please upgrade PyTorch and try again."
)
Copy link
Contributor

Choose a reason for hiding this comment

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

I would change this to compile requires PyTorch 2.0.0 or higher. Please upgrade PyTorch and try again

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

That's what I get for letting Copilot write the error messages.

@tgaddair tgaddair merged commit 60bdb85 into master Mar 17, 2023
@tgaddair tgaddair deleted the torch-compile branch March 17, 2023 04:04
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
release-0.7 Needs cherry-pick into 0.7 release branch
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants