Feature/unswizzle by int-smart · Pull Request #2732 · NVIDIA/TransformerEngine

Commits on Mar 4, 2026

Add unswizzling functions for scaling factors in swizzle module
int-smart
committed
Add swizzle/unswizzle roundtrip test for scaling factors
int-smart
committed
Added another unswizzling functionality test for scaling factors
int-smart
committed
Moved swizzle_row_scaling_kernel implementation at its original place
int-smart
committed
Add multi-tensor unswizzling functions for scaling factors
int-smart
committed
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
committed

Commits on Mar 5, 2026

Commits on Mar 13, 2026

Commits on Mar 17, 2026

Commits on Mar 19, 2026

Refactor unswizzling scaling factors to improve input validation and streamline processing. Need to check if rowwise and columnwise both can be true. If yes the if else needs to account for that
int-smart
committed

Commits on Mar 20, 2026

Fix multi_tensor_unswizzle_scaling_factors to correctly reference output tensors for scaling mode and data validation. Updated checks for input and output tensor shapes to ensure proper handling of…
int-smart
committed

Commits on Mar 21, 2026

Commits on Apr 1, 2026

Commits on Apr 3, 2026

Merge branch 'main' into feature/unswizzle
ptrendx
committed

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Feature/unswizzle#2732

Feature/unswizzle#2732
ptrendx merged 23 commits intoNVIDIA:mainfrom
int-smart:feature/unswizzle

Commits on Mar 4, 2026

Add unswizzling functions for scaling factors in swizzle module

Add swizzle/unswizzle roundtrip test for scaling factors

Added another unswizzling functionality test for scaling factors

Moved swizzle_row_scaling_kernel implementation at its original place

Add multi-tensor unswizzling functions for scaling factors

[pre-commit.ci] auto fixes from pre-commit.com hooks

Commits on Mar 5, 2026

Added greptile suggestions

[pre-commit.ci] auto fixes from pre-commit.com hooks

Removed unused check from tests and reading input directly as const rather than casting

[pre-commit.ci] auto fixes from pre-commit.com hooks

Commits on Mar 13, 2026

Refactor unswizzling functions and update test cases for scaling factors

[pre-commit.ci] auto fixes from pre-commit.com hooks

Commits on Mar 17, 2026

Refactor unswizzling scaling factors to use a launch function

Change unswizzling to use output as gt.

Commits on Mar 19, 2026

Refactor unswizzling scaling factors to improve input validation and streamline processing. Need to check if rowwise and columnwise both can be true. If yes the if else needs to account for that

Commits on Mar 20, 2026

Fix multi_tensor_unswizzle_scaling_factors to correctly reference output tensors for scaling mode and data validation. Updated checks for input and output tensor shapes to ensure proper handling of…

Commits on Mar 21, 2026

Enhance swizzle tests and unswizzling validation

Fix typos and update validation checks in swizzle.cu

Update validation checks in multi_tensor_unswizzle_scaling_factors to use input numel

Typo

Commits on Apr 1, 2026

Merge branch 'main' into feature/unswizzle

[pre-commit.ci] auto fixes from pre-commit.com hooks

Commits on Apr 3, 2026

Merge branch 'main' into feature/unswizzle