transformer_engine/pytorch/graph.py · 44c8924f67b8c1da2cc8cff0cde99a9fdad08050 · OpenDAS / TransformerEngine

Bug fix for num_warmup_iters=0 case (#1095) · 44c8924f

vasunvidia authored Aug 12, 2024



Buf fix for num_warmup_iters=0 case
Signed-off-by: Vasudevan Rengasamy <vrengasamy@nvidia.com>
Co-authored-by: Kirthi Shankar Sivamani <ksivamani@nvidia.com>

44c8924f

graph.py 27.3 KB

Replace graph.py