benchmark
93b58047 - Do not treat user defined nn module attributes static for dynamic shape infra (#136516)

Commit
1 year ago
Do not treat user defined nn module attributes static for dynamic shape infra (#136516) Summary: Fixes https://github.com/pytorch/pytorch/issues/136254 Th regression was introduced in https://github.com/pytorch/pytorch/pull/132736 where originally we were trying to fix another regression. This PR and the offending PR together say - "treat user defined nn module attributes as automatic dynamic, but for cudagraphs they will be considered static". This avoid recompilations. This can lead to a cudagraph recording, which is ok. This also maintains the state before inline_inbuilt_nn_modules flag was introduced. X-link: https://github.com/pytorch/pytorch/pull/136516 Approved by: https://github.com/williamwen42 Reviewed By: atalman Differential Revision: D63351217 Pulled By: anijain2305 fbshipit-source-id: 074bc2af485a7b73eb70876c205a8ef6b611abb3
Author
Parents
Loading