OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-08 07:39:33 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[dynamo][nn module] Use correct sources for _call_impl (#124970)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/124970
Approved by: https://github.com/jansel
ghstack dependencies: #124779, #124627
This commit is contained in:
Animesh Jain 2024-04-26 11:14:24 -07:00 committed by PyTorch MergeBot
parent 43069c460e
commit fd24d8c05a
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
torch/_dynamo/variables/nn_module.py
View File

@ -340,9 +340,10 @@ class NNModuleVariable(VariableTracker):
# If so at least some changes are needed, we don't allow inlining # If so at least some changes are needed, we don't allow inlining
# the call_wrapped currently, and maybe other issues too # the call_wrapped currently, and maybe other issues too
fn = mod.forward fn = mod.forward
fn_source = AttrSource(self.source, "forward")
else: else:
fn = mod._call_impl fn = mod._call_impl
fn_source = AttrSource(self.source, "__call__") fn_source = AttrSource(self.source, "_call_impl")
if istype(fn, types.MethodType): if istype(fn, types.MethodType):
fn = fn.__func__ fn = fn.__func__
fn_source = AttrSource(fn_source, "__func__") fn_source = AttrSource(fn_source, "__func__")