CudnnHelpers.h File Reference

Context management functions for working with cuDNN. More...

#include <cudnn.h>
#include <nvneural/CoreTypes.h>
#include <nvneural/CoreHelpers.h>
#include <nvneural/CudaTypes.h>

Go to the source code of this file.

Functions
cudnnHandle_t	nvneural::GetCudnnHandle (const INetworkBackendCuda *pBackend) noexcept
	Retrieves the cuDNN handle associated with a CUDA backend. More...
NeuralResult	nvneural::InitializeCudnn (INetworkBackendCuda *pBackend) noexcept
	Initializes cuDNN and registers it with the backend. More...

Detailed Description

Context management functions for working with cuDNN.

Function Documentation

GetCudnnHandle()

cudnnHandle_t nvneural::GetCudnnHandle ( const INetworkBackendCuda *  pBackend )

noexcept

Retrieves the cuDNN handle associated with a CUDA backend.

Parameters

pBackend

CUDA backend to query

Returns

The initialized cudnnHandle_t, or nullptr if InitializeCudnn has not been called.

InitializeCudnn()

NeuralResult nvneural::InitializeCudnn ( INetworkBackendCuda *  pBackend )

noexcept

Initializes cuDNN and registers it with the backend.

Parameters

pBackend

CUDA backend to enable for cuDNN