caffe2::CuDNNWorkspace Struct Reference

CuDNNWorkspace is a wrapper around a raw cuda pointer that holds the cudnn scratch space. More...

#include <common_cudnn.h>

Public Member Functions
void *	get (size_t nbytes)
void	reset ()

Data Fields
void *	data_ {nullptr}
size_t	nbytes_ {0}

Detailed Description

CuDNNWorkspace is a wrapper around a raw cuda pointer that holds the cudnn scratch space.

This struct is meant to be only used in CuDNNWrapper to provide a program-wide scratch space for CuDNN. The reason behind it is that cudnn function calls are usually very efficient, hence one probably does not want to run multiple cudnn calls at the same time. As a result, one should not need more than one cudnn workspace per device.

Definition at line 321 of file common_cudnn.h.

---

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

• caffe2/core/common_cudnn.h