Skip to content

Commit

Permalink
row major eigen backend
Browse files Browse the repository at this point in the history
  • Loading branch information
jwaldrop107 committed Nov 15, 2024
1 parent 4d36c58 commit 70fc3f6
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 3 deletions.
2 changes: 1 addition & 1 deletion include/tensorwrapper/backends/eigen.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ namespace tensorwrapper::eigen {
#ifdef TENSORWRAPPER_HAS_EIGEN

template<typename FloatType, unsigned short Rank>
using data_type = Eigen::Tensor<FloatType, int(Rank)>;
using data_type = Eigen::Tensor<FloatType, int(Rank), Eigen::RowMajor>;

#else

Expand Down
5 changes: 3 additions & 2 deletions tests/cxx/unit_tests/tensorwrapper/buffer/eigen.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -26,8 +26,9 @@ namespace {

template<typename FloatType, typename LHSType, typename RHSType>
bool compare_eigen(const LHSType& lhs, const RHSType& rhs) {
auto d = lhs - rhs;
Eigen::Tensor<FloatType, 0> r = d.sum();
using r_type = Eigen::Tensor<FloatType, 0, Eigen::RowMajor>;
auto d = lhs - rhs;
r_type r = d.sum();

return (r() == 0.0);
}
Expand Down

0 comments on commit 70fc3f6

Please sign in to comment.