Merge branch 'main' of https://github.com/NVIDIA/TransformerEngine

a207db1d · yuguo · fbee8990 · 69365f88 · a207db1d
Commit a207db1d authored Apr 01, 2025 by yuguo
Expand all Hide whitespace changes
Inline Side-by-side

Showing with 13 additions and 0 deletions

transformer_engine/pytorch/utils.py transformer_engine/pytorch/utils.py +13 -0

No files found.
--- a/transformer_engine/pytorch/utils.py
+++ b/transformer_engine/pytorch/utils.py