types.h

Go to the documentation of this file.

#pragma once

#include <vector> // for tensor_filler_t
#if defined(_HAS_CUDA_)
#include "cudnn.h"
#include "cublas_v2.h"
#endif

namespace magmadnn {

typedef enum memory_t {
    HOST,
    #if defined(_HAS_CUDA_)
    DEVICE,
    MANAGED,
    CUDA_MANAGED
    #endif
} memory_t;

typedef unsigned int device_t;
typedef unsigned int magmadnn_error_t;



struct magmadnn_settings_t {
    unsigned int n_devices;
    #if defined(_HAS_CUDA_)
    cudnnHandle_t cudnn_handle;
    cublasHandle_t cublas_handle;
    #endif
};
namespace internal {
extern magmadnn_settings_t *MAGMADNN_SETTINGS;  /* make this available to everything */
}   // namespace internal


enum tensor_fill_t {
    UNIFORM,
    GLOROT,
    MASK,
    CONSTANT,
    ZERO,
    ONE,
    DIAGONAL,
    IDENTITY,
    NONE
};  

template <typename T>
struct tensor_filler_t {
    tensor_fill_t fill_type;
    std::vector<T> values;
};

} // namespace magmadnn