config.h

Go to the documentation of this file.

 
#ifndef GMATTENSOR_CONFIG_H
#define GMATTENSOR_CONFIG_H
 
#include <algorithm>
#include <string>
 
#ifdef GMATTENSOR_USE_XTENSOR_PYTHON
#include <xtensor-python/pytensor.hpp>
#else
#include <xtensor/xtensor.hpp>
#endif
 
#define Q(x) #x
#define QUOTE(x) Q(x)
 
#define UNUSED(p) ((void)(p))
 
#define GMATTENSOR_ASSERT_IMPL(expr, file, line) \
    if (!(expr)) { \
        throw std::runtime_error( \
            std::string(file) + ':' + std::to_string(line) + \
            ": assertion failed (" #expr ") \n\t"); \
    }
#ifdef GMATTENSOR_ENABLE_ASSERT
#define GMATTENSOR_ASSERT(expr) GMATTENSOR_ASSERT_IMPL(expr, __FILE__, __LINE__)
#else
#define GMATTENSOR_ASSERT(expr)
#endif
 
namespace GMatTensor {
 
namespace array_type {
 
#ifdef GMATTENSOR_USE_XTENSOR_PYTHON
 
template <typename T, size_t N>
using tensor = xt::pytensor<T, N>;
 
#else
 
template <typename T, size_t N>
using tensor = xt::xtensor<T, N>;
 
#endif
 
} // namespace array_type
 
template <size_t RANK, class T>
struct allocate {
};
 
template <size_t RANK, class EC, size_t N, xt::layout_type L, class Tag>
struct allocate<RANK, xt::xtensor<EC, N, L, Tag>> {
    using type = typename xt::xtensor<EC, RANK, L, Tag>;
};
 
#ifdef XTENSOR_FIXED_HPP
template <size_t RANK, class EC, class S, xt::layout_type L>
struct allocate<RANK, xt::xtensor_fixed<EC, S, L>> {
    using type = typename xt::xtensor<EC, RANK, L>;
};
#endif
 
#ifdef XTENSOR_FIXED_HPP
template <size_t RANK, class EC, class S, xt::layout_type L, bool SH, class Tag>
struct allocate<RANK, xt::xfixed_container<EC, S, L, SH, Tag>> {
    using type = typename xt::xtensor<EC, RANK, L, Tag>;
};
#endif
 
#ifdef PY_TENSOR_HPP
template <size_t RANK, class EC, size_t N, xt::layout_type L>
struct allocate<RANK, xt::pytensor<EC, N, L>> {
    using type = typename xt::pytensor<EC, RANK, L>;
};
#endif
} // namespace GMatTensor
 
#endif