Skip to content

Conversation

tushar00jain
Copy link
Contributor

@tushar00jain tushar00jain commented Aug 4, 2025

Summary:

  • improve abstraction and api contract offered by managed work - users can change future return types, ensures users use correct cuda streams
  • added a managed future that is lazily attaches callbacks - based on when users call wait on the work or the future

Stack created with Sapling. Best reviewed with ReviewStack.

@meta-cla meta-cla bot added the CLA Signed This label is managed by the Meta Open Source bot. label Aug 4, 2025
@tushar00jain tushar00jain marked this pull request as draft August 4, 2025 22:08
@tushar00jain tushar00jain force-pushed the pr253 branch 3 times, most recently from 2915aa5 to 4870513 Compare August 5, 2025 17:35
@tushar00jain tushar00jain marked this pull request as ready for review August 5, 2025 18:04
@tushar00jain tushar00jain force-pushed the pr253 branch 3 times, most recently from b72ddcc to eb099b5 Compare August 6, 2025 20:16
@tushar00jain tushar00jain mentioned this pull request Aug 6, 2025
@tushar00jain tushar00jain changed the title make managed futures generic clean up managed work Aug 7, 2025
@tushar00jain tushar00jain requested a review from d4l3k August 7, 2025 20:17
Copy link
Member

@d4l3k d4l3k left a comment

Choose a reason for hiding this comment

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

LGTM thanks for cleaning this up!

Summary:
- improve abstraction and api contract offered by  managed work - users can change future return types, ensures users use correct cuda streams
- added a managed future that is lazily attaches callbacks - based on when users call wait on the work  or the future
@tushar00jain tushar00jain merged commit 855bcad into pytorch:main Aug 14, 2025
14 of 15 checks passed
@tushar00jain tushar00jain deleted the pr253 branch August 14, 2025 06:02
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
CLA Signed This label is managed by the Meta Open Source bot.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants