matrix.h

Go to the documentation of this file.

 
#pragma once
 
#include <tvm/internal/meta.h>
 
#include <Eigen/Core>
 
namespace tvm::diagnostic
{
template<typename MatrixOrMapType>
inline bool isInMatrix(typename MatrixOrMapType::Scalar const * ptr, MatrixOrMapType & M)
{
  if constexpr(MatrixOrMapType::IsRowMajor)
  {
    static_assert(::tvm::internal::always_false<MatrixOrMapType>::value, "Only implemented for column-major matrices");
  }
  else
  {
    auto d = ptr - M.data();
    if(d < 0)
      return false;
 
    // within the column number
    if(d >= M.cols() * M.colStride())
      return false;
 
    // position in column
    auto r = (d % M.colStride());
    if(r >= M.rows() * M.rowStride())
      return false;
 
    return (r % M.rowStride() == 0);
  }
}
 
template<typename MatrixT, typename MatrixOrMapType>
inline bool isInMatrix(const MatrixT & A, Eigen::Index i, Eigen::Index j, MatrixOrMapType & M)
{ return isInMatrix(&A.coeff(i, j), M); }
 
template<typename Derived>
inline bool hasElemAbsInRange(const Eigen::MatrixBase<Derived> & A,
                              typename Derived::Scalar emin,
                              typename Derived::Scalar emax)
{ return (A.array().abs() >= emin && A.array().abs() <= emax).any(); }
 
} // namespace tvm::diagnostic