cufft_wrapper.hpp File Reference

#include <lbann/base.hpp>
#include <lbann/utils/dim_helpers.hpp>
#include <lbann/utils/exception.hpp>
#include <lbann/utils/fft_common.hpp>
#include <cufft.h>

Include dependency graph for cufft_wrapper.hpp:

Go to the source code of this file.

Classes
struct	lbann::cufft::cuFFTTypeT< T >
struct	lbann::cufft::cuFFTTypeT< float >
struct	lbann::cufft::cuFFTTypeT< double >
struct	lbann::cufft::cuFFTTypeT< El::Complex< float > >
struct	lbann::cufft::cuFFTTypeT< El::Complex< double > >
struct	lbann::cufft::cuFFTExecutor< InType, OutType >
	Alias around the C-compatible API. More...
struct	lbann::cufft::cuFFTExecutor< El::Complex< float >, El::Complex< float > >
struct	lbann::cufft::cuFFTExecutor< El::Complex< double >, El::Complex< double > >
class	lbann::cufft::cuFFTWrapper< InputTypeT >
	Wrapper around cuFFT. More...
struct	lbann::cufft::cuFFTWrapper< InputTypeT >::InternalPlanType

Namespaces
	lbann
	lbann::cufft

Macros
#define	LBANN_CHECK_CUFFT(cmd)

Typedefs
template<typename T >
using	lbann::cufft::cuFFTType = typename cuFFTTypeT< T >::type

Functions
std::string	lbann::cufft::value_as_string (cufftResult_t)
	The stringified name of the enumerated value. More...
std::string	lbann::cufft::result_string (cufftResult_t)
	The docstring for the given result. More...
template<typename T >
auto	lbann::cufft::AsCUFFTType (T *buffer)

Macro Definition Documentation

LBANN_CHECK_CUFFT

#define LBANN_CHECK_CUFFT

(

cmd

)

Value:

do {                                                                         \
    auto const lbann_check_cufft_result_ = (cmd);                              \
    if (lbann_check_cufft_result_ != CUFFT_SUCCESS) {                          \
      LBANN_ERROR("cuFFT error!\n\n"                                           \
                  "      cmd: " #cmd "\n"                                      \
                  "   result: ",                                               \
                  lbann::cufft::value_as_string(lbann_check_cufft_result_),    \
                  "\n"                                                         \
                  "  message: ",                                               \
                  lbann::cufft::result_string(lbann_check_cufft_result_),      \
                  "\n\n");                                                     \
    }                                                                          \
  } while (0)

Definition at line 36 of file cufft_wrapper.hpp.