diff --git a/userbenchmark/dynamo/common.py b/userbenchmark/dynamo/common.py index dfd650c782..075b88f4a0 100644 --- a/userbenchmark/dynamo/common.py +++ b/userbenchmark/dynamo/common.py @@ -1134,7 +1134,7 @@ def load(cls, model, example_inputs, eager_forward): # Use a utility function for easier benchmarking source = """ - #include + #include torch::aot_inductor::AOTInductorModel model;