tests/pytorch/test_fused_optimizer.py · 1470116ebc147a890f2e9384c940d1c1efa36d9f · OpenDAS / TransformerEngine

[C][PyTorch] Remove deprecated `device_id` arg for multi tensor API (#1994) · 1470116e

Kirthi Shankar Sivamani authored Jul 25, 2025



* Remove deprecated device arg
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

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

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

1470116e

test_fused_optimizer.py 27.5 KB

Replace test_fused_optimizer.py