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(
only_inputs=True,
allow_unused=allow_unused_input,
)
del outputs, grad_inputs
del outputs, grad_inputs
torch.cuda.synchronize()
# 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