tests/pytorch/test_deferred_init.py · 0d874a4e7f1b73692cc6f4d7ee06542680bc15a5 · 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