Skip to content

Commit 639d4ca

Browse files
committed
Add fix for ROCm GPU in pytorch profiler
1 parent f05f050 commit 639d4ca

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

intermediate_source/tensorboard_profiler_tutorial.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
Introduction
88
------------
99
PyTorch 1.8 includes an updated profiler API capable of
10-
recording the CPU side operations as well as the CUDA kernel launches on the GPU side.
10+
recording the CPU side operations as well as the CUDA kernel launches on the GPU side (ROCm AMD GPUs are not supported).
1111
The profiler can visualize this information
1212
in TensorBoard Plugin and provide analysis of the performance bottlenecks.
1313

0 commit comments

Comments
 (0)