numo-linalg-alt 0.2.0

data/ext/numo/linalg/linalg.h

@@ -0,0 +1,85 @@
+/**
+ * Copyright (c) 2025 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_LINALG_ALT_LINALG_H
+#define NUMO_LINALG_ALT_LINALG_H 1
+
+#include <string.h>
+
+#include <ruby.h>
+
+#include <cblas.h>
+#include <lapacke.h>
+#include <openblas_config.h>
+
+#include <numo/narray.h>
+#include <numo/template.h>
+
+#include "converter.h"
+#include "util.h"
+
+#include "blas/dot.h"
+#include "blas/dot_sub.h"
+#include "blas/gemm.h"
+#include "blas/gemv.h"
+#include "blas/nrm2.h"
+
+#include "lapack/geev.h"
+#include "lapack/gelsd.h"
+#include "lapack/geqrf.h"
+#include "lapack/gesdd.h"
+#include "lapack/gesv.h"
+#include "lapack/gesvd.h"
+#include "lapack/getrf.h"
+#include "lapack/getri.h"
+#include "lapack/getrs.h"
+#include "lapack/heev.h"
+#include "lapack/heevd.h"
+#include "lapack/heevr.h"
+#include "lapack/hegv.h"
+#include "lapack/hegvd.h"
+#include "lapack/hegvx.h"
+#include "lapack/hetrf.h"
+#include "lapack/lange.h"
+#include "lapack/orgqr.h"
+#include "lapack/potrf.h"
+#include "lapack/potri.h"
+#include "lapack/potrs.h"
+#include "lapack/syev.h"
+#include "lapack/syevd.h"
+#include "lapack/syevr.h"
+#include "lapack/sygv.h"
+#include "lapack/sygvd.h"
+#include "lapack/sygvx.h"
+#include "lapack/sytrf.h"
+#include "lapack/trtrs.h"
+#include "lapack/ungqr.h"
+
+#endif /* NUMO_LINALG_ALT_LINALG_H */

data/ext/numo/linalg/util.c

@@ -0,0 +1,95 @@
+#include "util.h"
+
+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;
+}
+
+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;
+}
+
+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;
+}
+
+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;
+}
+
+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::Linalg does not support column major.");
+    break;
+  }
+
+  return LAPACK_ROW_MAJOR;
+}
+
+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;
+}
+
+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::Linalg does not support column major.");
+    break;
+  }
+
+  return CblasRowMajor;
+}

data/ext/numo/linalg/util.h

@@ -0,0 +1,17 @@
+#ifndef NUMO_LINALG_ALT_UTIL_H
+#define NUMO_LINALG_ALT_UTIL_H 1
+
+#include <ruby.h>
+
+#include <cblas.h>
+#include <lapacke.h>
+
+lapack_int get_itype(VALUE val);
+char get_jobz(VALUE val);
+char get_range(VALUE val);
+char get_uplo(VALUE val);
+int get_matrix_layout(VALUE val);
+enum CBLAS_TRANSPOSE get_cblas_trans(VALUE val);
+enum CBLAS_ORDER get_cblas_order(VALUE val);
+
+#endif // NUMO_LINALG_ALT_UTIL_H

data/lib/numo/linalg/version.rb

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+# Ruby/Numo (NUmerical MOdules)
+module Numo
+  # Numo::Linalg Alternative (numo-linalg-alt) is an alternative to Numo::Linalg.
+  module Linalg
+    # The version of numo-linalg-alt you install.
+    VERSION = '0.2.0'
+  end
+end