File CUDAContext.h

↰ Parent directory (aten/src/ATen/cuda)

Definition (aten/src/ATen/cuda/CUDAContext.h)

• Program Listing for File CUDAContext.h

Includes

• ATen/Context.h

• ATen/core/ATenGeneral.h

• ATen/cuda/Exceptions.h

• c10/cuda/CUDAFunctions.h

• c10/cuda/CUDAStream.h (File CUDAStream.h)

• cstdint

• cublas_v2.h

• cuda_runtime_api.h

• cusparse.h

Included By

• File Descriptors.h

Namespaces

• Namespace at

• Namespace at::cuda

Functions

• Function at::cuda::canDeviceAccessPeer

• Function at::cuda::getCUDADeviceAllocator

• Function at::cuda::getCurrentCUDABlasHandle

• Function at::cuda::getCurrentCUDASparseHandle

• Function at::cuda::getCurrentDeviceProperties

• Function at::cuda::getDeviceProperties

• Function at::cuda::getNumGPUs

• Function at::cuda::is_available

• Function at::cuda::warp_size