cutensor_support.hpp File Reference

#include "lbann/base.hpp"
#include "lbann/utils/exception.hpp"
#include <cuda_runtime.h>
#include <cutensor.h>

Include dependency graph for cutensor_support.hpp:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	lbann::CUDATypeT< CppType >
struct	lbann::CUDATypeT< __half >
struct	lbann::CUDATypeT< float >
struct	lbann::CUDATypeT< double >
struct	lbann::CUDATypeT< El::Complex< float > >
struct	lbann::CUDATypeT< El::Complex< double > >

Namespaces
	lbann

Macros
#define	CHECK_CUTENSOR(cmd)

Typedefs
template<typename CppType >
using	lbann::CUDAScalar = typename CUDATypeT< CppType >::scalar_type

Functions
static cutensorHandle_t	lbann::make_handle ()
static cutensorHandle_t *	lbann::get_handle_ptr ()

Variables
template<typename CppType >
constexpr auto	lbann::CUDAType = CUDATypeT<CppType>::value
template<typename CppType >
constexpr auto	lbann::CUDAScalarType = CUDATypeT<CUDAScalar<CppType>>::value

Macro Definition Documentation

CHECK_CUTENSOR

#define CHECK_CUTENSOR

(

cmd

)

Value:

do {                                                                         \
    auto const lbann_chk_cutensor_status__ = (cmd);                            \
    if (CUTENSOR_STATUS_SUCCESS != lbann_chk_cutensor_status__) {              \
      LBANN_ERROR("cuTENSOR error (status=",                                   \
                  lbann_chk_cutensor_status__,                                 \
                  "): ",                                                       \
                  cutensorGetErrorString(lbann_chk_cutensor_status__));        \
    }                                                                          \
  } while (false)

Definition at line 35 of file cutensor_support.hpp.