Dragon - C++ API
A Computation Graph Virtual Machine Based Deep Learning Framework
Classes | Namespaces | Macros | Functions
cudnn_device.h File Reference
#include <stdint.h>
#include <vector>
#include <cudnn.h>
#include "core/types.h"

Go to the source code of this file.

Classes

class  dragon::CuDNNType< T >
 
class  dragon::CuDNNType< float >
 
class  dragon::CuDNNType< double >
 
class  dragon::CuDNNType< float16 >
 

Namespaces

 dragon
 

Macros

#define CUDNN_VERSION_MIN(major, minor, patch)   (CUDNN_VERSION >= (major * 1000 + minor * 100 + patch))
 
#define CUDNN_VERSION_MAX(major, minor, patch)   (CUDNN_VERSION < (major * 1000 + minor * 100 + patch))
 
#define CUDNN_CHECK(condition)
 

Functions

void dragon::CuDNNCreateTensorDesc (cudnnTensorDescriptor_t *desc)
 
void dragon::CuDNNDestroyTensorDesc (cudnnTensorDescriptor_t *desc)
 
template<typename T >
void dragon::CuDNNSetTensorDesc (cudnnTensorDescriptor_t *desc, Tensor *tensor)
 
template<typename T >
void dragon::CuDNNSetTensor4dDesc (cudnnTensorDescriptor_t *desc, const string &data_format, Tensor *tensor)
 
template<typename T >
void dragon::CuDNNSetTensor5dDesc (cudnnTensorDescriptor_t *desc, const string &data_format, Tensor *tensor)
 
template<typename T >
void dragon::CuDNNSetTensor3dDesc (cudnnTensorDescriptor_t *desc, const string &data_format, Tensor *tensor)
 
template<typename T >
void dragon::CuDNNSetTensorDesc (cudnnTensorDescriptor_t *desc, const vec64_t &dims)
 
template<typename T >
void dragon::CuDNNSetTensor4dDesc (cudnnTensorDescriptor_t *desc, const string &data_format, const vec64_t &dims)
 
template<typename T >
void dragon::CuDNNSetTensor4dDescWithGroup (cudnnTensorDescriptor_t *desc, const string &data_format, const vec64_t &dims, const int64_t group)
 
template<typename T >
void dragon::CuDNNSetTensor5dDesc (cudnnTensorDescriptor_t *desc, const string &data_format, const vec64_t &dims)
 
template<typename T >
void dragon::CuDNNSetTensor3dDesc (cudnnTensorDescriptor_t *desc, const string &data_format, const vec64_t &dims)
 
template<typename T >
void dragon::CuDNNSetTensorDesc (cudnnTensorDescriptor_t *desc, const vec64_t &dims, const vec64_t &strides)
 

Macro Definition Documentation

◆ CUDNN_CHECK

#define CUDNN_CHECK (   condition)
Value:
do { \
cudnnStatus_t status = condition; \
CHECK_EQ(status, CUDNN_STATUS_SUCCESS) << "\n" \
<< cudnnGetErrorString(status); \
} while (0)

◆ CUDNN_VERSION_MAX

#define CUDNN_VERSION_MAX (   major,
  minor,
  patch 
)    (CUDNN_VERSION < (major * 1000 + minor * 100 + patch))

◆ CUDNN_VERSION_MIN

#define CUDNN_VERSION_MIN (   major,
  minor,
  patch 
)    (CUDNN_VERSION >= (major * 1000 + minor * 100 + patch))