numo-tiny_linalg 0.0.4 → 0.1.1

data/ext/numo/tiny_linalg/lapack/sygvx.hpp

@@ -18,7 +18,7 @@ struct SSyGvx {
   }
 };
 
-template <int nary_dtype_id, typename DType, typename FncType>
+template <int nary_dtype_id, typename dtype, class LapackFn>
 class SyGvx {
 public:
   static void define_module_function(VALUE mLapack, const char* fnc_name) {
@@ -32,18 +32,18 @@ private:
     char jobz;
     char range;
     char uplo;
-    DType vl;
-    DType vu;
+    dtype vl;
+    dtype vu;
     lapack_int il;
     lapack_int iu;
   };
 
   static void iter_sygvx(na_loop_t* const lp) {
-    DType* a = (DType*)NDL_PTR(lp, 0);
-    DType* b = (DType*)NDL_PTR(lp, 1);
+    dtype* a = (dtype*)NDL_PTR(lp, 0);
+    dtype* b = (dtype*)NDL_PTR(lp, 1);
     int* m = (int*)NDL_PTR(lp, 2);
-    DType* w = (DType*)NDL_PTR(lp, 3);
-    DType* z = (DType*)NDL_PTR(lp, 4);
+    dtype* w = (dtype*)NDL_PTR(lp, 3);
+    dtype* z = (dtype*)NDL_PTR(lp, 4);
     int* ifail = (int*)NDL_PTR(lp, 5);
     int* info = (int*)NDL_PTR(lp, 6);
     sygvx_opt* opt = (sygvx_opt*)(lp->opt_ptr);
@@ -51,8 +51,8 @@ private:
     const lapack_int lda = NDL_SHAPE(lp, 0)[0];
     const lapack_int ldb = NDL_SHAPE(lp, 1)[0];
     const lapack_int ldz = opt->range != 'I' ? n : opt->iu - opt->il + 1;
-    const DType abstol = 0.0;
-    const lapack_int i = FncType().call(
+    const dtype abstol = 0.0;
+    const lapack_int i = LapackFn().call(
       opt->matrix_layout, opt->itype, opt->jobz, opt->range, opt->uplo, n, a, lda, b, ldb,
       opt->vl, opt->vu, opt->il, opt->iu, abstol, m, w, z, ldz, ifail);
     *info = static_cast<int>(i);
@@ -69,15 +69,15 @@ private:
                        rb_intern("vl"), rb_intern("vu"), rb_intern("il"), rb_intern("iu"), rb_intern("order") };
     VALUE kw_values[9] = { Qundef, Qundef, Qundef, Qundef, Qundef, Qundef, Qundef, Qundef, Qundef };
     rb_get_kwargs(kw_args, kw_table, 0, 9, kw_values);
-    const lapack_int itype = kw_values[0] != Qundef ? get_itype(kw_values[0]) : 1;
-    const char jobz = kw_values[1] != Qundef ? get_jobz(kw_values[1]) : 'V';
-    const char range = kw_values[2] != Qundef ? get_range(kw_values[2]) : 'A';
-    const char uplo = kw_values[3] != Qundef ? get_uplo(kw_values[3]) : 'U';
-    const DType vl = kw_values[4] != Qundef ? NUM2DBL(kw_values[4]) : 0.0;
-    const DType vu = kw_values[5] != Qundef ? NUM2DBL(kw_values[5]) : 0.0;
+    const lapack_int itype = kw_values[0] != Qundef ? Util().get_itype(kw_values[0]) : 1;
+    const char jobz = kw_values[1] != Qundef ? Util().get_jobz(kw_values[1]) : 'V';
+    const char range = kw_values[2] != Qundef ? Util().get_range(kw_values[2]) : 'A';
+    const char uplo = kw_values[3] != Qundef ? Util().get_uplo(kw_values[3]) : 'U';
+    const dtype vl = kw_values[4] != Qundef ? NUM2DBL(kw_values[4]) : 0.0;
+    const dtype vu = kw_values[5] != Qundef ? NUM2DBL(kw_values[5]) : 0.0;
     const lapack_int il = kw_values[6] != Qundef ? NUM2INT(kw_values[6]) : 0;
     const lapack_int iu = kw_values[7] != Qundef ? NUM2INT(kw_values[7]) : 0;
-    const int matrix_layout = kw_values[8] != Qundef ? get_matrix_layout(kw_values[8]) : LAPACK_ROW_MAJOR;
+    const int matrix_layout = kw_values[8] != Qundef ? Util().get_matrix_layout(kw_values[8]) : LAPACK_ROW_MAJOR;
 
     if (CLASS_OF(a_vnary) != nary_dtype) {
       a_vnary = rb_funcall(nary_dtype, rb_intern("cast"), 1, a_vnary);
@@ -131,62 +131,6 @@ private:
 
     return ret;
   }
-
-  static lapack_int get_itype(VALUE val) {
-    const lapack_int itype = NUM2INT(val);
-
-    if (itype != 1 && itype != 2 && itype != 3) {
-      rb_raise(rb_eArgError, "itype must be 1, 2 or 3");
-    }
-
-    return itype;
-  }
-
-  static char get_jobz(VALUE val) {
-    const char jobz = NUM2CHR(val);
-
-    if (jobz != 'N' && jobz != 'V') {
-      rb_raise(rb_eArgError, "jobz must be 'N' or 'V'");
-    }
-
-    return jobz;
-  }
-
-  static char get_range(VALUE val) {
-    const char range = NUM2CHR(val);
-
-    if (range != 'A' && range != 'V' && range != 'I') {
-      rb_raise(rb_eArgError, "range must be 'A', 'V' or 'I'");
-    }
-
-    return range;
-  }
-
-  static char get_uplo(VALUE val) {
-    const char uplo = NUM2CHR(val);
-
-    if (uplo != 'U' && uplo != 'L') {
-      rb_raise(rb_eArgError, "uplo must be 'U' or 'L'");
-    }
-
-    return uplo;
-  }
-
-  static int get_matrix_layout(VALUE val) {
-    const char option = NUM2CHR(val);
-
-    switch (option) {
-    case 'r':
-    case 'R':
-      break;
-    case 'c':
-    case 'C':
-      rb_warn("Numo::TinyLinalg::Lapack.sygvx does not support column major.");
-      break;
-    }
-
-    return LAPACK_ROW_MAJOR;
-  }
 };
 
 } // namespace TinyLinalg

