In nutshell, with these two flags, the recompiles are in control torch._dynamo.config.inline_inbuilt_nn_modules = True ## we are hoping this will be turned on by default in few weeks torch._dynamo.config.force_parameter_static_shapes = False ...
anijain2305 mentioned this pull request Jul 18, 2024 [dynamo] Revert https://github.com/pytorch/pytorch/pull/130416 #131058 Closed anijain2305 added a commit that referenced this pull request Jul 18, 2024 [dynamo] Turn on inline_inbuilt_nn_modules … 7d347f3 Update on "[dynamo]...
anijain2305 commented Jul 13, 2024 • edited by pytorch-bot bot Stack from ghstack (oldest at bottom): [DONT MERGE YET][dynamo] Inline inbuilt nn modules #130680 -> [easy][inline-inbuilt-nn-module] Update test output #130681 [dynamo][unsoundness but very controlled] Skip guards on...
[dynamo] Turn on inline_inbuilt_nn_modules … 914244f pytorch-bot bot commented Jul 20, 2024 • edited 🔗 Helpful Links 🧪 See artifacts and rendered test results at hud.pytorch.org/pr/131275 📄 Preview Python docs built from this PR 📄 Preview C++ docs built from this PR ...
Tensors and Dynamic neural networks in Python with strong GPU acceleration - Add _dynamo.config inline_inbuilt_nn_modules and specialize_float logging · pytorch/pytorch@9e24eb7
Tensors and Dynamic neural networks in Python with strong GPU acceleration - [dynamo][compile-time][inlining-inbuilt-nn-modules] Manually implement nn.Module._call_impl · pytorch/pytorch@148badb
Tensors and Dynamic neural networks in Python with strong GPU acceleration - [dynamo][reland][inline-inbuilt-nn-modules] Mark attributes of nn mod… · pytorch/pytorch@dd83398