Skip to content

cp: fix: NotImplementedError: aten::equal on meta tensors during multi-GPU init (1769) into r0.4.0#1797

Merged
akoumpa merged 1 commit intor0.4.0from
cherry-pick-1769-r0.4.0
Apr 13, 2026
Merged

cp: fix: NotImplementedError: aten::equal on meta tensors during multi-GPU init (1769) into r0.4.0#1797
akoumpa merged 1 commit intor0.4.0from
cherry-pick-1769-r0.4.0

Conversation

@svcnvidia-nemo-ci
Copy link
Copy Markdown
Contributor

beep boop [🤖]: Hi @harshareddy832 👋,

we've cherry picked #1769 into  for you! 🚀

Please review and approve this cherry pick by your convenience!

…GPU init (#1769)

* fix: `notimplementederror: aten::equal` on meta tensors during multi-gpu model init with transformers >= 5.4.0 (#1765)

Signed-off-by: Harsha Pasham <pashamharsha018@gmail.com>

* fix: `notimplementederror: aten::equal` on meta tensors during multi-gpu model init with transformers >= 5.4.0 (#1765)

Signed-off-by: Harsha Pasham <pashamharsha018@gmail.com>

* Add pull_request types to workflow triggers

* Apply suggestion from @akoumpa

---------

Signed-off-by: Harsha Pasham <pashamharsha018@gmail.com>
Co-authored-by: Alexandros Koumparoulis <153118171+akoumpa@users.noreply.github.com>
Signed-off-by: NeMo Bot <nemo-bot@nvidia.com>
@svcnvidia-nemo-ci svcnvidia-nemo-ci added cherry-pick Run CICD Trigger Testing CICD labels Apr 13, 2026
@svcnvidia-nemo-ci
Copy link
Copy Markdown
Contributor Author

/ok to test 78237b2

@copy-pr-bot
Copy link
Copy Markdown

copy-pr-bot Bot commented Apr 13, 2026

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@akoumpa akoumpa changed the title cp: fix: NotImplementedError: aten::equal on meta tensors during multi-GPU init (1769) into r0.4.0 cp: fix: NotImplementedError: aten::equal on meta tensors during multi-GPU init (1769) into r0.4.0 Apr 13, 2026
@akoumpa akoumpa merged commit e3517d8 into r0.4.0 Apr 13, 2026
52 of 56 checks passed
@akoumpa akoumpa deleted the cherry-pick-1769-r0.4.0 branch April 13, 2026 16:23
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

cherry-pick Run CICD Trigger Testing CICD

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants