tests/jax/test_custom_call_compute.py · c57ffc51a83ce16f7961df5b7b65c08080eb6639 · OpenDAS / TransformerEngine

[JAX] L1_jax_distributed_test suit with individual executions (#2321) · c57ffc51

Phuong Nguyen authored Nov 03, 2025



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

* comment out test_multi_process_grouped_gemm for now
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>

* rm e5m2 from test norm + MXFP8
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>

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

c57ffc51

test_custom_call_compute.py 76.7 KB

Replace test_custom_call_compute.py