tests/pytorch/test_float8tensor.py · 9588109d4c412aa58bc08f523421d82f71a0cc15 · OpenDAS / TransformerEngine

Fix shape of new quantized tensor in `make_like` (#1515) · 9588109d

Kirthi Shankar Sivamani authored Feb 28, 2025



* Fix quantized tensor shape
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* add shape to make_like; add test for chunk
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

* Fix typo from suggestion
Signed-off-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

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

9588109d

test_float8tensor.py 10.7 KB

Replace test_float8tensor.py