Bug fix for num_warmup_iters=0 case (#1095)

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>

Bug fix for num_warmup_iters=0 case (#1095)
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 · vasunvidia · GitHub · e0aa7992 · 44c8924f
Unverified Commit 44c8924f authored Aug 12, 2024 by vasunvidia Committed by GitHub Aug 12, 2024
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

transformer_engine/pytorch/graph.py transformer_engine/pytorch/graph.py +1 -1

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