[https://nvbugs/6501376][fix] Test-only fix — drop the if hidden_size % 2 != 0: with pytest.raises(...)…

[trtllm-agent]

Summary

• Root cause: Stale pytest.raises(AssertionError) on the odd-hidden_size branch expected a divisibility assert that PR [TRTLLM-13117][feat] Implement Uneven TP Linear for VisualGen models #14875's _auto_tp_sharding no longer fires for VANILLA row/column Linear.
• Fix: Test-only fix — drop the if hidden_size % 2 != 0: with pytest.raises(...) branch in both test_row_linear and test_column_linear; unconditionally for r in results: assert r is True.
• Automated fix generated by repair-bot

Test plan

• Verify fix on the same GPU type as the original failure
• Check for regressions in related tests

Links

• Bug: https://nvbugs/6501376

Dev Engineer Review

• Updated test_column_linear and test_row_linear to reflect uneven tensor-parallel support introduced for VANILLA Linear modules.
• Removed stale pytest.raises(AssertionError) expectations for odd hidden_size values.
• Tests now consistently assert that all ranks return True; no API, configuration, or production-code changes were made.
• Verdict: sufficient.

QA Engineer Review

• Modified tests: test_column_linear, test_row_linear.
• No test-list changes were identified; test-list coverage should be verified in test-db/ or qa/.
• Verdict: needs follow-up.

[coderabbitai]

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info

⚙️ Run configuration

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Enterprise

Run ID: 8a29dc41-f1c6-4ee5-a50a-e62d27f896ee

📥 Commits

Reviewing files that changed from the base of the PR and between 021b435 and 77df102.

📒 Files selected for processing (1)

• tests/unittest/_torch/multi_gpu/test_linear.py

---

Walkthrough

The multi-GPU column and row linear tests now unconditionally require all returned results to be True, including for odd hidden_size values.

Changes

Linear test expectations

Layer / File(s)	Summary
Unconditional linear result assertions tests/unittest/_torch/multi_gpu/test_linear.py	test_column_linear and test_row_linear no longer expect AssertionError for odd hidden_size values and instead assert every result is True.

Estimated code review effort: 1 (Trivial) | ~2 minutes

Possibly related PRs

• NVIDIA/TensorRT-LLM#16840: Updates related odd hidden_size handling in the multi-GPU linear tests.

Suggested reviewers: lori-ren

🚥 Pre-merge checks | ✅ 5

✅ Passed checks (5 passed)

Check name	Status	Explanation
Title check	✅ Passed	The title matches the test-only bug fix and includes the required bug ID and fix type.
Description check	✅ Passed	The description has Summary, Test plan, and Links, which covers the template's main required sections.
Docstring Coverage	✅ Passed	No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check.
Linked Issues check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check	✅ Passed	Check skipped because no linked issues were found for this pull request.

✨ Finishing Touches

🧪 Generate unit tests (beta)

• Create PR with unit tests

---

_{Comment @coderabbitai help to get the list of available commands.}