Skip to content

Commit 7d7440e

Browse files
committed
add no-op tensor subclass transform
Signed-off-by: Masaki Kozuki <mkozuki@nvidia.com>
1 parent 519e813 commit 7d7440e

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed
Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,18 @@
1+
from __future__ import annotations
2+
from typing import TYPE_CHECKING
3+
4+
from torch._subclasses.fake_tensor import FakeTensor, FakeTensorMode
5+
6+
if TYPE_CHECKING:
7+
from thunder.core.symbol import BoundSymbol, VariableInterface
8+
from thunder.core.trace import TraceCtx
9+
10+
11+
__all__ = [
12+
"flatten_tensor_subclasses",
13+
]
14+
15+
16+
def flatten_tensor_subclasses(computation_trace: TraceCtx) -> TraceCtx:
17+
18+
return computation_trace

0 commit comments

Comments
 (0)