transformer_engine/pytorch/csrc/quantizer.cpp · 47902e96c5414750b5e61c8e49968b0dc1b944e5 · OpenDAS / TransformerEngine

Add separate RNG states for column-wise quantization with Stochastic Rounding (#2487) · a5694f26

Evgeny Tsykunov authored Dec 11, 2025



* Add separate RNG states for columnwise quantization with Stochastic Rounding
Signed-off-by: Evgeny <etsykunov@nvidia.com>

* Fix single tensor path
Signed-off-by: Evgeny <etsykunov@nvidia.com>

---------
Signed-off-by: Evgeny <etsykunov@nvidia.com>

a5694f26

quantizer.cpp 74.7 KB

Replace quantizer.cpp