Skip to content

Commit 4e41a4b

Browse files
Update example
1 parent a2543dc commit 4e41a4b

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

examples/compile_debug.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,13 +7,15 @@
77
torch._logging.set_logs(
88
dynamo=logging.INFO,
99
graph_breaks=True,
10-
recompiles=True,
11-
recompiles_verbose=True,
12-
compiled_autograd_verbose=True,
10+
recompiles=False,
11+
recompiles_verbose=False,
12+
compiled_autograd_verbose=False,
1313
)
1414

1515
torch._dynamo.config.suppress_errors = False
1616

17+
torch._dynamo.config.capture_scalar_outputs = True
18+
torch._dynamo.config.capture_dynamic_output_shape_ops = True
1719

1820
torch.set_float32_matmul_precision("high")
1921

0 commit comments

Comments
 (0)