Skip to content

Commit 27cdd1d

Browse files
anijain2305facebook-github-bot
authored andcommitted
Return gm.forward for eager backend (#124109)
Summary: X-link: pytorch/pytorch#124109 Approved by: https://github.com/yanboliang, https://github.com/jansel ghstack dependencies: #124445 Reviewed By: jeanschmidt Differential Revision: D56408649 Pulled By: anijain2305 fbshipit-source-id: c15bd1c2034bb6e01bc9a909f91021702579e779
1 parent a0692ac commit 27cdd1d

File tree

1 file changed

+1
-1
lines changed
  • userbenchmark/dynamo/dynamobench/_dynamo

1 file changed

+1
-1
lines changed

userbenchmark/dynamo/dynamobench/_dynamo/testing.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -208,7 +208,7 @@ def __init__(self):
208208

209209
def __call__(self, gm: torch.fx.GraphModule, example_inputs: List[torch.Tensor]):
210210
self.graphs.append(gm)
211-
return gm
211+
return gm.forward
212212

213213

214214
def strip_comment(code) -> str:

0 commit comments

Comments
 (0)