Enhance GroupedLinear with integrating AITER triton kernels by sudhu2k · Pull Request #413 · ROCm/TransformerEngine

Commits on Jan 19, 2026

Enhance GroupedLinear with Triton kernel support and update setup.py for JSON configs
sugovind
committed
Added copyright and fixed env var for triton grouped gemm
sugovind
committed
Added grouped_linear module test with triton
sugovind
committed
Fix for unit test, set back the env variable to 0
sugovind
committed
Update numerical test tolerances for float32 in grouped linear accuracy test
sugovind
committed
Relaxed rtol tolerance for float32 in grouped linear accuracy test
sugovind
committed
Update copyright years in multiple files to 2026
sugovind
committed
Addressed PR comments to refactor grouped linear implementation and update setup for Triton kernel support
sugovind
committed
Reverting setup.py
sugovind
committed
Updated test for TEv2.8
sugovind
committed
Re-add GMM configs to core package for dual-install support
sugovind
committed

Commits on Jan 21, 2026

Unified group linear accuracy test for triton and hip. Reverted setup.py in pytorch dir
sugovind
committed

Commits on Jan 22, 2026

Commits on Jan 23, 2026

Commits on Jan 26, 2026

Skip tests for large M values in grouped GEMM to avoid significant numerical differences. Updated test logic to conditionally skip based on M size.
sugovind
committed

Commits on Jan 27, 2026

Commits on Jan 31, 2026

Refactor test_numerics.py to improve clarity and organization; move environment variable setup for Triton to a cleaner structure, and streamline the parameter sharing logic in grouped linear accura…
sugovind
committed

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Enhance GroupedLinear with integrating AITER triton kernels#413

Enhance GroupedLinear with integrating AITER triton kernels#413
sudhu2k merged 25 commits intodevfrom
sudhu/aiter_grouped_gemm_integration

Commits on Jan 19, 2026

Enhance GroupedLinear with Triton kernel support and update setup.py for JSON configs

Added copyright and fixed env var for triton grouped gemm

Added grouped_linear module test with triton

Fix for unit test, set back the env variable to 0

Update numerical test tolerances for float32 in grouped linear accuracy test

Relaxed rtol tolerance for float32 in grouped linear accuracy test

Update copyright years in multiple files to 2026

Addressed PR comments to refactor grouped linear implementation and update setup for Triton kernel support

Reverting setup.py

Updated test for TEv2.8

Re-add GMM configs to core package for dual-install support

Commits on Jan 21, 2026

Unified group linear accuracy test for triton and hip. Reverted setup.py in pytorch dir

Commits on Jan 22, 2026

Refactor grouped GEMM function calls in _GroupedLinear to streamline tensor handling for Triton support. Removed unused kwargs parameter in general_grouped_gemm function signature.

Fix typo in variable name for transposed LHS in grouped GEMM tests to ensure consistency in parameterization.

Commits on Jan 23, 2026

Refactor GMM data type handling in tests and common module. Replaced dtype_from_str with str_to_torch_dtype for improved consistency and clarity. Updated supported data types to include fp32, fp16,…

Revert supported data types in GMM and grouped GEMM tests to remove fp32. T

Enhance GMM t by adding fp32 to supported data types and updating configuration handling. Adjusted tolerance levels in tests and modified get_config to adapt block sizes based on dtype.

Commits on Jan 26, 2026

Skip tests for large M values in grouped GEMM to avoid significant numerical differences. Updated test logic to conditionally skip based on M size.

Commits on Jan 27, 2026

Merge remote-tracking branch 'origin/dev' into sudhu/aiter_grouped_gemm_integration

Revert copyright notice gmm.py

Move import of general_grouped_gemm_triton to conditional block for HIP extension support in grouped_linear.py

move os import to the HIP extension conditional block

Commits on Jan 31, 2026

Refactor test_numerics.py to improve clarity and organization; move environment variable setup for Triton to a cleaner structure, and streamline the parameter sharing logic in grouped linear accura…

Commits on Feb 2, 2026

Modified clean up environment handling in test_grouped_linear_accuracy;

Addressed PR Comments