Inheritance diagram for at::cuda::detail::CUDAHooks:

Public Member Functions
	CUDAHooks (at::CUDAHooksArgs)

std::unique_ptr< THCState, void()(THCState )>	initCUDA () const override

std::unique_ptr< Generator >	initCUDAGenerator (Context *) const override

bool	hasCUDA () const override

bool	hasMAGMA () const override

bool	hasCuDNN () const override

int64_t	current_device () const override

Allocator *	getPinnedMemoryAllocator () const override

void	registerCUDATypes (Context *) const override

bool	compiledWithCuDNN () const override

bool	compiledWithMIOpen () const override

bool	supportsDilatedConvolutionWithCuDNN () const override

long	versionCuDNN () const override

double	batchnormMinEpsilonCuDNN () const override

int64_t	cuFFTGetPlanCacheMaxSize () const override

void	cuFFTSetPlanCacheMaxSize (int64_t max_size) const override

int64_t	cuFFTGetPlanCacheSize () const override

void	cuFFTClearPlanCache () const override

int	getNumGPUs () const override

Detailed Description

Definition at line 11 of file CUDAHooks.h.

The documentation for this struct was generated from the following files:

Public Member Functions