caffe2::MIOPENWorkspace Struct Reference

MIOPENWorkspace is a wrapper around a raw cuda pointer that holds the miopen scratch space. More...

#include <miopen_wrapper.h>

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

Detailed Description

MIOPENWorkspace is a wrapper around a raw cuda pointer that holds the miopen scratch space.

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

Definition at line 22 of file miopen_wrapper.h.

---

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

• caffe2/core/hip/miopen_wrapper.h