data/ext/numo/tiny_linalg/lapack/ungqr.hpp

@@ -14,7 +14,7 @@ struct CUngQr {
   }
 };
 
-template <int nary_dtype_id, typename DType, typename FncType>
+template <int nary_dtype_id, typename dtype, class LapackFn>
 class UngQr {
 public:
   static void define_module_function(VALUE mLapack, const char* fnc_name) {
@@ -27,15 +27,15 @@ private:
   };
 
   static void iter_ungqr(na_loop_t* const lp) {
-    DType* a = (DType*)NDL_PTR(lp, 0);
-    DType* tau = (DType*)NDL_PTR(lp, 1);
+    dtype* a = (dtype*)NDL_PTR(lp, 0);
+    dtype* tau = (dtype*)NDL_PTR(lp, 1);
     int* info = (int*)NDL_PTR(lp, 2);
     ungqr_opt* opt = (ungqr_opt*)(lp->opt_ptr);
     const lapack_int m = NDL_SHAPE(lp, 0)[0];
     const lapack_int n = NDL_SHAPE(lp, 0)[1];
     const lapack_int k = NDL_SHAPE(lp, 1)[0];
     const lapack_int lda = n;
-    const lapack_int i = FncType().call(opt->matrix_layout, m, n, k, a, lda, tau);
+    const lapack_int i = LapackFn().call(opt->matrix_layout, m, n, k, a, lda, tau);
     *info = static_cast<int>(i);
   }
 
@@ -49,7 +49,7 @@ private:
     ID kw_table[1] = { rb_intern("order") };
     VALUE kw_values[1] = { Qundef };
     rb_get_kwargs(kw_args, kw_table, 0, 1, kw_values);
-    const int matrix_layout = kw_values[0] != Qundef ? get_matrix_layout(kw_values[0]) : LAPACK_ROW_MAJOR;
+    const int matrix_layout = kw_values[0] != Qundef ? Util().get_matrix_layout(kw_values[0]) : LAPACK_ROW_MAJOR;
 
     if (CLASS_OF(a_vnary) != nary_dtype) {
       a_vnary = rb_funcall(nary_dtype, rb_intern("cast"), 1, a_vnary);
@@ -90,26 +90,6 @@ private:
 
     return ret;
   }
-
-  static int get_matrix_layout(VALUE val) {
-    const char* option_str = StringValueCStr(val);
-
-    if (std::strlen(option_str) > 0) {
-      switch (option_str[0]) {
-      case 'r':
-      case 'R':
-        break;
-      case 'c':
-      case 'C':
-        rb_warn("Numo::TinyLinalg::Lapack.getrf does not support column major.");
-        break;
-      }
-    }
-
-    RB_GC_GUARD(val);
-
-    return LAPACK_ROW_MAJOR;
-  }
 };
 
 } // namespace TinyLinalg

