File tree Expand file tree Collapse file tree 3 files changed +4
-2
lines changed Expand file tree Collapse file tree 3 files changed +4
-2
lines changed Original file line number Diff line number Diff line change 1
1
import sys
2
+
2
3
import torch
3
4
from . import convert_frame , eval_frame , resume_execution
4
5
from .backends .registry import list_backends , lookup_backend , register_backend
@@ -85,6 +86,7 @@ def reset() -> None:
85
86
torch ._C ._dynamo .compiled_autograd .clear_cache ()
86
87
code_context .clear ()
87
88
89
+
88
90
def is_win32 ():
89
91
if sys .platform == "win32" :
90
92
return True
Original file line number Diff line number Diff line change 1
1
# mypy: ignore-errors
2
2
3
- from torch ._dynamo import register_backend , is_win32
3
+ from torch ._dynamo import is_win32 , register_backend
4
4
5
5
6
6
@register_backend
Original file line number Diff line number Diff line change @@ -2118,8 +2118,8 @@ def build_checkpoint_variable(**options):
2118
2118
2119
2119
2120
2120
def is_compile_supported (device_type ):
2121
- from .eval_frame import is_dynamo_supported
2122
2121
from . import is_win32
2122
+ from .eval_frame import is_dynamo_supported
2123
2123
2124
2124
compile_supported = is_dynamo_supported ()
2125
2125
if device_type == "cpu" :
You can’t perform that action at this time.
0 commit comments