transformer_engine/common/triton/cross_entropy.py · c1a1c04e6341778562a45cff847dcaface8b33cc · OpenDAS / TransformerEngine

Move Triton to common (#2359) · 5ea83432

Teddy Do authored Nov 10, 2025



* move triton to common and change paths
Signed-off-by: tdophung <tdophung@nvidia.com>

* Formatting
Signed-off-by: tdophung <tdophung@nvidia.com>

---------
Signed-off-by: tdophung <tdophung@nvidia.com>

5ea83432

cross_entropy.py 8.88 KB

Replace cross_entropy.py