tests/pytorch/test_deferred_init.py · c90a9214091badd1234b2d9ca851bd97f8edb0f6 · OpenDAS / TransformerEngine

Add tests that reset_parameters doesn't change parameter initial value ranges (#2550) · c90a9214

Peter St. John authored Jan 04, 2026



* Add tests for 2528 and 2529
Signed-off-by: Peter St. John <pstjohn@nvidia.com>

* Update tests/pytorch/test_deferred_init.py
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* Update tests/pytorch/test_deferred_init.py
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

---------
Signed-off-by: Peter St. John <pstjohn@nvidia.com>
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>
Co-authored-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

c90a9214

test_deferred_init.py 4.08 KB

Replace test_deferred_init.py