numo-tiny_linalg 0.2.0 → 0.3.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 911223ba40879d779bcc9bc77b840fb7b5e41ed68d6ccf7d26816913c42aed73
-  data.tar.gz: 9dce9c353da5466e1104c604c2749369131cb6640e14ddd33c697cee5d973563
+  metadata.gz: a1bff84bdd946b37325d2561e03141fec2ac75ac363e2ac640fc5d10d7f41950
+  data.tar.gz: 141adbf0789217c8d77c934e545ab8fae9a4c5178a94d6f4716f4a62c3106118
 SHA512:
-  metadata.gz: 9d666bdd04f48132005a9aee455047c80d83184815fac5c688ec593412ff8e670f1f31ba3376b1ee2985b8e829680590c1dad6a0ec3bf5366e584a82b523d8e4
-  data.tar.gz: 22027c54b56bad7214060ba7d5939b986be660a75afd65d74de7fdae095a6699ceea35422eee56831b621f2132e5bce921cb6b70bc9709e77e3efdf6cfc8a098
+  metadata.gz: 973e0075155ea849a9a7cc681d3e59242d19e6d649c8210329fdc21dae2aa07ac79495c4a32c56c580d819e37504f7064716f261aaa3d76c1ab2f5b42d414dfd
+  data.tar.gz: f2a9f4d7575e89f6a740955f74e13692fbd04844072ef0b7c519ce2ac4ce21bf3bdd32f72434c0dc0c1c6b358a0cbe279f101afdff139702b3574e9ab1d014c0

data/CHANGELOG.md

@@ -1,5 +1,16 @@
 ## [Unreleased]
 
+## [[0.3.1](https://github.com/yoshoku/numo-tiny_linalg/compare/v0.3.0...v0.3.1)] - 2023-08-15
+- Support automatic build of OpenBLAS on Windows.
+  - The author does not have a Windows PC. It will probably work.
+- Add OPENBLAS_VERSION constant to TinyLinalg.
+
+## [[0.3.0](https://github.com/yoshoku/numo-tiny_linalg/compare/v0.2.0...v0.3.0)] - 2023-08-13
+- Add cholesky and cho_solve module functions to TinyLinalg.
+
+**Breaking change**
+- Change to raise NotImplementedError when calling a method not yet implemented in Numo::TinyLinalg.
+
 ## [[0.2.0](https://github.com/yoshoku/numo-tiny_linalg/compare/v0.1.2...v0.2.0)] - 2023-08-11
 **Breaking change**
 - Change LAPACK function to call when array b is not given to TinyLinalg.eigh method.

data/README.md

