benchmark
c13df576 - Trace through nn.Module dunder methods for UnspecializedNNModule (#126578)

Commit
1 year ago
Trace through nn.Module dunder methods for UnspecializedNNModule (#126578) Summary: Tracing through `__init__` is important because it initializes (calls STORE_ATTR) on members. By doing that, we kick in the mutation tracking for these objects. So, things like mutating `_modules` etc is tracked automatically. Fixes https://github.com/pytorch/pytorch/issues/111837 X-link: https://github.com/pytorch/pytorch/pull/126578 Approved by: https://github.com/jansel Reviewed By: clee2000 Differential Revision: D58451174 Pulled By: anijain2305 fbshipit-source-id: a9dca5754b57247ed6c3bf97dc6ec0f00670c477
Author
Parents
Loading