Unverified Commit 44c8924f authored by vasunvidia's avatar vasunvidia Committed by GitHub
Browse files

Bug fix for num_warmup_iters=0 case (#1095)



Buf fix for num_warmup_iters=0 case
Signed-off-by: default avatarVasudevan Rengasamy <vrengasamy@nvidia.com>
Co-authored-by: default avatarKirthi Shankar Sivamani <ksivamani@nvidia.com>
parent e0aa7992
...@@ -213,7 +213,7 @@ def _make_graphed_callables( ...@@ -213,7 +213,7 @@ def _make_graphed_callables(
only_inputs=True, only_inputs=True,
allow_unused=allow_unused_input, allow_unused=allow_unused_input,
) )
del outputs, grad_inputs del outputs, grad_inputs
torch.cuda.synchronize() torch.cuda.synchronize()
# All captures here share a mempool. To avoid replays corrupting each other's memory, # All captures here share a mempool. To avoid replays corrupting each other's memory,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment