diff --git a/userbenchmark/dynamo/dynamobench/_dynamo/testing.py b/userbenchmark/dynamo/dynamobench/_dynamo/testing.py index 527e0138f..96191ea66 100644 --- a/userbenchmark/dynamo/dynamobench/_dynamo/testing.py +++ b/userbenchmark/dynamo/dynamobench/_dynamo/testing.py @@ -29,7 +29,7 @@ is_generator, transform_code_object, ) -from .guards import CheckFunctionManager, GuardedCode +from .guards import CheckFunctionManager, CompileId, GuardedCode from .utils import same unsupported = eval_frame.unsupported @@ -163,7 +163,7 @@ def insert_nops(instructions, code_options): f_code=frame.f_code, ) - return GuardedCode(code, CheckFunctionManager(graph).check_fn) + return GuardedCode(code, CheckFunctionManager(graph).check_fn, CompileId(0, 0)) class CompileCounter: