Skip to content

Commit 36417b9

Browse files
authored
Merge branch 'main' into remove-tensorboard-profiler-tutorial
2 parents 7487a56 + 05c08f2 commit 36417b9

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

intermediate_source/torch_compile_tutorial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -330,7 +330,7 @@ def bar(a, b):
330330
#
331331
# The second time we run ``bar``, we take the other branch of the if statement
332332
# and we get 1 traced graph corresponding to the code ``b = b * -1; return x * b``.
333-
# We do not see a graph of ``x = a / (torch.abs(a) + 1)`` outputted the second time
333+
# We do not see a graph of ``x = a / (torch.abs(a) + 1); b.sum()`` outputted the second time
334334
# since ``torch.compile`` cached this graph from the first run and re-used it.
335335
#
336336
# Let's investigate by example how TorchDynamo would step through ``bar``.

0 commit comments

Comments
 (0)