frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

PEP585 update - torch/_dynamo (#145105)

Browse Source 
See #145101 for details.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/145105
Approved by: https://github.com/bobrenjc93
This commit is contained in:
Aaron Orenstein
2025-01-18 08:56:06 -08:00
committed by PyTorch MergeBot
parent c95efc37ba
commit a79100ab11
69 changed files with 847 additions and 864 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
torch/_dynamo/bytecode_analysis.py
View File

@ -3,7 +3,7 @@ import bisect
import dataclasses
import dis
import sys
from typing import Any, Set, Union
from typing import Any, Union
TERMINAL_OPCODES = {
@ -12,8 +12,7 @@ TERMINAL_OPCODES = {
dis.opmap["RAISE_VARARGS"],
# TODO(jansel): double check exception handling
}
if sys.version_info >= (3, 9):
TERMINAL_OPCODES.add(dis.opmap["RERAISE"])
TERMINAL_OPCODES.add(dis.opmap["RERAISE"])
if sys.version_info >= (3, 11):
TERMINAL_OPCODES.add(dis.opmap["JUMP_BACKWARD"])
TERMINAL_OPCODES.add(dis.opmap["JUMP_FORWARD"])
@ -133,9 +132,9 @@ def remove_extra_line_nums(instructions):
@dataclasses.dataclass
class ReadsWrites:
reads: Set[Any]
writes: Set[Any]
visited: Set[Any]
reads: set[Any]
writes: set[Any]
visited: set[Any]
def livevars_analysis(instructions, instruction):