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 to enable torch.compile with Deepspeed #1171

Closed
wants to merge 1 commit into from

Conversation

vivekgoe
Copy link
Collaborator

@vivekgoe vivekgoe commented Jul 30, 2024

Update prepare_deepspeed() to support new t.compile API (refer to deepspeedai/DeepSpeed#5581) in Deepspeed

What does this PR do?

Fixes # (issue)

Before submitting

  • This PR fixes a typo or improves the docs (you can dismiss the other checks if that's the case).
  • Did you make sure to update the documentation with your changes?
  • Did you write any new necessary tests?

* Update prepare_deepspeed() to support new t.compile API in Deepspeed

* Update condition

* Update condition
@vivekgoe vivekgoe requested a review from regisss as a code owner July 30, 2024 04:12
@vivekgoe vivekgoe requested review from libinta and a user July 30, 2024 04:12
@vivekgoe vivekgoe added the synapse 1.17_dependency PR not backward compatible can be merged only when synapse 1.17 is available. label Jul 30, 2024
@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.

yafshar
yafshar approved these changes Jul 31, 2024
vidyasiv pushed a commit to emascarenhas/optimum-habana that referenced this pull request Aug 1, 2024
Add support to enable torch.compile with Deepspeed huggingface#1171
vidyasiv added a commit to emascarenhas/optimum-habana that referenced this pull request Aug 2, 2024
Add support to enable torch.compile with Deepspeed huggingface#1171
@regisss
Copy link
Collaborator

regisss commented Aug 7, 2024

Closing as it was integrated into #1163.

@regisss regisss closed this Aug 7, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
synapse 1.17_dependency PR not backward compatible can be merged only when synapse 1.17 is available.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants