lbfgsb 0.4.1 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 771754be56f6a3954337488ed44d237a9d97899c28800916d89ff134e8d2d12d
-  data.tar.gz: 7db51dd7443056af4e174199752ba7ef136c7898bb9bd086b5316a784fb0a70f
+  metadata.gz: d0558e429ff4c590e681165e0af885de04e9b78d4d679afc89f9c0bcfc1b2764
+  data.tar.gz: 5e1a8a157614da1a8541fbe28aad8cb04bf4096a4d3022f1b519c35abf071544
 SHA512:
-  metadata.gz: a1eb043cbdd14c0db2d9bd1a74b934d1719a95ee943281f915a11dbff508cdaf55c501d6ba250cec5569b405ff0ab4702124c1388cc6e14bd7df229727ffbfcf
-  data.tar.gz: db4f459a6b3711415c78d8a94ff0bbc4039581f7923e0bbabd833e2222a0e5462979b2319f3284725411e6542d84f4d319003cf685ccc9c9d4d27b909d79a443
+  metadata.gz: 66997d33a6fd0fb750ae236934fa866a94592161cbba49375bd7d46d94a9c98f0177cd207431d24cd1812bbf10be050106c941de9ac972f7699d0687b97a41c8
+  data.tar.gz: d8ba46b919591239180642f9a45fe5f14337f2aa703ac527ba533449c647be4b64de700a49a88616019241010d37e6118904c992c3886ed7e44c620f771cadc9

data/CHANGELOG.md

@@ -1,3 +1,19 @@
+## 0.5.0
+- Add build option to select FORTRAN integer bit size.
+
+```
+$ gem install lbfgsb -- --with-use-int64
+```
+
+- Add build option to use any blas library.
+
+```
+$ gem install lbfgsb -- --with-blas-dir=/opt/local/openblas/lib --with-blas-lib=openblas
+```
+
+- Change to use 32-bit integer for FORTRAN integer in internal functions by default.
+- Introduce conventional commits.
+
 ## 0.4.1
 - Remove dependent gem's type declaration file from installation files.
 

data/LICENSE.txt

@@ -1,4 +1,4 @@
-Copyright (c) 2020-2021 Atsushi Tatsuma
+Copyright (c) 2020-2022 Atsushi Tatsuma
 All rights reserved.
 
 Redistribution and use in source and binary forms, with or without

data/ext/lbfgsb/extconf.rb

@@ -1,6 +1,8 @@
 require 'mkmf'
 require 'numo/narray'
 
+$defs << '-DUSE_INT64' if with_config('use-int64', false)
+
 $LOAD_PATH.each do |lp|
   if File.exist?(File.join(lp, 'numo/numo/narray.h'))
     $INCFLAGS = "-I#{lp}/numo #{$INCFLAGS}"
@@ -26,10 +28,17 @@ if RUBY_PLATFORM =~ /mswin|cygwin|mingw/
   end
 end
 
-$srcs = Dir.glob("#{$srcdir}/*.c").map { |path| File.basename(path) }
-$srcs.concat(%w[blas.c linpack.c lbfgsb.c])
+$srcs = Dir.glob("#{$srcdir}/**/*.c").map { |path| File.basename(path) }
+
+blas_dir = with_config('blas-dir')
+$LDFLAGS = "-L#{blas_dir} #{$LDFLAGS}" unless blas_dir.nil?
+
+blas_lib = with_config('blas-lib')
+unless blas_lib.nil?
+  abort "#{blas_lib} not found." unless have_library(blas_lib)
+  $srcs.delete('blas.c')
+end
 
-$INCFLAGS << " -I$(srcdir)/src"
 $VPATH << "$(srcdir)/src"
 
 create_makefile('lbfgsb/lbfgsbext')

data/ext/lbfgsb/lbfgsbext.c

