Skip to content

GitLab

Unverified Commit 0e2a5ae7 authored by vfdev's avatar vfdev Committed by GitHub
Browse files

Added support for tv tensors in torch compile for func ops (#8110) 


Co-authored-by: default avatarNicolas Hug <nh.nicolas.hug@gmail.com>
parent 517f6d3b
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
torchvision/tv_tensors/__init__.py
View file @ 0e2a5ae7
...@@ -8,6 +8,10 @@ from ._tv_tensor import TVTensor ...@@ -8,6 +8,10 @@ from ._tv_tensor import TVTensor
from ._video import Video from ._video import Video
# TODO: Fix this. We skip this method as it leads to
# RecursionError: maximum recursion depth exceeded while calling a Python object
# Until `disable` is removed, there will be graph breaks after all calls to functional transforms
@torch.compiler.disable
def wrap(wrappee, *, like, **kwargs): def wrap(wrappee, *, like, **kwargs):
"""[BETA] Convert a :class:`torch.Tensor` (``wrappee``) into the same :class:`~torchvision.tv_tensors.TVTensor` subclass as ``like``. """[BETA] Convert a :class:`torch.Tensor` (``wrappee``) into the same :class:`~torchvision.tv_tensors.TVTensor` subclass as ``like``.
......
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