KEMBAR78
[CUDA] Fix missing `__syncthreads` in MultiMarginLoss backward by eqy · Pull Request #158994 · pytorch/pytorch · GitHub
Skip to content

Conversation

@eqy
Copy link
Collaborator

@eqy eqy commented Jul 23, 2025

Turns out issue in #158921 is detectable with a simple unit test and adding the missing sync fixes it

cc @ptrblck @msaroufim @jerryzh168

@eqy eqy requested a review from syed-ahmed as a code owner July 23, 2025 23:43
@eqy eqy added module: loss Problem is related to loss function module: cuda Related to torch.cuda, and CUDA support in general labels Jul 23, 2025
@pytorch-bot
Copy link

pytorch-bot bot commented Jul 23, 2025

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/158994

Note: Links to docs will display an error until the docs builds have been completed.

✅ You can merge normally! (1 Unrelated Failure)

As of commit 8ae0666 with merge base 4d5d56a (image):

UNSTABLE - The following job is marked as unstable, possibly due to flakiness on trunk:

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@malfet malfet added the topic: bug fixes topic category label Jul 24, 2025
Co-authored-by: Nikita Shulga <2453524+malfet@users.noreply.github.com>
@eqy
Copy link
Collaborator Author

eqy commented Jul 24, 2025

@pytorchmergebot merge

@pytorch-bot pytorch-bot bot added the ciflow/trunk Trigger trunk jobs on your pull request label Jul 24, 2025
@pytorchmergebot
Copy link
Collaborator

Merge started

Your change will be merged once all checks pass (ETA 0-4 Hours).

Learn more about merging in the wiki.

Questions? Feedback? Please reach out to the PyTorch DevX Team

Advanced Debugging
Check the merge workflow status
here

yangw-dev pushed a commit that referenced this pull request Aug 1, 2025
Turns out issue in #158921 is detectable with a simple unit test and adding the missing sync fixes it

Pull Request resolved: #158994
Approved by: https://github.com/malfet, https://github.com/Skylion007

Co-authored-by: Nikita Shulga <2453524+malfet@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/trunk Trigger trunk jobs on your pull request Merged module: cuda Related to torch.cuda, and CUDA support in general module: loss Problem is related to loss function open source release notes: nn release notes category topic: bug fixes topic category

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants