tests/pytorch/distributed/test_sanity.py · e1221735c9104ee6a718c2d9a2002e587b2b97dc · OpenDAS / TransformerEngine

[PyTorch] Cache RHT device tensors properly (#2395) · e1221735

Kirthi Shankar Sivamani authored Nov 18, 2025



* Cache device tensors properly
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* Fix annotation and add test
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* skip nvfp4 test if not supported
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

---------
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

e1221735

test_sanity.py 5.62 KB

Replace test_sanity.py