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 53 Packages Projects Releases Wiki Activity

Move CUDAGuard, CUDAStream and CUDAGuardImpl to c10/cuda (#14248)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14248

This diff also introduces a horrifying hack to override CUDA's DeviceGuardImpl
with a HIPGuardImplMasqueradingAsCUDA, to accommodate PyTorch's current
behavior of pretending CUDA is HIP when you build with ROCm enabled.

Reviewed By: bddppq

Differential Revision: D13145293

fbshipit-source-id: ee0e207b6fd132f0d435512957424a002d588f02
This commit is contained in:
Edward Yang
2018-12-12 11:19:03 -08:00
committed by Facebook Github Bot
parent 9943cf2378
commit 2d485ffb17
47 changed files with 229 additions and 83 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/cpp/source/Doxyfile
View File

@@ -35,8 +35,6 @@ INPUT = ../../../aten/src/ATen/ATen.h \
../../../aten/src/ATen/core/ScalarType.h \
../../../aten/src/ATen/core/Tensor.h \
../../../aten/src/ATen/cuda/CUDAContext.h \
../../../aten/src/ATen/cuda/CUDAGuard.h \
../../../aten/src/ATen/cuda/CUDAStream.h \
../../../aten/src/ATen/cudnn/Descriptors.h \
../../../aten/src/ATen/cudnn/Handles.h \
../../../aten/src/ATen/cudnn/Types.h \
@@ -52,6 +50,8 @@ INPUT = ../../../aten/src/ATen/ATen.h \
../../../c10/util/ArrayRef.h \
../../../c10/util/Exception.h \
../../../c10/util/Optional.h \
../../../c10/cuda/CUDAGuard.h \
../../../c10/cuda/CUDAStream.h \
../../../torch/csrc/api/include \
../../../torch/csrc/api/src \
../../../torch/csrc/autograd/generated/variable_factories.h \