Skip to content

fix: support batch size > 1#58

Merged
akoumpa merged 9 commits intomainfrom
akoumparouli/fix_batch_unsqueezing_size
Nov 19, 2025
Merged

fix: support batch size > 1#58
akoumpa merged 9 commits intomainfrom
akoumparouli/fix_batch_unsqueezing_size

Conversation

@akoumpa
Copy link
Contributor

@akoumpa akoumpa commented Nov 18, 2025

No description provided.

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>
@copy-pr-bot
Copy link

copy-pr-bot bot commented Nov 18, 2025

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.

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>
@akoumpa
Copy link
Contributor Author

akoumpa commented Nov 18, 2025

/ok to test 0a1f4c0

linnanwang
linnanwang previously approved these changes Nov 18, 2025
Copy link
Contributor

@linnanwang linnanwang left a comment

Choose a reason for hiding this comment

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

LGTM

@akoumpa
Copy link
Contributor Author

akoumpa commented Nov 18, 2025

/ok to test 0c9a912

@akoumpa
Copy link
Contributor Author

akoumpa commented Nov 18, 2025

/ok to test 0c9a912

@copy-pr-bot
Copy link

copy-pr-bot bot commented Nov 18, 2025

/ok to test 0c9a912

@akoumpa, there was an error processing your request: E2

See the following link for more information: https://docs.gha-runners.nvidia.com/cpr/e/2/

@akoumpa
Copy link
Contributor Author

akoumpa commented Nov 18, 2025

/ok to test 00eae77

@akoumpa
Copy link
Contributor Author

akoumpa commented Nov 19, 2025

/ok to test f47b81e

@linnanwang
Copy link
Contributor

/ok to test d381b9a

Signed-off-by: linnan wang <wangnan318@gmail.com>
@linnanwang
Copy link
Contributor

/okay to test acd96c3

linnanwang
linnanwang previously approved these changes Nov 19, 2025
Copy link
Contributor

@linnanwang linnanwang left a comment

Choose a reason for hiding this comment

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

LGTM

Signed-off-by: linnan wang <wangnan318@gmail.com>
Signed-off-by: linnan wang <wangnan318@gmail.com>
@linnanwang
Copy link
Contributor

/ok to test ac21bde

Copy link
Contributor

@linnanwang linnanwang left a comment

Choose a reason for hiding this comment

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

LGTM

@akoumpa akoumpa merged commit fd56e93 into main Nov 19, 2025
15 checks passed
lbliii pushed a commit that referenced this pull request Nov 19, 2025
* replace torch.stack with torch.cat

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* fix

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

---------

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>
Signed-off-by: Lawrence Lane <llane@nvidia.com>
huvunvidia pushed a commit that referenced this pull request Feb 12, 2026
* replace torch.stack with torch.cat

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

* fix

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>

---------

Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants

Comments