diff --git a/thunder/core/proxies.py b/thunder/core/proxies.py index ecb2e680f..9832af300 100644 --- a/thunder/core/proxies.py +++ b/thunder/core/proxies.py @@ -2193,7 +2193,7 @@ def tensorproxy(t: torch.Tensor, /, *, name: None | str, history: None | tuple = lambda: f"{t=} seems to be a tensor subclass but not traceable", ) tensor_attr_names, metadata = t.__tensor_flatten__() - tensors = [getattr(t, name) for name in tensor_attr_names] + tensors = [tensorproxy(getattr(t, name), name=None, history=history) for name in tensor_attr_names] ctor_kwargs.update( { "tensors": tensors,