Skip to content

Commit 44c2ac2

Browse files
authored
Added support for tv tensors in torch compile for func ops
1 parent 15c166a commit 44c2ac2

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

torchvision/tv_tensors/__init__.py

+4
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,10 @@
88
from ._video import Video
99

1010

11+
# TODO: Fix this. We skip this method as it leads to
12+
# RecursionError: maximum recursion depth exceeded while calling a Python object
13+
# Keeping it here, leads to graph breaks between multiple functional ops instead of having a single graph
14+
@torch.compiler.disable
1115
def wrap(wrappee, *, like, **kwargs):
1216
"""[BETA] Convert a :class:`torch.Tensor` (``wrappee``) into the same :class:`~torchvision.tv_tensors.TVTensor` subclass as ``like``.
1317

0 commit comments

Comments
 (0)