transformer_engine/jax/csrc/extensions/gemm.cpp · 9d4e11eaa508383e35b510dc338e58b09c30be73 · OpenDAS / TransformerEngine

Removing NVTE_NO_SCALING (#1650) · b362a6e0

Phuong Nguyen authored Apr 07, 2025



* rm no scaling enum
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>

* update jax enum
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>

---------
Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>

b362a6e0

gemm.cpp 10.6 KB

Replace gemm.cpp