blas3.hpp

// ops/include/nk/ops/blas3.hpp
#pragma once
 
#include <cstddef>
#include <complex>
 
namespace nk::ops {
 
enum class MatrixSide { Left, Right };
enum class MatrixUplo { Upper, Lower };
enum class MatrixTranspose { NoTrans, Trans, ConjTrans };
enum class MatrixDiag { NonUnit, Unit };

void gemm(MatrixTranspose transa, MatrixTranspose transb,
          std::size_t m, std::size_t n, std::size_t k,
          double alpha, const double *A, std::size_t lda,
          const double *B, std::size_t ldb,
          double beta, double *C, std::size_t ldc);

void gemm(MatrixTranspose transa, MatrixTranspose transb,
          std::size_t m, std::size_t n, std::size_t k,
          std::complex<double> alpha, const std::complex<double> *A, std::size_t lda,
          const std::complex<double> *B, std::size_t ldb,
          std::complex<double> beta, std::complex<double> *C, std::size_t ldc);

void trsm(MatrixSide side, MatrixUplo uplo, MatrixTranspose trans, MatrixDiag diag,
          std::size_t m, std::size_t n,
          double alpha, const double *A, std::size_t lda,
          double *B, std::size_t ldb);

void trsm(MatrixSide side, MatrixUplo uplo, MatrixTranspose trans, MatrixDiag diag,
          std::size_t m, std::size_t n,
          std::complex<double> alpha, const std::complex<double> *A, std::size_t lda,
          std::complex<double> *B, std::size_t ldb);

void syrk(MatrixUplo uplo, MatrixTranspose trans,
          std::size_t n, std::size_t k,
          double alpha, const double *A, std::size_t lda,
          double beta, double *C, std::size_t ldc);

void syrk(MatrixUplo uplo, MatrixTranspose trans,
          std::size_t n, std::size_t k,
          std::complex<double> alpha, const std::complex<double> *A, std::size_t lda,
          std::complex<double> beta, std::complex<double> *C, std::size_t ldc);
 
} // namespace nk::ops