OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2026-01-15 12:15:51 +00:00
Code Issues Actions 52 Packages Projects Releases Wiki Activity

Fix typos in docs (#80602)

Browse Source 
I hope it helps.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/80602
Approved by: https://github.com/kit1980
This commit is contained in:
apeltop
2022-08-29 23:32:44 +00:00
committed by PyTorch MergeBot
parent 372a19d2c6
commit e7635c06ce
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/cpp/source/notes/tensor_cuda_stream.rst
View File

@@ -92,7 +92,7 @@ CUDA Stream Usage Examples
// get the default CUDA stream on device 0
at::cuda::CUDAStream defaultStream = at::cuda::getDefaultCUDAStream();
// set current CUDA stream back to default CUDA stream on devide 0
// set current CUDA stream back to default CUDA stream on device 0
at::cuda::setCurrentCUDAStream(defaultStream);
// sum() on tensor0 uses `defaultStream` as current CUDA stream
tensor0.sum();
@@ -120,7 +120,7 @@ CUDA Stream Usage Examples
.. attention::
Above code is running on the same CUDA device. `setCurrentCUDAStream` will always set current CUDA stream on current device,
but note that `setCurrentCUDASteram` actually set current stream on the device of passed in CUDA stream.
but note that `setCurrentCUDAStream` actually set current stream on the device of passed in CUDA stream.
2. Acquiring and setting CUDA streams on multiple devices.