frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-11-04 16:04:58 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Dynamo] Remove ignored modes workaround (#135502)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/135502
Approved by: https://github.com/anijain2305
ghstack dependencies: #134732, #133137, #135443, #135444, #135422
This commit is contained in:
Michael Lazos
2024-09-13 10:40:55 -07:00
committed by PyTorch MergeBot
parent 7743149b2b
commit 5c67cf180e
4 changed files with 5 additions and 88 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
torch/_dynamo/variables/torch_function.py
View File

@ -69,12 +69,6 @@ banned_attrs = [
if is_tensor_base_attr_getter(fn)
]
# Today set default device is placed in the graph and guarded on separately
# so we should not trace through it. In the future we can trace it once
# mode tracing is implemented and not put in the graph, but this is more
# of a BE project and can be evaluated later
IGNORED_MODES = {DeviceContext}
@functools.lru_cache(None)
def get_prev_stack_var_name():