Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
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: