Skip to content

Commit

Permalink
Use guard_manager consistently instead of check_fn (#138896)
Browse files Browse the repository at this point in the history
Summary:
X-link: pytorch/pytorch#138896
Approved by: https://github.com/williamwen42, https://github.com/jansel
ghstack dependencies: #138512

Reviewed By: wdvr

Differential Revision: D65030963

Pulled By: anijain2305

fbshipit-source-id: 7423473e4c3613aea42e13a64eae9c417c876964
  • Loading branch information
anijain2305 authored and facebook-github-bot committed Oct 27, 2024
1 parent bd23811 commit 34ea1a1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion userbenchmark/dynamo/dynamobench/_dynamo/testing.py
Original file line number Diff line number Diff line change
Expand Up @@ -191,7 +191,7 @@ def insert_nops(instructions: List[Any], code_options: Any) -> None:
torch_function_mode_stack=[],
)

return GuardedCode(code, CheckFunctionManager(graph).check_fn, CompileId(0, 0))
return GuardedCode(code, CheckFunctionManager(graph).guard_manager, CompileId(0, 0)) # type: ignore[arg-type]


class CompileCounter:
Expand Down

0 comments on commit 34ea1a1

Please sign in to comment.