KEMBAR78
small bamba test for parallel execution by lucaslie · Pull Request #142 · nv-auto-deploy/TensorRT-LLM · GitHub
Skip to content

Conversation

@lucaslie
Copy link
Collaborator

@lucaslie lucaslie commented Sep 25, 2025

minituarized bamba test to stabilize parallel test execution

Signed-off-by: Lucas Liebenwein <11156568+lucaslie@users.noreply.github.com>
@lucaslie lucaslie self-assigned this Sep 25, 2025
@Copilot Copilot AI review requested due to automatic review settings September 25, 2025 22:19
Copy link

Copilot AI left a comment

Choose a reason for hiding this comment

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

Pull Request Overview

This PR adds a small test configuration for the Bamba model to enable parallel test execution. The changes streamline the test by removing commented code, adding a small model configuration option, and restructuring the test logic.

  • Adds small model configuration support for faster test execution
  • Removes commented pytest parametrization and debug code
  • Refactors test structure to use configurable model sizes

Reviewed Changes

Copilot reviewed 2 out of 2 changed files in this pull request and generated 2 comments.

File Description
tests/unittest/_torch/auto_deploy/unit/singlegpu/models/test_bamba.py Refactors test to support small model config and removes commented code
tests/unittest/_torch/auto_deploy/_utils_test/_model_test_utils.py Adds small model configuration for Bamba-9B-v2 model

Tip: Customize your code reviews with copilot-instructions.md. Create the file or learn how to get started.

@lucaslie lucaslie merged commit 066bfe9 into feat/ad_linear_attention Sep 25, 2025
2 of 5 checks passed
lucaslie added a commit that referenced this pull request Sep 29, 2025
Signed-off-by: Lucas Liebenwein <11156568+lucaslie@users.noreply.github.com>
nvchenghaoz pushed a commit that referenced this pull request Oct 1, 2025
Signed-off-by: Lucas Liebenwein <11156568+lucaslie@users.noreply.github.com>
nvchenghaoz pushed a commit that referenced this pull request Oct 3, 2025
Signed-off-by: Lucas Liebenwein <11156568+lucaslie@users.noreply.github.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.

1 participant