Skip to content

Commit 2b241f0

Browse files
authored
[Bug Fix]Enabling a specific torch.dynamo config (#133)
* Update extractor.py * fix
1 parent 5ba2b8b commit 2b241f0

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

graph_net/torch/extractor.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,8 @@
44
from typing import Union, Callable
55
from . import utils
66

7+
torch._dynamo.config.capture_scalar_outputs = True
8+
79

810
def extract(name, dynamic=True, mut_graph_codes=None, placeholder_auto_rename=False):
911
"""

0 commit comments

Comments
 (0)