getrf.h

Go to the documentation of this file.

#ifndef TATOOINE_LAPACK_GETRF_H
#define TATOOINE_LAPACK_GETRF_H
//==============================================================================
extern "C" {
//==============================================================================
auto dgetrf_(int* M, int* N, double* A, int* LDA, int* IPIV, int* INFO) -> void;
auto sgetrf_(int* M, int* N, double* A, int* LDA, int* IPIV, int* INFO) -> void;
//==============================================================================
}
//==============================================================================
#include <tatooine/lapack/base.h>
//==============================================================================
namespace tatooine::lapack {
//==============================================================================
//==============================================================================
//==============================================================================
template <std::floating_point Float>
auto getrf(int M, int N, double* A, int LDA, int* IPIV) -> int {
  auto INFO = int{};
  if constexpr (std::same_as<Float, double>) {
    dgetrf_(&M, &N, A, &LDA, IPIV, &INFO);
  } else if constexpr (std::same_as<Float, float>) {
    sgetrf_(&M, &N, A, &LDA, IPIV, &INFO);
  }
  return INFO;
}
//------------------------------------------------------------------------------
template <typename T, size_t M, size_t N>
auto getrf(tensor<T, M, N>& A, tensor<int, tatooine::min(M, N)>& p) {
  return getrf(M, N, A.data(), M, p.data());
}
//==============================================================================
//==============================================================================
}  // namespace tatooine::lapack
//==============================================================================
#endif