data/ext/numo/tiny_linalg/tiny_linalg.cpp

@@ -1,10 +1,43 @@
+/**
+ * Copyright (c) 2023 Atsushi Tatsuma
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * * Redistributions of source code must retain the above copyright notice, this
+ *   list of conditions and the following disclaimer.
+ *
+ * * Redistributions in binary form must reproduce the above copyright notice,
+ *   this list of conditions and the following disclaimer in the documentation
+ *   and/or other materials provided with the distribution.
+ *
+ * * Neither the name of the copyright holder nor the names of its
+ *   contributors may be used to endorse or promote products derived from
+ *   this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+ * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+ * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+ * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
 #include "tiny_linalg.hpp"
+
+#include "converter.hpp"
+#include "util.hpp"
+
 #include "blas/dot.hpp"
 #include "blas/dot_sub.hpp"
 #include "blas/gemm.hpp"
 #include "blas/gemv.hpp"
 #include "blas/nrm2.hpp"
-#include "converter.hpp"
 #include "lapack/geqrf.hpp"
 #include "lapack/gesdd.hpp"
 #include "lapack/gesv.hpp"
@@ -197,11 +230,13 @@ extern "C" void Init_tiny_linalg(void) {
   /**
    * Document-module: Numo::TinyLinalg::Blas
    * Numo::TinyLinalg::Blas is wrapper module of BLAS functions.
+   * @!visibility private
    */
   rb_mTinyLinalgBlas = rb_define_module_under(rb_mTinyLinalg, "Blas");
   /**
    * Document-module: Numo::TinyLinalg::Lapack
    * Numo::TinyLinalg::Lapack is wrapper module of LAPACK functions.
+   * @!visibility private
    */
   rb_mTinyLinalgLapack = rb_define_module_under(rb_mTinyLinalg, "Lapack");
 
