Skip to content

Dense grouped gemm tests - #1

Merged
sraman-rgb merged 1 commit into
sraman-rgb:dense_grouped_gemmfrom
michal2409:dense_grouped_gemm-tests
Apr 17, 2026
Merged

Dense grouped gemm tests#1
sraman-rgb merged 1 commit into
sraman-rgb:dense_grouped_gemmfrom
michal2409:dense_grouped_gemm-tests

Conversation

@michal2409

Copy link
Copy Markdown

What does this PR do ?

Add a one line overview of what this PR aims to accomplish.

Changelog

  • Add specific line by line info of high level changes in this PR.

GitHub Actions CI

See the CI sectionin the Contributing doc for how to trigger the CI. A Nvidia developer will need to approve and trigger the CI for external contributors.

Before your PR is "Ready for review"

Pre checks:

  • Make sure you read and followed Contributor guidelines
  • Did you write any new necessary tests?
  • Did you add or update any necessary documentation?
  • Does the PR affect components that are optional to install? (Ex: Numba, Pynini, Apex etc)
    • Reviewer: Does the PR have correct import guards for all optional libraries?

If you haven't finished some of the above items you can still open "Draft" PR.

Additional Information

  • Related to # (issue)

The dense_grouped_gemm feature introduces a feature-detection branch
(via inspect.signature) in transformer_engine_layer_spec that was
uncovered — codecov flagged it on the fork's PR. Adds four unit tests:
attribute default, attribute settable, and both branches of the
signature check (forwarded when supported, omitted otherwise) using
patched fakes whose signatures include or exclude the parameter.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
Signed-off-by: Michal Futrega <mfutrega@nvidia.com>
@michal2409
michal2409 force-pushed the dense_grouped_gemm-tests branch from 0f3a037 to 2feb7e4 Compare April 17, 2026 12:54
@sraman-rgb
sraman-rgb merged commit 51b889c into sraman-rgb:dense_grouped_gemm Apr 17, 2026
1 check passed
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