[DTensor] Assert shard dim is less than tensor ndim #112404
Conversation
joshlk
requested review from
H-Huang,
awgu,
d4l3k,
fduwjj,
fegin,
kiukchung,
kwen2501,
mrshenli,
rohan-varma,
wanchaol,
wz337 and
zhaojuanmao
as code owners
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/112404
Note: Links to docs will display an error until the docs builds have been completed. ❌ 1 New Failure, 1 Unrelated FailureAs of commit e7a2d86 with merge base d6724a5 ( NEW FAILURE - The following job has failed:
BROKEN TRUNK - The following job failed but were present on the merge base:👉 Rebase onto the `viable/strict` branch to avoid these failures
This comment was automatically generated by Dr. CI and updates every 15 minutes. |
|
CLA issue has now been resolved. it should be ready to merge |
|
@pytorchbot merge -i |
Merge startedYour change will be merged while ignoring the following 2 checks: pull / linux-focal-py3-clang9-android-ndk-r21e-gradle-custom-build-single / build-and-test (default, 1, 1, linux.2xlarge), pull / linux-focal-py3-clang9-android-ndk-r21e-gradle-custom-build-single-full-jit / build-and-test (default, 1, 1, linux.2xlarge) Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
Assert shard dim is less than tensor ndim. Previously, an index error on line 154 is thrown and the error is not clear Pull Request resolved: pytorch#112404 Approved by: https://github.com/wanchaol, https://github.com/fduwjj
Assert shard dim is less than tensor ndim. Previously, an index error on line 154 is thrown and the error is not clear Pull Request resolved: pytorch#112404 Approved by: https://github.com/wanchaol, https://github.com/fduwjj
5 participants
Assert shard dim is less than tensor ndim. Previously, an index error on line 154 is thrown and the error is not clear
cc: @wanchaol