From 491779c90edbf5e28d4d085e3035408b197d8efa Mon Sep 17 00:00:00 2001 From: Kaining Zhong Date: Wed, 11 Mar 2026 00:31:28 +0000 Subject: [PATCH] [PyTorch] Fix fuser so it releases tensors properly Signed-off-by: Kaining Zhong --- transformer_engine/pytorch/ops/fuser.py | 1 + 1 file changed, 1 insertion(+) diff --git a/transformer_engine/pytorch/ops/fuser.py b/transformer_engine/pytorch/ops/fuser.py index bd3bc94b60..80386db2d9 100644 --- a/transformer_engine/pytorch/ops/fuser.py +++ b/transformer_engine/pytorch/ops/fuser.py @@ -213,6 +213,7 @@ def backward( # Restore saved tensors saved_tensors = restore_from_saved(func_ctx.tensor_objects, func_ctx.saved_tensors) + func_ctx.tensor_objects = None # Unflatten list of saved tensors for ctx in basic_op_ctxs: