Skip to content

Commit 060c1dd

Browse files
author
Svetlana Karslioglu
committed
Update torch_compile_tutorial.py
1 parent 82544b4 commit 060c1dd

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

intermediate_source/torch_compile_tutorial.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -233,7 +233,7 @@ def train(mod, data):
233233

234234
######################################################################
235235
# Again, we can see that ``torch.compile`` takes longer in the first
236-
# iteration, as it must compile the model, but afterward, we see
236+
# iteration, as it must compile the model, but in subsequent iterations, we see
237237
# significant speedups compared to eager. On an NVIDIA A100 GPU, we
238238
# observe a ~1.8x speedup.
239239

@@ -493,4 +493,4 @@ def bar(a, b):
493493
# In this tutorial, we introduced ``torch.compile`` by covering
494494
# basic usage, demonstrating speedups over eager mode, comparing to previous
495495
# PyTorch compiler solutions, and briefly investigating TorchDynamo and its interactions
496-
# with FX graphs. We hope that you will give ``torch.compile`` a try!
496+
# with FX graphs. We hope that you will give ``torch.compile`` a try!

0 commit comments

Comments
 (0)