Add dim checks for internal embedding_bag functions

[kurtamohler]

Fixes #85213

[pytorch-bot]

🔗 Helpful Links

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

• 📄 Preview Python docs built from this PR
• 📄 Preview C++ docs built from this PR
• ❓ Need help or want to give feedback on the CI? Visit the bot commands wiki or our office hours

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

❌ 1 Failures

As of commit e48ffff:

NEW FAILURES - The following jobs have failed:

• linux-focal-py3.7-clang7-asan / test (default, 1, 5, linux.2xlarge)

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

[facebook-github-bot]

/easycla

As part of the transition to the PyTorch Foundation, this project now requires contributions be covered under the new CLA. See #85559 for additional details.

This comment will trigger a new check of this PR. If you are already covered, you will simply see a new "EasyCLA" check that passes. If you are not covered, a bot will leave a new comment with a link to sign.

[linux-foundation-easycla]

The committers listed above are authorized under a signed CLA.

• ✅ login: kurtamohler (2a9e168)
• ✅ login: malfet / name: Nikita Shulga (20ecb68)

[malfet]

@pytorchbot rebase

[pytorchmergebot]

@pytorchbot successfully started a rebase job. Check the current status here

[pytorchmergebot]

Rebase failed due to Command git -C /home/runner/work/pytorch/pytorch rebase refs/remotes/origin/viable/strict pull/85433/head returned non-zero exit code 1

Rebasing (1/16)
Auto-merging aten/src/ATen/native/EmbeddingBag.cpp
Auto-merging test/test_nn.py
CONFLICT (content): Merge conflict in test/test_nn.py
error: could not apply 2a9e168e7f... Add dim checks for internal `embedding_bag` functions
hint: Resolve all conflicts manually, mark them as resolved with
hint: "git add/rm <conflicted_files>", then run "git rebase --continue".
hint: You can instead skip this commit: run "git rebase --skip".
hint: To abort and get back to the state before "git rebase", run "git rebase --abort".
Could not apply 2a9e168e7f... Add dim checks for internal `embedding_bag` functions

Raised by https://github.com/pytorch/pytorch/actions/runs/3568393500

[malfet]

@pytorchbot rebase

[pytorchmergebot]

@pytorchbot successfully started a rebase job. Check the current status here

[pytorchmergebot]

Rebase failed due to Command git -C /home/runner/work/pytorch/pytorch rebase refs/remotes/origin/viable/strict pull/85433/head returned non-zero exit code 1

Rebasing (1/7)
Auto-merging aten/src/ATen/native/EmbeddingBag.cpp
Auto-merging test/test_nn.py
CONFLICT (content): Merge conflict in test/test_nn.py
error: could not apply 2a9e168e7f... Add dim checks for internal `embedding_bag` functions
hint: Resolve all conflicts manually, mark them as resolved with
hint: "git add/rm <conflicted_files>", then run "git rebase --continue".
hint: You can instead skip this commit: run "git rebase --skip".
hint: To abort and get back to the state before "git rebase", run "git rebase --abort".
Could not apply 2a9e168e7f... Add dim checks for internal `embedding_bag` functions

Raised by https://github.com/pytorch/pytorch/actions/runs/3569126478

[kurtamohler]

The xla test failure is upstream

[kurtamohler]

@pytorchbot merge -f "unrelated failure"

[pytorchmergebot]

Merge started

Your change will be merged immediately since you used the force (-f) flag, bypassing any CI checks (ETA: 1-5 minutes).

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