@@ -249,26 +284,26 @@ extern "C" void Init_tiny_linalg(void) {
   TinyLinalg::Nrm2<TinyLinalg::numo_cSFloatId, float, TinyLinalg::SNrm2>::define_module_function(rb_mTinyLinalgBlas, "snrm2");
   TinyLinalg::Nrm2<TinyLinalg::numo_cDComplexId, double, TinyLinalg::DZNrm2>::define_module_function(rb_mTinyLinalgBlas, "dznrm2");
   TinyLinalg::Nrm2<TinyLinalg::numo_cSComplexId, float, TinyLinalg::SCNrm2>::define_module_function(rb_mTinyLinalgBlas, "scnrm2");
-  TinyLinalg::GESV<TinyLinalg::numo_cDFloatId, double, TinyLinalg::DGESV>::define_module_function(rb_mTinyLinalgLapack, "dgesv");
-  TinyLinalg::GESV<TinyLinalg::numo_cSFloatId, float, TinyLinalg::SGESV>::define_module_function(rb_mTinyLinalgLapack, "sgesv");
-  TinyLinalg::GESV<TinyLinalg::numo_cDComplexId, lapack_complex_double, TinyLinalg::ZGESV>::define_module_function(rb_mTinyLinalgLapack, "zgesv");
-  TinyLinalg::GESV<TinyLinalg::numo_cSComplexId, lapack_complex_float, TinyLinalg::CGESV>::define_module_function(rb_mTinyLinalgLapack, "cgesv");
-  TinyLinalg::GESVD<TinyLinalg::numo_cDFloatId, TinyLinalg::numo_cDFloatId, double, double, TinyLinalg::DGESVD>::define_module_function(rb_mTinyLinalgLapack, "dgesvd");
-  TinyLinalg::GESVD<TinyLinalg::numo_cSFloatId, TinyLinalg::numo_cSFloatId, float, float, TinyLinalg::SGESVD>::define_module_function(rb_mTinyLinalgLapack, "sgesvd");
-  TinyLinalg::GESVD<TinyLinalg::numo_cDComplexId, TinyLinalg::numo_cDFloatId, lapack_complex_double, double, TinyLinalg::ZGESVD>::define_module_function(rb_mTinyLinalgLapack, "zgesvd");
-  TinyLinalg::GESVD<TinyLinalg::numo_cSComplexId, TinyLinalg::numo_cSFloatId, lapack_complex_float, float, TinyLinalg::CGESVD>::define_module_function(rb_mTinyLinalgLapack, "cgesvd");
-  TinyLinalg::GESDD<TinyLinalg::numo_cDFloatId, TinyLinalg::numo_cDFloatId, double, double, TinyLinalg::DGESDD>::define_module_function(rb_mTinyLinalgLapack, "dgesdd");
-  TinyLinalg::GESDD<TinyLinalg::numo_cSFloatId, TinyLinalg::numo_cSFloatId, float, float, TinyLinalg::SGESDD>::define_module_function(rb_mTinyLinalgLapack, "sgesdd");
-  TinyLinalg::GESDD<TinyLinalg::numo_cDComplexId, TinyLinalg::numo_cDFloatId, lapack_complex_double, double, TinyLinalg::ZGESDD>::define_module_function(rb_mTinyLinalgLapack, "zgesdd");
-  TinyLinalg::GESDD<TinyLinalg::numo_cSComplexId, TinyLinalg::numo_cSFloatId, lapack_complex_float, float, TinyLinalg::CGESDD>::define_module_function(rb_mTinyLinalgLapack, "cgesdd");
-  TinyLinalg::GETRF<TinyLinalg::numo_cDFloatId, double, TinyLinalg::DGETRF>::define_module_function(rb_mTinyLinalgLapack, "dgetrf");
-  TinyLinalg::GETRF<TinyLinalg::numo_cSFloatId, float, TinyLinalg::SGETRF>::define_module_function(rb_mTinyLinalgLapack, "sgetrf");
-  TinyLinalg::GETRF<TinyLinalg::numo_cDComplexId, lapack_complex_double, TinyLinalg::ZGETRF>::define_module_function(rb_mTinyLinalgLapack, "zgetrf");
-  TinyLinalg::GETRF<TinyLinalg::numo_cSComplexId, lapack_complex_float, TinyLinalg::CGETRF>::define_module_function(rb_mTinyLinalgLapack, "cgetrf");
-  TinyLinalg::GETRI<TinyLinalg::numo_cDFloatId, double, TinyLinalg::DGETRI>::define_module_function(rb_mTinyLinalgLapack, "dgetri");
-  TinyLinalg::GETRI<TinyLinalg::numo_cSFloatId, float, TinyLinalg::SGETRI>::define_module_function(rb_mTinyLinalgLapack, "sgetri");
-  TinyLinalg::GETRI<TinyLinalg::numo_cDComplexId, lapack_complex_double, TinyLinalg::ZGETRI>::define_module_function(rb_mTinyLinalgLapack, "zgetri");
-  TinyLinalg::GETRI<TinyLinalg::numo_cSComplexId, lapack_complex_float, TinyLinalg::CGETRI>::define_module_function(rb_mTinyLinalgLapack, "cgetri");
+  TinyLinalg::GeSv<TinyLinalg::numo_cDFloatId, double, TinyLinalg::DGeSv>::define_module_function(rb_mTinyLinalgLapack, "dgesv");
+  TinyLinalg::GeSv<TinyLinalg::numo_cSFloatId, float, TinyLinalg::SGeSv>::define_module_function(rb_mTinyLinalgLapack, "sgesv");
+  TinyLinalg::GeSv<TinyLinalg::numo_cDComplexId, lapack_complex_double, TinyLinalg::ZGeSv>::define_module_function(rb_mTinyLinalgLapack, "zgesv");
+  TinyLinalg::GeSv<TinyLinalg::numo_cSComplexId, lapack_complex_float, TinyLinalg::CGeSv>::define_module_function(rb_mTinyLinalgLapack, "cgesv");
+  TinyLinalg::GeSvd<TinyLinalg::numo_cDFloatId, TinyLinalg::numo_cDFloatId, double, double, TinyLinalg::DGeSvd>::define_module_function(rb_mTinyLinalgLapack, "dgesvd");
+  TinyLinalg::GeSvd<TinyLinalg::numo_cSFloatId, TinyLinalg::numo_cSFloatId, float, float, TinyLinalg::SGeSvd>::define_module_function(rb_mTinyLinalgLapack, "sgesvd");
+  TinyLinalg::GeSvd<TinyLinalg::numo_cDComplexId, TinyLinalg::numo_cDFloatId, lapack_complex_double, double, TinyLinalg::ZGeSvd>::define_module_function(rb_mTinyLinalgLapack, "zgesvd");
+  TinyLinalg::GeSvd<TinyLinalg::numo_cSComplexId, TinyLinalg::numo_cSFloatId, lapack_complex_float, float, TinyLinalg::CGeSvd>::define_module_function(rb_mTinyLinalgLapack, "cgesvd");
+  TinyLinalg::GeSdd<TinyLinalg::numo_cDFloatId, TinyLinalg::numo_cDFloatId, double, double, TinyLinalg::DGeSdd>::define_module_function(rb_mTinyLinalgLapack, "dgesdd");
+  TinyLinalg::GeSdd<TinyLinalg::numo_cSFloatId, TinyLinalg::numo_cSFloatId, float, float, TinyLinalg::SGeSdd>::define_module_function(rb_mTinyLinalgLapack, "sgesdd");
+  TinyLinalg::GeSdd<TinyLinalg::numo_cDComplexId, TinyLinalg::numo_cDFloatId, lapack_complex_double, double, TinyLinalg::ZGeSdd>::define_module_function(rb_mTinyLinalgLapack, "zgesdd");
+  TinyLinalg::GeSdd<TinyLinalg::numo_cSComplexId, TinyLinalg::numo_cSFloatId, lapack_complex_float, float, TinyLinalg::CGeSdd>::define_module_function(rb_mTinyLinalgLapack, "cgesdd");
+  TinyLinalg::GeTrf<TinyLinalg::numo_cDFloatId, double, TinyLinalg::DGeTrf>::define_module_function(rb_mTinyLinalgLapack, "dgetrf");
+  TinyLinalg::GeTrf<TinyLinalg::numo_cSFloatId, float, TinyLinalg::SGeTrf>::define_module_function(rb_mTinyLinalgLapack, "sgetrf");
+  TinyLinalg::GeTrf<TinyLinalg::numo_cDComplexId, lapack_complex_double, TinyLinalg::ZGeTrf>::define_module_function(rb_mTinyLinalgLapack, "zgetrf");
+  TinyLinalg::GeTrf<TinyLinalg::numo_cSComplexId, lapack_complex_float, TinyLinalg::CGeTrf>::define_module_function(rb_mTinyLinalgLapack, "cgetrf");
+  TinyLinalg::GeTri<TinyLinalg::numo_cDFloatId, double, TinyLinalg::DGeTri>::define_module_function(rb_mTinyLinalgLapack, "dgetri");
+  TinyLinalg::GeTri<TinyLinalg::numo_cSFloatId, float, TinyLinalg::SGeTri>::define_module_function(rb_mTinyLinalgLapack, "sgetri");
+  TinyLinalg::GeTri<TinyLinalg::numo_cDComplexId, lapack_complex_double, TinyLinalg::ZGeTri>::define_module_function(rb_mTinyLinalgLapack, "zgetri");
+  TinyLinalg::GeTri<TinyLinalg::numo_cSComplexId, lapack_complex_float, TinyLinalg::CGeTri>::define_module_function(rb_mTinyLinalgLapack, "cgetri");
   TinyLinalg::GeQrf<TinyLinalg::numo_cDFloatId, double, TinyLinalg::DGeQrf>::define_module_function(rb_mTinyLinalgLapack, "dgeqrf");
   TinyLinalg::GeQrf<TinyLinalg::numo_cSFloatId, float, TinyLinalg::SGeQrf>::define_module_function(rb_mTinyLinalgLapack, "sgeqrf");
   TinyLinalg::GeQrf<TinyLinalg::numo_cDComplexId, lapack_complex_double, TinyLinalg::ZGeQrf>::define_module_function(rb_mTinyLinalgLapack, "zgeqrf");

data/ext/numo/tiny_linalg/tiny_linalg.hpp

@@ -1,17 +1,41 @@
+/**
+ * Copyright (c) 2023 Atsushi Tatsuma
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * * Redistributions of source code must retain the above copyright notice, this
+ *   list of conditions and the following disclaimer.
+ *
+ * * Redistributions in binary form must reproduce the above copyright notice,
+ *   this list of conditions and the following disclaimer in the documentation
+ *   and/or other materials provided with the distribution.
+ *
+ * * Neither the name of the copyright holder nor the names of its
+ *   contributors may be used to endorse or promote products derived from
+ *   this software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
+ * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
+ * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
+ * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
 #ifndef NUMO_TINY_LINALG_H
 #define NUMO_TINY_LINALG_H 1
 
-#if defined(TINYLINALG_USE_ACCELERATE)
-#include <Accelerate/Accelerate.h>
-#else
 #include <cblas.h>
 #include <lapacke.h>
-#endif
 
 #include <string>
 
-#include <cstring>
-
 #include <ruby.h>
 
 #include <numo/narray.h>

data/ext/numo/tiny_linalg/util.hpp

@@ -0,0 +1,100 @@
+namespace TinyLinalg {
+
+class Util {
+public:
+  static lapack_int get_itype(VALUE val) {
+    const lapack_int itype = NUM2INT(val);
+
+    if (itype != 1 && itype != 2 && itype != 3) {
+      rb_raise(rb_eArgError, "itype must be 1, 2 or 3");
+    }
+
+    return itype;
+  }
+
+  static char get_jobz(VALUE val) {
+    const char jobz = NUM2CHR(val);
+
+    if (jobz != 'N' && jobz != 'V') {
+      rb_raise(rb_eArgError, "jobz must be 'N' or 'V'");
+    }
+
+    return jobz;
+  }
+
+  static char get_range(VALUE val) {
+    const char range = NUM2CHR(val);
+
+    if (range != 'A' && range != 'V' && range != 'I') {
+      rb_raise(rb_eArgError, "range must be 'A', 'V' or 'I'");
+    }
+
+    return range;
+  }
+
+  static char get_uplo(VALUE val) {
+    const char uplo = NUM2CHR(val);
+
+    if (uplo != 'U' && uplo != 'L') {
+      rb_raise(rb_eArgError, "uplo must be 'U' or 'L'");
+    }
+
+    return uplo;
+  }
+
+  static int get_matrix_layout(VALUE val) {
+    const char option = NUM2CHR(val);
+
+    switch (option) {
+    case 'r':
+    case 'R':
+      break;
+    case 'c':
+    case 'C':
+      rb_warn("Numo::TinyLinalg does not support column major.");
+      break;
+    }
+
+    return LAPACK_ROW_MAJOR;
+  }
+
+  static enum CBLAS_TRANSPOSE get_cblas_trans(VALUE val) {
+    const char option = NUM2CHR(val);
+    enum CBLAS_TRANSPOSE res = CblasNoTrans;
+
+    switch (option) {
+    case 'n':
+    case 'N':
+      res = CblasNoTrans;
+      break;
+    case 't':
+    case 'T':
+      res = CblasTrans;
+      break;
+    case 'c':
+    case 'C':
+      res = CblasConjTrans;
+      break;
+    }
+
+    return res;
+  }
+
+  static enum CBLAS_ORDER get_cblas_order(VALUE val) {
+    const char option = NUM2CHR(val);
+
+    switch (option) {
+    case 'r':
+    case 'R':
+      break;
+    case 'c':
+    case 'C':
+      rb_warn("Numo::TinyLinalg does not support column major.");
+      break;
+    }
+
+    return CblasRowMajor;
+  }
+};
+
+} // namespace TinyLinalg

data/lib/numo/tiny_linalg/version.rb

@@ -5,6 +5,6 @@ module Numo
   # Numo::TinyLinalg is a subset library from Numo::Linalg consisting only of methods used in Machine Learning algorithms.
   module TinyLinalg
     # The version of Numo::TinyLinalg you install.
-    VERSION = '0.0.4'
+    VERSION = '0.1.1'
   end
 end