tests/jax/test_distributed_layernorm.py · 5350f277c26b8dfc40178194e1ddb46e2efd8d5e · OpenDAS / TransformerEngine

"vscode:/vscode.git/clone" did not exist on "8567b09290fe402cf01923e2131c5635b8ed851b"

[JAX] Remove unneccessary MXFP8 scale_inv padding (#1954) · 5350f277

Phuong Nguyen authored Jul 17, 2025



* remove unnecessary padding
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>

* adapt the test_distributed_layernorm byte count
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>


---------
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>

5350f277

test_distributed_layernorm.py 10.1 KB

Replace test_distributed_layernorm.py