Skip to content

Commit 6eb17f1

Browse files
williamwen42facebook-github-bot
authored andcommitted
reconstruct functions decorated in the compiled region properly (#150645)
Summary: We were previously unable to reconstruct functions that were decorated in the compiled region. X-link: pytorch/pytorch#150645 Approved by: https://github.com/jansel Reviewed By: atalman Differential Revision: D72656343 fbshipit-source-id: 43cb8c14ad0f4c1b4c8275bd43d4e8a6b2b476de
1 parent 98b06f0 commit 6eb17f1

File tree

1 file changed

+6
-0
lines changed
  • userbenchmark/dynamo/dynamobench/_dynamo

1 file changed

+6
-0
lines changed

userbenchmark/dynamo/dynamobench/_dynamo/testing.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -524,3 +524,9 @@ def reset_rng_state(use_xla: bool = False) -> None:
524524
import torch_xla.core.xla_model as xm
525525

526526
xm.set_rng_state(1337, str(xm.xla_device()))
527+
528+
529+
def _skipped_function_for_test_reconstruct(
530+
f: Callable[_P, _T], *args: _P.args, **kwargs: _P.kwargs
531+
) -> _T:
532+
return f(*args, **kwargs)

0 commit comments

Comments
 (0)