Compare commits

...

1 Commits

Author SHA1 Message Date
1d92e7401c [DONT MERGE][dynamo] Turn on inlining of inbuilt nn modules
ghstack-source-id: e2a479b73506358a9f269df545e92c580005a522
Pull Request resolved: https://github.com/pytorch/pytorch/pull/128148
2024-06-17 23:04:53 -07:00

View File

@ -368,7 +368,7 @@ enable_cpp_guard_manager = os.environ.get("TORCHDYNAMO_CPP_GUARD_MANAGER", "1")
# Inline inbuilt nn modules
inline_inbuilt_nn_modules = (
os.environ.get("TORCHDYNAMO_INLINE_INBUILT_NN_MODULES", "0") == "1"
os.environ.get("TORCHDYNAMO_INLINE_INBUILT_NN_MODULES", "1") == "1"
)