cutt_permuteimpl.hpp File Reference

#include "lbann/base.hpp"
#include "lbann/utils/exception.hpp"
#include "lbann/utils/typename.hpp"
#include "tensor_dims_utils.hpp"
#include <cutt.h>
#include <iterator>
#include <sstream>
#include <string>
#include <unordered_map>
#include <utility>
#include <vector>

Include dependency graph for cutt_permuteimpl.hpp:

Go to the source code of this file.

Classes
class	lbann::cuTT_PermuteImpl
	cuTT-based implementation of tensor permute. More...

Namespaces
	lbann

Macros
#define	LBANN_CHECK_CUTT(cmd)

Functions
static char const *	cutt_err_string (cuttResult err) noexcept

Macro Definition Documentation

LBANN_CHECK_CUTT

#define LBANN_CHECK_CUTT

(

cmd

)

Value:

do {                                                                         \
    cuttResult _check_cutt_err_result = (cmd);                                 \
    if (CUTT_SUCCESS != _check_cutt_err_result) {                              \
      LBANN_ERROR("cuTT operation \"" #cmd "\" FAILED (",                      \
                  cutt_err_string(_check_cutt_err_result),                     \
                  ")");                                                        \
    }                                                                          \
  } while (0)

Definition at line 44 of file cutt_permuteimpl.hpp.

Function Documentation

cutt_err_string()

static char const* cutt_err_string ( cuttResult  err )

inlinestaticnoexcept

Definition at line 54 of file cutt_permuteimpl.hpp.