@@ -6,7 +6,7 @@
 [![Documentation](https://img.shields.io/badge/api-reference-blue.svg)](https://yoshoku.github.io/numo-tiny_linalg/doc/)
 
 Numo::TinyLinalg is a subset library from [Numo::Linalg](https://github.com/ruby-numo/numo-linalg) consisting only of methods used in Machine Learning algorithms.
-The functions Numo::TinyLinalg supports are dot, det, eigh, inv, pinv, qr, solve, and svd.
+The functions Numo::TinyLinalg supports are dot, det, eigh, inv, pinv, qr, solve, cholesky, cho_solve and svd.
 
 Note that the version numbering rule of Numo::TinyLinalg is not compatible with that of Numo::Linalg.
 

data/ext/numo/tiny_linalg/extconf.rb

@@ -18,7 +18,9 @@ end
 
 abort 'numo/narray.h is not found' unless have_header('numo/narray.h')
 
-if RUBY_PLATFORM.match?(/mswin|cygwin|mingw/)
+on_windows = RUBY_PLATFORM.match?(/mswin|cygwin|mingw/)
+
+if on_windows
   $LOAD_PATH.each do |lp|
     if File.exist?(File.join(lp, 'numo/libnarray.a'))
       $LDFLAGS = "-L#{lp}/numo #{$LDFLAGS}"
@@ -31,13 +33,6 @@ end
 abort 'libstdc++ is not found.' unless have_library('stdc++')
 $CXXFLAGS << ' -std=c++11'
 
-# NOTE: Accelerate framework does not support LAPACKE.
-# use_accelerate = false
-# if RUBY_PLATFORM.include?('darwin') && have_framework('Accelerate')
-#   $CFLAGS << ' -DTINYLINALG_USE_ACCELERATE'
-#   use_accelerate = true
-# end
-
 build_openblas = false
 unless find_library('openblas', 'LAPACKE_dsyevr')
   build_openblas = true unless have_library('openblas')
@@ -51,6 +46,7 @@ if build_openblas
   warn 'BLAS and LAPACKE APIs are not found. Downloading and Building OpenBLAS...'
 
   VENDOR_DIR = File.expand_path("#{__dir__}/../../../vendor")
+  TINYLINALG_DIR = File.expand_path("#{__dir__}/../../../lib/numo/tiny_linalg")
   OPENBLAS_VER = '0.3.23'
   OPENBLAS_KEY = '115634b39007de71eb7e75cf7591dfb2'
   OPENBLAS_URI = "https://github.com/xianyi/OpenBLAS/archive/v#{OPENBLAS_VER}.tar.gz"
@@ -82,18 +78,19 @@ if build_openblas
       File.open("#{VENDOR_DIR}/tmp/openblas.log", 'a') { |f| f.puts(insstdout) }
       abort("Failed to install OpenBLAS. Check the openblas.log file for more details: #{VENDOR_DIR}/tmp/openblas.log") unless insstatus.success?
 
+      FileUtils.cp("#{VENDOR_DIR}/lib/libopenblas.a", TINYLINALG_DIR) if on_windows
+
       FileUtils.touch("#{VENDOR_DIR}/installed_#{OPENBLAS_VER}")
     end
   end
 
-  abort('libopenblas is not found.') unless find_library('openblas', nil, "#{VENDOR_DIR}/lib")
+  libopenblas_dir = on_windows ? TINYLINALG_DIR : "#{VENDOR_DIR}/lib"
+  abort('libopenblas is not found.') unless find_library('openblas', nil, libopenblas_dir)
   abort('openblas_config.h is not found.') unless find_header('openblas_config.h', nil, "#{VENDOR_DIR}/include")
   abort('cblas.h is not found.') unless find_header('cblas.h', nil, "#{VENDOR_DIR}/include")
   abort('lapacke.h is not found.') unless find_header('lapacke.h', nil, "#{VENDOR_DIR}/include")
 end
 
-$CFLAGS << ' -DNUMO_TINY_LINALG_USE_OPENBLAS'
-
 if RUBY_PLATFORM.include?('darwin') && Gem::Version.new('3.1.0') <= Gem::Version.new(RUBY_VERSION) &&
    try_link('int main(void){return 0;}', '-Wl,-undefined,dynamic_lookup')
   $LDFLAGS << ' -Wl,-undefined,dynamic_lookup'

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

@@ -0,0 +1,93 @@
+namespace TinyLinalg {
+
+struct DPoTrf {
+  lapack_int call(int matrix_layout, char uplo, lapack_int n, double* a, lapack_int lda) {
+    return LAPACKE_dpotrf(matrix_layout, uplo, n, a, lda);
+  }
+};
+
+struct SPoTrf {
+  lapack_int call(int matrix_layout, char uplo, lapack_int n, float* a, lapack_int lda) {
+    return LAPACKE_spotrf(matrix_layout, uplo, n, a, lda);
+  }
+};
+
+struct ZPoTrf {
+  lapack_int call(int matrix_layout, char uplo, lapack_int n, lapack_complex_double* a, lapack_int lda) {
+    return LAPACKE_zpotrf(matrix_layout, uplo, n, a, lda);
+  }
+};
+
+struct CPoTrf {
+  lapack_int call(int matrix_layout, char uplo, lapack_int n, lapack_complex_float* a, lapack_int lda) {
+    return LAPACKE_cpotrf(matrix_layout, uplo, n, a, lda);
+  }
+};
+
+template <int nary_dtype_id, typename dtype, class LapackFn>
+class PoTrf {
+public:
+  static void define_module_function(VALUE mLapack, const char* fnc_name) {
+    rb_define_module_function(mLapack, fnc_name, RUBY_METHOD_FUNC(tiny_linalg_potrf), -1);
+  }
+
+private:
+  struct potrf_opt {
+    int matrix_layout;
+    char uplo;
+  };
+
+  static void iter_potrf(na_loop_t* const lp) {
+    dtype* a = (dtype*)NDL_PTR(lp, 0);
+    int* info = (int*)NDL_PTR(lp, 1);
+    potrf_opt* opt = (potrf_opt*)(lp->opt_ptr);
+    const lapack_int n = NDL_SHAPE(lp, 0)[0];
+    const lapack_int lda = NDL_SHAPE(lp, 0)[1];
+    const lapack_int i = LapackFn().call(opt->matrix_layout, opt->uplo, n, a, lda);
+    *info = static_cast<int>(i);
+  }
+
+  static VALUE tiny_linalg_potrf(int argc, VALUE* argv, VALUE self) {
+    VALUE nary_dtype = NaryTypes[nary_dtype_id];
+
+    VALUE a_vnary = Qnil;
+    VALUE kw_args = Qnil;
+    rb_scan_args(argc, argv, "1:", &a_vnary, &kw_args);
+    ID kw_table[2] = { rb_intern("order"), rb_intern("uplo") };
+    VALUE kw_values[2] = { Qundef, Qundef };
+    rb_get_kwargs(kw_args, kw_table, 0, 2, kw_values);
+    const int matrix_layout = kw_values[0] != Qundef ? Util().get_matrix_layout(kw_values[0]) : LAPACK_ROW_MAJOR;
+    const char uplo = kw_values[1] != Qundef ? Util().get_uplo(kw_values[1]) : 'U';
+
+    if (CLASS_OF(a_vnary) != nary_dtype) {
+      a_vnary = rb_funcall(nary_dtype, rb_intern("cast"), 1, a_vnary);
+    }
+    if (!RTEST(nary_check_contiguous(a_vnary))) {
+      a_vnary = nary_dup(a_vnary);
+    }
+
+    narray_t* a_nary = NULL;
+    GetNArray(a_vnary, a_nary);
+    if (NA_NDIM(a_nary) != 2) {
+      rb_raise(rb_eArgError, "input array a must be 2-dimensional");
+      return Qnil;
+    }
+    if (NA_SHAPE(a_nary)[0] != NA_SHAPE(a_nary)[1]) {
+      rb_raise(rb_eArgError, "input array a must be square");
+      return Qnil;
+    }
+
+    ndfunc_arg_in_t ain[1] = { { OVERWRITE, 2 } };
+    ndfunc_arg_out_t aout[1] = { { numo_cInt32, 0 } };
+    ndfunc_t ndf = { iter_potrf, NO_LOOP | NDF_EXTRACT, 1, 1, ain, aout };
+    potrf_opt opt = { matrix_layout, uplo };
+    VALUE res = na_ndloop3(&ndf, &opt, 1, a_vnary);
+    VALUE ret = rb_ary_new3(2, a_vnary, res);
+
+    RB_GC_GUARD(a_vnary);
+
+    return ret;
+  }
+};
+
+} // namespace TinyLinalg

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

@@ -0,0 +1,121 @@
+namespace TinyLinalg {
+
+struct DPoTrs {
+  lapack_int call(int matrix_layout, char uplo, lapack_int n, lapack_int nrhs,
+                  const double* a, lapack_int lda, double* b, lapack_int ldb) {
+    return LAPACKE_dpotrs(matrix_layout, uplo, n, nrhs, a, lda, b, ldb);
+  }
+};
+
+struct SPoTrs {
+  lapack_int call(int matrix_layout, char uplo, lapack_int n, lapack_int nrhs,
+                  const float* a, lapack_int lda, float* b, lapack_int ldb) {
+    return LAPACKE_spotrs(matrix_layout, uplo, n, nrhs, a, lda, b, ldb);
+  }
+};
+
+struct ZPoTrs {
+  lapack_int call(int matrix_layout, char uplo, lapack_int n, lapack_int nrhs,
+                  const lapack_complex_double* a, lapack_int lda, lapack_complex_double* b, lapack_int ldb) {
+    return LAPACKE_zpotrs(matrix_layout, uplo, n, nrhs, a, lda, b, ldb);
+  }
+};
+
+struct CPoTrs {
+  lapack_int call(int matrix_layout, char uplo, lapack_int n, lapack_int nrhs,
+                  const lapack_complex_float* a, lapack_int lda, lapack_complex_float* b, lapack_int ldb) {
+    return LAPACKE_cpotrs(matrix_layout, uplo, n, nrhs, a, lda, b, ldb);
+  }
+};
+
+template <int nary_dtype_id, typename dtype, class LapackFn>
+class PoTrs {
+public:
+  static void define_module_function(VALUE mLapack, const char* fnc_name) {
+    rb_define_module_function(mLapack, fnc_name, RUBY_METHOD_FUNC(tiny_linalg_potrs), -1);
+  }
+
+private:
+  struct potrs_opt {
+    int matrix_layout;
+    char uplo;
+  };
+
+  static void iter_potrs(na_loop_t* const lp) {
+    dtype* a = (dtype*)NDL_PTR(lp, 0);
+    dtype* b = (dtype*)NDL_PTR(lp, 1);
+    int* info = (int*)NDL_PTR(lp, 2);
+    potrs_opt* opt = (potrs_opt*)(lp->opt_ptr);
+    const lapack_int n = NDL_SHAPE(lp, 0)[0];
+    const lapack_int nrhs = lp->args[1].ndim == 1 ? 1 : NDL_SHAPE(lp, 1)[1];
+    const lapack_int lda = n;
+    const lapack_int ldb = nrhs;
+    const lapack_int i = LapackFn().call(opt->matrix_layout, opt->uplo, n, nrhs, a, lda, b, ldb);
+    *info = static_cast<int>(i);
+  }
+
+  static VALUE tiny_linalg_potrs(int argc, VALUE* argv, VALUE self) {
+    VALUE nary_dtype = NaryTypes[nary_dtype_id];
+
+    VALUE a_vnary = Qnil;
+    VALUE b_vnary = Qnil;
+    VALUE kw_args = Qnil;
+    rb_scan_args(argc, argv, "2:", &a_vnary, &b_vnary, &kw_args);
+    ID kw_table[2] = { rb_intern("order"), rb_intern("uplo") };
+    VALUE kw_values[2] = { Qundef, Qundef };
+    rb_get_kwargs(kw_args, kw_table, 0, 2, kw_values);
+    const int matrix_layout = kw_values[0] != Qundef ? Util().get_matrix_layout(kw_values[0]) : LAPACK_ROW_MAJOR;
+    const char uplo = kw_values[1] != Qundef ? Util().get_uplo(kw_values[1]) : 'U';
+
+    if (CLASS_OF(a_vnary) != nary_dtype) {
+      a_vnary = rb_funcall(nary_dtype, rb_intern("cast"), 1, a_vnary);
+    }
+    if (!RTEST(nary_check_contiguous(a_vnary))) {
+      a_vnary = nary_dup(a_vnary);
+    }
+    if (CLASS_OF(b_vnary) != nary_dtype) {
+      b_vnary = rb_funcall(nary_dtype, rb_intern("cast"), 1, b_vnary);
+    }
+    if (!RTEST(nary_check_contiguous(b_vnary))) {
+      b_vnary = nary_dup(b_vnary);
+    }
+
+    narray_t* a_nary = NULL;
+    GetNArray(a_vnary, a_nary);
+    if (NA_NDIM(a_nary) != 2) {
+      rb_raise(rb_eArgError, "input array a must be 2-dimensional");
+      return Qnil;
+    }
+    if (NA_SHAPE(a_nary)[0] != NA_SHAPE(a_nary)[1]) {
+      rb_raise(rb_eArgError, "input array a must be square");
+      return Qnil;
+    }
+    narray_t* b_nary = NULL;
+    GetNArray(b_vnary, b_nary);
+    const int b_n_dims = NA_NDIM(b_nary);
+    if (b_n_dims != 1 && b_n_dims != 2) {
+      rb_raise(rb_eArgError, "input array b must be 1- or 2-dimensional");
+      return Qnil;
+    }
+
+    lapack_int n = NA_SHAPE(a_nary)[0];
+    lapack_int nb = NA_SHAPE(b_nary)[0];
+    if (n != nb) {
+      rb_raise(nary_eShapeError, "shape1[0](=%d) != shape2[0](=%d)", n, nb);
+    }
+
+    ndfunc_arg_in_t ain[2] = { { nary_dtype, 2 }, { OVERWRITE, b_n_dims } };
+    ndfunc_arg_out_t aout[1] = { { numo_cInt32, 0 } };
+    ndfunc_t ndf = { iter_potrs, NO_LOOP | NDF_EXTRACT, 2, 1, ain, aout };
+    potrs_opt opt = { matrix_layout, uplo };
+    VALUE res = na_ndloop3(&ndf, &opt, 2, a_vnary, b_vnary);
+    VALUE ret = rb_ary_new3(2, b_vnary, res);
+
+    RB_GC_GUARD(a_vnary);
+    RB_GC_GUARD(b_vnary);
+
+    return ret;
+  }
+};
+
+} // namespace TinyLinalg

data/ext/numo/tiny_linalg/tiny_linalg.cpp

@@ -51,6 +51,8 @@
 #include "lapack/hegvd.hpp"
 #include "lapack/hegvx.hpp"
 #include "lapack/orgqr.hpp"
+#include "lapack/potrf.hpp"
+#include "lapack/potrs.hpp"
 #include "lapack/syev.hpp"
 #include "lapack/syevd.hpp"
 #include "lapack/syevr.hpp"
@@ -246,6 +248,9 @@ extern "C" void Init_tiny_linalg(void) {
    */
   rb_mTinyLinalgLapack = rb_define_module_under(rb_mTinyLinalg, "Lapack");
 
+  /* The version of OpenBLAS used in background library. */
+  rb_define_const(rb_mTinyLinalg, "OPENBLAS_VERSION", rb_str_new_cstr(OPENBLAS_VERSION));
+
   /**
    * Returns BLAS char ([sdcz]) defined by data-type of arguments.
    *
@@ -310,6 +315,14 @@ extern "C" void Init_tiny_linalg(void) {
   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::PoTrf<TinyLinalg::numo_cDFloatId, double, TinyLinalg::DPoTrf>::define_module_function(rb_mTinyLinalgLapack, "dpotrf");
+  TinyLinalg::PoTrf<TinyLinalg::numo_cSFloatId, float, TinyLinalg::SPoTrf>::define_module_function(rb_mTinyLinalgLapack, "spotrf");
+  TinyLinalg::PoTrf<TinyLinalg::numo_cDComplexId, lapack_complex_double, TinyLinalg::ZPoTrf>::define_module_function(rb_mTinyLinalgLapack, "zpotrf");
+  TinyLinalg::PoTrf<TinyLinalg::numo_cSComplexId, lapack_complex_float, TinyLinalg::CPoTrf>::define_module_function(rb_mTinyLinalgLapack, "cpotrf");
+  TinyLinalg::PoTrs<TinyLinalg::numo_cDFloatId, double, TinyLinalg::DPoTrs>::define_module_function(rb_mTinyLinalgLapack, "dpotrs");
+  TinyLinalg::PoTrs<TinyLinalg::numo_cSFloatId, float, TinyLinalg::SPoTrs>::define_module_function(rb_mTinyLinalgLapack, "spotrs");
+  TinyLinalg::PoTrs<TinyLinalg::numo_cDComplexId, lapack_complex_double, TinyLinalg::ZPoTrs>::define_module_function(rb_mTinyLinalgLapack, "zpotrs");
+  TinyLinalg::PoTrs<TinyLinalg::numo_cSComplexId, lapack_complex_float, TinyLinalg::CPoTrs>::define_module_function(rb_mTinyLinalgLapack, "cpotrs");
   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

@@ -33,6 +33,7 @@
 
 #include <cblas.h>
 #include <lapacke.h>
+#include <openblas_config.h>
 
 #include <string>
 

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.2.0'
+    VERSION = '0.3.1'
   end
 end

data/lib/numo/tiny_linalg.rb

@@ -95,6 +95,96 @@ module Numo
       [vals, vecs]
     end
 
+    # Computes the Cholesky decomposition of a symmetric / Hermitian positive-definite matrix.
+    #
+    # @example
+    #   require 'numo/tiny_linalg'
+    #
+    #   Numo::Linalg = Numo::TinyLinalg unless defined?(Numo::Linalg)
+    #
+    #   s = Numo::DFloat.new(3, 3).rand - 0.5
+    #   a = s.transpose.dot(s)
+    #   u = Numo::Linalg.cholesky(a)
+    #
+    #   pp u
+    #   # =>
+    #   # Numo::DFloat#shape=[3,3]
+    #   # [[0.532006, 0.338183, -0.18036],
+    #   #  [0, 0.325153, 0.011721],
+    #   #  [0, 0, 0.436738]]
+    #
+    #   pp (a - u.transpose.dot(u)).abs.max
+    #   # => 1.3877787807814457e-17
+    #
+    #   l = Numo::Linalg.cholesky(a, uplo: 'L')
+    #
+    #   pp l
+    #   # =>
+    #   # Numo::DFloat#shape=[3,3]
+    #   # [[0.532006, 0, 0],
+    #   #  [0.338183, 0.325153, 0],
+    #   #  [-0.18036, 0.011721, 0.436738]]
+    #
+    #   pp (a - l.dot(l.transpose)).abs.max
+    #   # => 1.3877787807814457e-17
+    #
+    # @param a [Numo::NArray] The n-by-n symmetric matrix.
+    # @param uplo [String] Whether to compute the upper- or lower-triangular Cholesky factor ('U' or 'L').
+    # @return [Numo::NArray] The upper- or lower-triangular Cholesky factor of a.
+    def cholesky(a, uplo: 'U')
+      raise ArgumentError, 'input array a must be 2-dimensional' if a.ndim != 2
+      raise ArgumentError, 'input array a must be square' if a.shape[0] != a.shape[1]
+
+      bchr = blas_char(a)
+      raise ArgumentError, "invalid array type: #{a.class}" if bchr == 'n'
+
+      fnc = "#{bchr}potrf".to_sym
+      c, _info = Numo::TinyLinalg::Lapack.send(fnc, a.dup, uplo: uplo)
+
+      case uplo
+      when 'U'
+        c.triu
+      when 'L'
+        c.tril
+      else
+        raise ArgumentError, "invalid uplo: #{uplo}"
+      end
+    end
+
+    # Solves linear equation `A * x = b` or `A * X = B` for `x` with the Cholesky factorization of `A`.
+    #
+    # @example
+    #   require 'numo/tiny_linalg'
+    #
+    #   Numo::Linalg = Numo::TinyLinalg unless defined?(Numo::Linalg)
+    #
+    #   s = Numo::DFloat.new(3, 3).rand - 0.5
+    #   a = s.transpose.dot(s)
+    #   u = Numo::Linalg.cholesky(a)
+    #
+    #   b = Numo::DFloat.new(3).rand
+    #   x = Numo::Linalg.cho_solve(u, b)
+    #
+    #   puts (b - a.dot(x)).abs.max
+    #   => 0.0
+    #
+    # @param a [Numo::NArray] The n-by-n cholesky factor.
+    # @param b [Numo::NArray] The n right-hand side vector, or n-by-nrhs right-hand side matrix.
+    # @param uplo [String] Whether to compute the upper- or lower-triangular Cholesky factor ('U' or 'L').
+    # @return [Numo::NArray] The solution vector or matrix `X`.
+    def cho_solve(a, b, uplo: 'U')
+      raise ArgumentError, 'input array a must be 2-dimensional' if a.ndim != 2
+      raise ArgumentError, 'input array a must be square' if a.shape[0] != a.shape[1]
+      raise ArgumentError, "incompatible dimensions: a.shape[0] = #{a.shape[0]} != b.shape[0] = #{b.shape[0]}" if a.shape[0] != b.shape[0]
+
+      bchr = blas_char(a, b)
+      raise ArgumentError, "invalid array type: #{a.class}" if bchr == 'n'
+
+      fnc = "#{bchr}potrs".to_sym
+      x, _info = Numo::TinyLinalg::Lapack.send(fnc, a, b.dup, uplo: uplo)
+      x
+    end
+
     # Computes the determinant of matrix.
     #
     # @example
@@ -271,7 +361,7 @@ module Numo
       [q, r]
     end
 
-    # Solves linear equation `A * x = b` or `A * X = B` for `x` from square matrix `a`.
+    # Solves linear equation `A * x = b` or `A * X = B` for `x` from square matrix `A`.
     #
     # @example
     #   require 'numo/tiny_linalg'
@@ -294,10 +384,10 @@ module Numo
     #   # => 2.1081041547796492e-16
     #
     # @param a [Numo::NArray] The n-by-n square matrix.
-    # @param b [Numo::NArray] The n right-hand side vector, or n-by-nrhs right-hand side matrix (>= 1-dimensinal NArray).
+    # @param b [Numo::NArray] The n right-hand side vector, or n-by-nrhs right-hand side matrix.
     # @param driver [String] This argument is for compatibility with Numo::Linalg.solver, and is not used.
     # @param uplo [String] This argument is for compatibility with Numo::Linalg.solver, and is not used.
-    # @return [Numo::NArray] The solusion vector / matrix `x`.
+    # @return [Numo::NArray] The solusion vector / matrix `X`.
     def solve(a, b, driver: 'gen', uplo: 'U') # rubocop:disable Lint/UnusedMethodArgument
       raise ArgumentError, 'input array a must be 2-dimensional' if a.ndim != 2
       raise ArgumentError, 'input array a must be square' if a.shape[0] != a.shape[1]
@@ -368,5 +458,110 @@ module Numo
 
       [s, u, vt]
     end
+
+    # @!visibility private
+    def matmul(*args)
+      raise NotImplementedError, "#{__method__} is not yet implemented in Numo::TinyLinalg"
+    end
+
+    # @!visibility private
+    def matrix_power(*args)
+      raise NotImplementedError, "#{__method__} is not yet implemented in Numo::TinyLinalg"
+    end
+
+    # @!visibility private
+    def svdvals(*args)
+      raise NotImplementedError, "#{__method__} is not yet implemented in Numo::TinyLinalg"
+    end
+
+    # @!visibility private
+    def orth(*args)
+      raise NotImplementedError, "#{__method__} is not yet implemented in Numo::TinyLinalg"
+    end
+
+    # @!visibility private
+    def null_space(*args)
+      raise NotImplementedError, "#{__method__} is not yet implemented in Numo::TinyLinalg"
+    end
+
+    # @!visibility private
+    def lu(*args)
+      raise NotImplementedError, "#{__method__} is not yet implemented in Numo::TinyLinalg"
+    end
+
+    # @!visibility private
+    def lu_fact(*args)
+      raise NotImplementedError, "#{__method__} is not yet implemented in Numo::TinyLinalg"
+    end
+
+    # @!visibility private
+    def lu_inv(*args)
+      raise NotImplementedError, "#{__method__} is not yet implemented in Numo::TinyLinalg"
+    end
+
+    # @!visibility private
+    def lu_solve(*args)
+      raise NotImplementedError, "#{__method__} is not yet implemented in Numo::TinyLinalg"
+    end
+
+    # @!visibility private
+    def ldl(*args)
+      raise NotImplementedError, "#{__method__} is not yet implemented in Numo::TinyLinalg"
+    end
+
+    # @!visibility private
+    def cho_fact(*args)
+      raise NotImplementedError, "#{__method__} is not yet implemented in Numo::TinyLinalg"
+    end
+
+    # @!visibility private
+    def cho_inv(*args)
+      raise NotImplementedError, "#{__method__} is not yet implemented in Numo::TinyLinalg"
+    end
+
+    # @!visibility private
+    def eig(*args)
+      raise NotImplementedError, "#{__method__} is not yet implemented in Numo::TinyLinalg"
+    end
+
+    # @!visibility private
+    def eigvals(*args)
+      raise NotImplementedError, "#{__method__} is not yet implemented in Numo::TinyLinalg"
+    end
+
+    # @!visibility private
+    def eigvalsh(*args)
+      raise NotImplementedError, "#{__method__} is not yet implemented in Numo::TinyLinalg"
+    end
+
+    # @!visibility private
+    def norm(*args)
+      raise NotImplementedError, "#{__method__} is not yet implemented in Numo::TinyLinalg"
+    end
+
+    # @!visibility private
+    def cond(*args)
+      raise NotImplementedError, "#{__method__} is not yet implemented in Numo::TinyLinalg"
+    end
+
+    # @!visibility private
+    def slogdet(*args)
+      raise NotImplementedError, "#{__method__} is not yet implemented in Numo::TinyLinalg"
+    end
+
+    # @!visibility private
+    def matrix_rank(*args)
+      raise NotImplementedError, "#{__method__} is not yet implemented in Numo::TinyLinalg"
+    end
+
+    # @!visibility private
+    def lstsq(*args)
+      raise NotImplementedError, "#{__method__} is not yet implemented in Numo::TinyLinalg"
+    end
+
+    # @!visibility private
+    def expm(*args)
+      raise NotImplementedError, "#{__method__} is not yet implemented in Numo::TinyLinalg"
+    end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: numo-tiny_linalg
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.1
 platform: ruby
 authors:
 - yoshoku
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-08-10 00:00:00.000000000 Z
+date: 2023-08-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: numo-narray
@@ -59,6 +59,8 @@ files:
 - ext/numo/tiny_linalg/lapack/hegvd.hpp
 - ext/numo/tiny_linalg/lapack/hegvx.hpp
 - ext/numo/tiny_linalg/lapack/orgqr.hpp
+- ext/numo/tiny_linalg/lapack/potrf.hpp
+- ext/numo/tiny_linalg/lapack/potrs.hpp
 - ext/numo/tiny_linalg/lapack/syev.hpp
 - ext/numo/tiny_linalg/lapack/syevd.hpp
 - ext/numo/tiny_linalg/lapack/syevr.hpp