Skip to content

Commit 494bce3

Browse files
crcrparpytorchmergebot
authored andcommitted
fix typo in aot_compile error message: enable_aot_config -> enable_aot_compile (pytorch#170441)
As per title, this pull request corrects the error messages related to enabling AOT (Ahead-Of-Time) compilation in the `torch._dynamo.eval_frame` module. Pull Request resolved: pytorch#170441 Approved by: https://github.com/Lucaskabela
1 parent 9d9ecdb commit 494bce3

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

torch/_dynamo/eval_frame.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -475,7 +475,7 @@ def _aot_compile(self, inputs: list[torch._dynamo.aot_compile.ModelInput]) -> No
475475
assert hooks is not None
476476
if not config.enable_aot_compile:
477477
raise RuntimeError(
478-
"AOT Compile is not enabled, please set torch._dynamo.config.enable_aot_config=True"
478+
"AOT Compile is not enabled, please set torch._dynamo.config.enable_aot_compile=True"
479479
)
480480
if not self.dynamo_ctx.fullgraph:
481481
raise RuntimeError(
@@ -495,7 +495,7 @@ def _aot_compile(self, inputs: list[torch._dynamo.aot_compile.ModelInput]) -> No
495495
def _save_aot_compiled_module(self, path: Optional[str] = None) -> bytes:
496496
if not config.enable_aot_compile:
497497
raise RuntimeError(
498-
"AOT Compile is not enabled, please set torch._dynamo.config.enable_aot_config=True"
498+
"AOT Compile is not enabled, please set torch._dynamo.config.enable_aot_compile=True"
499499
)
500500
from torch._dynamo.aot_compile import AOTCompiledModel
501501

@@ -509,7 +509,7 @@ def _save_aot_compiled_module(self, path: Optional[str] = None) -> bytes:
509509
def _load_aot_compiled_module(self, data: bytes) -> None:
510510
if not config.enable_aot_compile:
511511
raise RuntimeError(
512-
"AOT Compile is not enabled, please set torch._dynamo.config.enable_aot_config=True"
512+
"AOT Compile is not enabled, please set torch._dynamo.config.enable_aot_compile=True"
513513
)
514514
from torch._dynamo.aot_compile import AOTCompiledModel
515515

0 commit comments

Comments
 (0)