@@ -4,31 +4,43 @@ VALUE rb_mLbfgsb;
 
 static VALUE lbfgsb_min_l_bfgs_b(VALUE self, VALUE fnc, VALUE x_val, VALUE jcb, VALUE args, VALUE l_val, VALUE u_val,
                                  VALUE nbd_val, VALUE maxcor, VALUE ftol, VALUE gtol, VALUE maxiter, VALUE disp) {
-  long n_iter;
-  long n_fev;
-  long n_jev;
-  long max_iter = NUM2LONG(maxiter);
+  F77_int n_iter;
+  F77_int n_fev;
+  F77_int n_jev;
+#ifdef USE_INT64
+  F77_int max_iter = NUM2LONG(maxiter);
+#else
+  F77_int max_iter = NUM2INT(maxiter);
+#endif
   narray_t* x_nary;
   narray_t* l_nary;
   narray_t* u_nary;
   narray_t* nbd_nary;
-  long n;
-  long m = NUM2LONG(maxcor);
+  F77_int n;
+#ifdef USE_INT64
+  F77_int m = NUM2LONG(maxcor);
+#else
+  F77_int m = NUM2INT(maxcor);
+#endif
   double* x_ptr;
   double* l_ptr;
   double* u_ptr;
-  long* nbd_ptr;
+  F77_int* nbd_ptr;
   double f;
   double* g;
   double factr = NUM2DBL(ftol);
   double pgtol = NUM2DBL(gtol);
   double* wa;
-  long* iwa;
+  F77_int* iwa;
   char task[60];
-  long iprint = NIL_P(disp) ? -1 : NUM2LONG(disp);
+#ifdef USE_INT64
+  F77_int iprint = NIL_P(disp) ? -1 : NUM2LONG(disp);
+#else
+  F77_int iprint = NIL_P(disp) ? -1 : NUM2INT(disp);
+#endif
   char csave[60];
-  long lsave[4];
-  long isave[44];
+  F77_int lsave[4];
+  F77_int isave[44];
   double dsave[29];
   VALUE g_val;
   VALUE fg_arr;
@@ -39,7 +51,7 @@ static VALUE lbfgsb_min_l_bfgs_b(VALUE self, VALUE fnc, VALUE x_val, VALUE jcb,
     rb_raise(rb_eArgError, "x must be a 1-D array.");
     return Qnil;
   }
-  n = (long)NA_SIZE(x_nary);
+  n = (F77_int)NA_SIZE(x_nary);
   if (CLASS_OF(x_val) != numo_cDFloat) {
     x_val = rb_funcall(numo_cDFloat, rb_intern("cast"), 1, x_val);
   }
@@ -52,7 +64,7 @@ static VALUE lbfgsb_min_l_bfgs_b(VALUE self, VALUE fnc, VALUE x_val, VALUE jcb,
     rb_raise(rb_eArgError, "l must be a 1-D array.");
     return Qnil;
   }
-  if ((long)NA_SIZE(l_nary) != n) {
+  if ((F77_int)NA_SIZE(l_nary) != n) {
     rb_raise(rb_eArgError, "The size of l must be equal to that of x.");
     return Qnil;
   }
@@ -68,7 +80,7 @@ static VALUE lbfgsb_min_l_bfgs_b(VALUE self, VALUE fnc, VALUE x_val, VALUE jcb,
     rb_raise(rb_eArgError, "u must be a 1-D array.");
     return Qnil;
   }
-  if ((long)NA_SIZE(u_nary) != n) {
+  if ((F77_int)NA_SIZE(u_nary) != n) {
     rb_raise(rb_eArgError, "The size of u must be equal to that of x.");
     return Qnil;
   }
@@ -84,13 +96,19 @@ static VALUE lbfgsb_min_l_bfgs_b(VALUE self, VALUE fnc, VALUE x_val, VALUE jcb,
     rb_raise(rb_eArgError, "nbd must be a 1-D array.");
     return Qnil;
   }
-  if ((long)NA_SIZE(nbd_nary) != n) {
+  if ((F77_int)NA_SIZE(nbd_nary) != n) {
     rb_raise(rb_eArgError, "The size of nbd must be equal to that of x.");
     return Qnil;
   }
+#ifdef USE_INT64
   if (CLASS_OF(nbd_val) != numo_cInt64) {
     nbd_val = rb_funcall(numo_cInt64, rb_intern("cast"), 1, nbd_val);
   }
+#else
+  if (CLASS_OF(nbd_val) != numo_cInt32) {
+    nbd_val = rb_funcall(numo_cInt32, rb_intern("cast"), 1, nbd_val);
+  }
+#endif
   if (!RTEST(nary_check_contiguous(nbd_val))) {
     nbd_val = nary_dup(nbd_val);
   }
@@ -98,10 +116,10 @@ static VALUE lbfgsb_min_l_bfgs_b(VALUE self, VALUE fnc, VALUE x_val, VALUE jcb,
   x_ptr = (double*)na_get_pointer_for_read_write(x_val);
   l_ptr = (double*)na_get_pointer_for_read(l_val);
   u_ptr = (double*)na_get_pointer_for_read(u_val);
-  nbd_ptr = (long*)na_get_pointer_for_read(nbd_val);
+  nbd_ptr = (F77_int*)na_get_pointer_for_read(nbd_val);
   g = ALLOC_N(double, n);
   wa = ALLOC_N(double, (2 * m + 5) * n + 12 * m * m + 12 * m);
-  iwa = ALLOC_N(long, 3 * n);
+  iwa = ALLOC_N(F77_int, 3 * n);
 
   g_val = Qnil;
   f = 0.0;
@@ -145,9 +163,15 @@ static VALUE lbfgsb_min_l_bfgs_b(VALUE self, VALUE fnc, VALUE x_val, VALUE jcb,
   rb_hash_aset(ret, ID2SYM(rb_intern("x")), x_val);
   rb_hash_aset(ret, ID2SYM(rb_intern("fnc")), DBL2NUM(f));
   rb_hash_aset(ret, ID2SYM(rb_intern("jcb")), g_val);
+#ifdef USE_INT64
   rb_hash_aset(ret, ID2SYM(rb_intern("n_iter")), LONG2NUM(n_iter));
   rb_hash_aset(ret, ID2SYM(rb_intern("n_fev")), LONG2NUM(n_fev));
   rb_hash_aset(ret, ID2SYM(rb_intern("n_jev")), LONG2NUM(n_jev));
+#else
+  rb_hash_aset(ret, ID2SYM(rb_intern("n_iter")), INT2NUM(n_iter));
+  rb_hash_aset(ret, ID2SYM(rb_intern("n_fev")), INT2NUM(n_fev));
+  rb_hash_aset(ret, ID2SYM(rb_intern("n_jev")), INT2NUM(n_jev));
+#endif
   rb_hash_aset(ret, ID2SYM(rb_intern("success")), strncmp(task, "CONV", 4) == 0 ? Qtrue : Qfalse);
 
   RB_GC_GUARD(x_val);
@@ -162,6 +186,13 @@ void Init_lbfgsbext(void) {
   rb_mLbfgsb = rb_define_module("Lbfgsb");
   /* The value of double epsilon used in the native extension. */
   rb_define_const(rb_mLbfgsb, "DBL_EPSILON", DBL2NUM(DBL_EPSILON));
+#ifdef USE_INT64
+  /* The bit size of fortran integer. */
+  rb_define_const(rb_mLbfgsb, "SZ_F77_INTEGER", INT2NUM(64));
+#else
+  /* The bit size of fortran integer. */
+  rb_define_const(rb_mLbfgsb, "SZ_F77_INTEGER", INT2NUM(32));
+#endif
   /* @!visibility private */
   rb_define_module_function(rb_mLbfgsb, "min_l_bfgs_b", lbfgsb_min_l_bfgs_b, 12);
 }

data/ext/lbfgsb/lbfgsbext.h

@@ -9,6 +9,6 @@
 #include <numo/narray.h>
 #include <numo/template.h>
 
-#include <lbfgsb.h>
+#include "src/lbfgsb.h"
 
 #endif /* LBFGSB_RB_H */

data/ext/lbfgsb/src/blas.c

@@ -5,9 +5,9 @@
  */
 #include "blas.h"
 
-int lbfgsb_rb_daxpy_(long* n, double* da, double* dx, long* incx, double* dy, long* incy) {
-  long i__1;
-  static long i__, m, ix, iy, mp1;
+void daxpy_(F77_int* n, double* da, double* dx, F77_int* incx, double* dy, F77_int* incy) {
+  F77_int i__1;
+  static F77_int i__, m, ix, iy, mp1;
 
   --dy;
   --dx;
@@ -16,10 +16,10 @@ int lbfgsb_rb_daxpy_(long* n, double* da, double* dx, long* incx, double* dy, lo
   /* uses unrolled loops for increments equal to one. */
   /* jack dongarra, linpack, 3/11/78. */
   if (*n <= 0) {
-    return 0;
+    return;
   }
   if (*da == 0.) {
-    return 0;
+    return;
   }
   if (*incx == 1 && *incy == 1) {
     goto L20;
@@ -41,7 +41,7 @@ int lbfgsb_rb_daxpy_(long* n, double* da, double* dx, long* incx, double* dy, lo
     ix += *incx;
     iy += *incy;
   }
-  return 0;
+  return;
 
   /* code for both increments equal to 1 */
   /* clean-up loop */
@@ -55,7 +55,7 @@ L20:
     dy[i__] += *da * dx[i__];
   }
   if (*n < 4) {
-    return 0;
+    return;
   }
 L40:
   mp1 = m + 1;
@@ -66,12 +66,12 @@ L40:
     dy[i__ + 2] += *da * dx[i__ + 2];
     dy[i__ + 3] += *da * dx[i__ + 3];
   }
-  return 0;
+  return;
 }
 
-int lbfgsb_rb_dcopy_(long* n, double* dx, long* incx, double* dy, long* incy) {
-  long i__1;
-  static long i__, m, ix, iy, mp1;
+void dcopy_(F77_int* n, double* dx, F77_int* incx, double* dy, F77_int* incy) {
+  F77_int i__1;
+  static F77_int i__, m, ix, iy, mp1;
 
   --dy;
   --dx;
@@ -80,7 +80,7 @@ int lbfgsb_rb_dcopy_(long* n, double* dx, long* incx, double* dy, long* incy) {
   /* uses unrolled loops for increments equal to one. */
   /* jack dongarra, linpack, 3/11/78. */
   if (*n <= 0) {
-    return 0;
+    return;
   }
   if (*incx == 1 && *incy == 1) {
     goto L20;
@@ -102,7 +102,7 @@ int lbfgsb_rb_dcopy_(long* n, double* dx, long* incx, double* dy, long* incy) {
     ix += *incx;
     iy += *incy;
   }
-  return 0;
+  return;
 
   /* code for both increments equal to 1 */
   /* clean-up loop */
@@ -116,7 +116,7 @@ L20:
     dy[i__] = dx[i__];
   }
   if (*n < 7) {
-    return 0;
+    return;
   }
 L40:
   mp1 = m + 1;
@@ -130,13 +130,13 @@ L40:
     dy[i__ + 5] = dx[i__ + 5];
     dy[i__ + 6] = dx[i__ + 6];
   }
-  return 0;
+  return;
 }
 
-double lbfgsb_rb_ddot_(long* n, double* dx, long* incx, double* dy, long* incy) {
-  long i__1;
+double ddot_(F77_int* n, double* dx, F77_int* incx, double* dy, F77_int* incy) {
+  F77_int i__1;
   double ret_val;
-  static long i__, m, ix, iy, mp1;
+  static F77_int i__, m, ix, iy, mp1;
   static double dtemp;
 
   --dy;
@@ -199,9 +199,9 @@ L60:
   return ret_val;
 }
 
-int lbfgsb_rb_dscal_(long* n, double* da, double* dx, long* incx) {
-  long i__1, i__2;
-  static long i__, m, mp1, nincx;
+void dscal_(F77_int* n, double* da, double* dx, F77_int* incx) {
+  F77_int i__1, i__2;
+  static F77_int i__, m, mp1, nincx;
 
   --dx;
 
@@ -210,7 +210,7 @@ int lbfgsb_rb_dscal_(long* n, double* da, double* dx, long* incx) {
   /* jack dongarra, linpack, 3/11/78. */
   /* modified 3/93 to return if incx .le. 0. */
   if (*n <= 0 || *incx <= 0) {
-    return 0;
+    return;
   }
   if (*incx == 1) {
     goto L20;
@@ -223,7 +223,7 @@ int lbfgsb_rb_dscal_(long* n, double* da, double* dx, long* incx) {
   for (i__ = 1; i__2 < 0 ? i__ >= i__1 : i__ <= i__1; i__ += i__2) {
     dx[i__] = *da * dx[i__];
   }
-  return 0;
+  return;
 
   /* code for increment equal to 1 */
   /* clean-up loop */
@@ -237,7 +237,7 @@ L20:
     dx[i__] = *da * dx[i__];
   }
   if (*n < 5) {
-    return 0;
+    return;
   }
 L40:
   mp1 = m + 1;
@@ -249,5 +249,5 @@ L40:
     dx[i__ + 3] = *da * dx[i__ + 3];
     dx[i__ + 4] = *da * dx[i__ + 4];
   }
-  return 0;
+  return;
 }

data/ext/lbfgsb/src/blas.h

@@ -1,11 +1,11 @@
 #ifndef LBFGSB_RB_BLAS_H_
 #define LBFGSB_RB_BLAS_H_
 
-#include <math.h>
+#include "common.h"
 
-extern int lbfgsb_rb_daxpy_(long* n, double* da, double* dx, long* incx, double* dy, long* incy);
-extern int lbfgsb_rb_dcopy_(long* n, double* dx, long* incx, double* dy, long* incy);
-extern double lbfgsb_rb_ddot_(long* n, double* dx, long* incx, double* dy, long* incy);
-extern int lbfgsb_rb_dscal_(long* n, double* da, double* dx, long* incx);
+extern void daxpy_(F77_int* n, double* da, double* dx, F77_int* incx, double* dy, F77_int* incy);
+extern void dcopy_(F77_int* n, double* dx, F77_int* incx, double* dy, F77_int* incy);
+extern double ddot_(F77_int* n, double* dx, F77_int* incx, double* dy, F77_int* incy);
+extern void dscal_(F77_int* n, double* da, double* dx, F77_int* incx);
 
 #endif /* LBFGSB_RB_BLAS_H_ */

data/ext/lbfgsb/src/common.h

@@ -0,0 +1,16 @@
+#ifndef LBFGSB_RB_COMMON_H_
+#define LBFGSB_RB_COMMON_H_
+
+#include <stdint.h>
+#include <inttypes.h>
+#include <math.h>
+
+#ifdef USE_INT64
+typedef int64_t F77_int;
+#define PRIdF77INT PRId64
+#else
+typedef int32_t F77_int;
+#define PRIdF77INT PRId32
+#endif
+
+#endif /* LBFGSB_RB_COMMON_H_ */