Skip to content

Commit 5304217

Browse files
authored
Update 2021-10-26-accelerating-pytorch-with-cuda-graphs.md
1 parent 049b4db commit 5304217

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

_posts/2021-10-26-accelerating-pytorch-with-cuda-graphs.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,7 @@ Deep learning frameworks use GPUs to accelerate computations, but a significant
193193
CUDA graphs can automatically eliminate CPU overhead when tensor shapes are static. A complete graph of all the kernel calls is captured during the first step, in subsequent steps the entire graph is launched with a single op, eliminating all the CPU overhead, as observed in Figure 4..
194194

195195
<p align="center">
196-
<img src="{{ site.url }}/assets/images/cuda-image-5.png" alt="With CUDA graph, the entire graph is launched with a single op, eliminating all the CPU overhead.">
196+
<img src="{{ site.url }}/assets/images/cuda-image-5.png" alt="With CUDA graph, the entire graph is launched with a single op, eliminating all the CPU overhead" width="100%">
197197
<br>
198198
Figure 4: CUDA graphs optimization
199199
</p>

0 commit comments

Comments
 (0)