numo-openblas 0.1.1 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 69ec5184f3c04827324d0fd0d5af5e9b92dd61b7710df099df26379525a48f09
-  data.tar.gz: 41fcc9d0a64991fe1e43d43467c6a98138fd3fa195428e2fb1663ffa571fccca
+  metadata.gz: eb28541f7f6f87b87410da68933898703b42d2987721e429b4d09ac7f1d23127
+  data.tar.gz: ea7a198b2c3d3c17c7b03898d9c9b31a4c747335401b521ae4415c63b7a45c74
 SHA512:
-  metadata.gz: 86d775545a967607cfce5c4e824a220224822a05fd2a5796eaa9e9a380b8c909829cf8b2d5e394349b6ff65a75d34729393bab8b25949afe2b9090c415373969
-  data.tar.gz: 158f89106e2611e1fdc29ce9c97f1b235c966eac637523230dca4a14db449febd18b32a7c42416cce973a1c34328b098b9e004f8670001489c8acfe7d0860e3f
+  metadata.gz: a2eec31d62b6cd6ee263b9219356835cb60fe52562dc2ab8090e7cb8261fa10de28eeb0c007a23d522def8af23a4d7d039ac5437f11270b2bc3e896e529ad645
+  data.tar.gz: dc3b17590715bf76411eb71e29b9cad3afea8c5ae13b57070638d55b63460e17914c0142428ebbf2ab71853e6d94f9e56e183f175c8378737037fb7d557d5d8b

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+## 0.2.0
+- Add constant values about OpenBLAS.
+- Change to download and build OpenBLAS when building native extension.
+
 ## 0.1.1
 - Fix to check downloaded file's SHA1.
 

data/Gemfile

@@ -4,4 +4,5 @@ source 'https://rubygems.org'
 gemspec
 
 gem 'rake', '~> 12.0'
+gem 'rake-compiler', '~> 1.0'
 gem 'rspec', '~> 3.0'

data/Rakefile

@@ -3,4 +3,13 @@ require 'rspec/core/rake_task'
 
 RSpec::Core::RakeTask.new(:spec)
 
-task :default => :spec
+require 'rake/extensiontask'
+
+task build: :compile
+
+Rake::ExtensionTask.new('openblas') do |ext|
+  ext.ext_dir = 'ext/numo/openblas'
+  ext.lib_dir = 'lib/numo/openblas'
+end
+
+task default: %i[clobber compile spec]

data/ext/numo/openblas/extconf.rb

@@ -0,0 +1,76 @@
+# frozen-string-literal: true
+
+require 'digest/sha1'
+require 'etc'
+require 'fileutils'
+require 'mkmf'
+require 'open-uri'
+require 'open3'
+require 'rubygems/package'
+
+OPENBLAS_VER = '0.3.10'
+OPENBLAS_KEY = 'cbe3fdd0e6ee235debc611d76976dac62f3ddc1c'
+OPENBLAS_URI = "https://github.com/xianyi/OpenBLAS/archive/v#{OPENBLAS_VER}.tar.gz"
+OPENBLAS_DIR = File.expand_path(__dir__ + '/../../../vendor')
+
+unless File.exist?("#{OPENBLAS_DIR}/installed_#{OPENBLAS_VER}")
+  puts "Downloading OpenBLAS #{OPENBLAS_VER}."
+  URI.open(OPENBLAS_URI) do |rf|
+    File.open("#{OPENBLAS_DIR}/tmp/openblas.tgz", 'wb') { |sf| sf.write(rf.read) }
+  end
+
+  if OPENBLAS_KEY != Digest::SHA1.file("#{OPENBLAS_DIR}/tmp/openblas.tgz").to_s
+    puts 'SHA1 digest of downloaded file does not match.'
+    exit(1)
+  end
+
+  puts 'Unpacking OpenBLAS tar.gz file.'
+  Gem::Package::TarReader.new(Zlib::GzipReader.open("#{OPENBLAS_DIR}/tmp/openblas.tgz")) do |tar|
+    tar.each do |entry|
+      next unless entry.file?
+
+      filename = "#{OPENBLAS_DIR}/tmp/#{entry.full_name}"
+      next if filename == File.dirname(filename)
+
+      FileUtils.mkdir_p("#{OPENBLAS_DIR}/tmp/#{File.dirname(entry.full_name)}")
+      File.open(filename, 'wb') { |f| f.write(entry.read) }
+      File.chmod(entry.header.mode, filename)
+    end
+  end
+
+  Dir.chdir("#{OPENBLAS_DIR}/tmp/OpenBLAS-#{OPENBLAS_VER}") do
+    puts 'Building OpenBLAS. This could take a while...'
+    mkstdout, _mkstderr, mkstatus = Open3.capture3("make -j#{Etc.nprocessors}")
+    File.open("#{OPENBLAS_DIR}/tmp/openblas.log", 'w') { |f| f.puts(mkstdout) }
+    unless mkstatus.success?
+      puts 'Failed to build OpenBLAS.'
+      puts 'Check the make-openblas.log file for more details:'
+      puts "#{OPENBLAS_DIR}/tmp/openblas.log"
+      exit(1)
+    end
+
+    puts 'Installing OpenBLAS.'
+    insstdout, _insstderr, insstatus = Open3.capture3("make install PREFIX=#{OPENBLAS_DIR}")
+    File.open("#{OPENBLAS_DIR}/tmp/openblas.log", 'a') { |f| f.puts(insstdout) }
+    unless insstatus.success?
+      puts 'Failed to install OpenBLAS.'
+      puts 'Check the make-openblas.log file for more details:'
+      puts "#{OPENBLAS_DIR}/tmp/openblas.log"
+      exit(1)
+    end
+
+    FileUtils.touch("#{OPENBLAS_DIR}/installed_#{OPENBLAS_VER}")
+  end
+end
+
+unless find_library('openblas', nil, "#{OPENBLAS_DIR}/lib")
+  puts 'libopenblas is not found.'
+  exit(1)
+end
+
+unless find_header('openblas_config.h', nil, "#{OPENBLAS_DIR}/include")
+  puts 'openblas_config.h is not found.'
+  exit(1)
+end
+
+create_makefile('numo/openblas/openblas')

data/ext/numo/openblas/openblas.c

@@ -0,0 +1,34 @@
+/**
+ * Numo::OpenBLAS downloads and builds OpenBLAS during installation and
+ * uses that as a background library for Numo::Linalg.
+ */
+
+#include <ruby.h>
+#include <openblas_config.h>
+
+VALUE mNumo;
+VALUE mOpenBLAS;
+
+void Init_openblas()
+{
+  /**
+  * Document-module: Numo
+  * Numo is the top level namespace of NUmerical MOdules for Ruby.
+  */
+  mNumo = rb_define_module("Numo");
+
+  /**
+   * Document-module: Numo::Liblinear
+   * Numo::OpenBLAS loads Numo::NArray and Linalg with OpenBLAS used as backend library.
+   */
+  mOpenBLAS = rb_define_module_under(mNumo, "OpenBLAS");
+
+  /* The number of cores detected by OpenBLAS. */
+  rb_define_const(mOpenBLAS, "OPENBLAS_NUM_CORES", INT2NUM(OPENBLAS_NUM_CORES));
+
+  /* The core name detected by OpenBLAS. */
+  rb_define_const(mOpenBLAS, "OPENBLAS_CHAR_CORENAME", rb_str_new_cstr(OPENBLAS_CHAR_CORENAME));
+
+  /* The version of OpenBLAS used in background library. */
+  rb_define_const(mOpenBLAS, "OPENBLAS_VERSION", rb_str_new_cstr(OPENBLAS_VERSION));
+}

data/lib/numo/openblas.rb

@@ -3,6 +3,7 @@
 require 'numo/narray'
 require 'numo/linalg/linalg'
 require 'numo/openblas/version'
+require 'numo/openblas/openblas'
 
 module Numo
   module Linalg

data/lib/numo/openblas/version.rb

@@ -1,12 +1,10 @@
 # frozen_string_literal: true
 
+# Numo is the top level namespace of NUmerical MOdules for Ruby.
 module Numo
   # Numo::OpenBLAS loads Numo::NArray and Linalg with OpenBLAS used as backend library.
   module OpenBLAS
     # The version of Numo::OpenBLAS you install.
-    VERSION = '0.1.1'
-
-    # The version OpenBLAS that Numo::OpenBLAS build and use.
-    OPENBLAS_VERSION = '0.3.10'
+    VERSION = '0.2.0'
   end
 end

data/numo-openblas.gemspec

@@ -30,6 +30,7 @@ Gem::Specification.new do |spec|
   spec.bindir        = 'exe'
   spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
   spec.require_paths = ['lib']
+  spec.extensions    = ['ext/numo/openblas/extconf.rb']
 
   spec.add_runtime_dependency 'numo-linalg', '>= 0.1.4'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: numo-openblas
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.2.0
 platform: ruby
 authors:
 - yoshoku
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2020-07-26 00:00:00.000000000 Z
+date: 2020-08-01 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: numo-linalg
@@ -30,7 +30,8 @@ description: |
 email:
 - yoshoku@outlook.com
 executables: []
-extensions: []
+extensions:
+- ext/numo/openblas/extconf.rb
 extra_rdoc_files: []
 files:
 - ".gitignore"
@@ -42,9 +43,10 @@ files:
 - LICENSE.txt
 - README.md
 - Rakefile
+- ext/numo/openblas/extconf.rb
+- ext/numo/openblas/openblas.c
 - lib/numo/openblas.rb
 - lib/numo/openblas/version.rb
-- lib/rubygems_plugin.rb
 - numo-openblas.gemspec
 - vendor/tmp/.gitkeep
 homepage: https://github.com/yoshoku/numo-openblas

data/lib/rubygems_plugin.rb

@@ -1,65 +0,0 @@
-# frozen-string-literal: true
-
-require 'rubygems'
-require 'rubygems/package'
-
-require 'digest/sha1'
-require 'etc'
-require 'fileutils'
-require 'open-uri'
-require 'open3'
-
-Gem.post_install do
-  openblas_ver = '0.3.10'
-  openblas_key = 'cbe3fdd0e6ee235debc611d76976dac62f3ddc1c'
-  openblas_uri = "https://github.com/xianyi/OpenBLAS/archive/v#{openblas_ver}.tar.gz"
-  openblas_dir = File.expand_path(__dir__ + '/../vendor')
-
-  next true if File.exist?("#{openblas_dir}/installed_#{openblas_ver}")
-
-  puts "Downloading OpenBLAS #{openblas_ver}."
-  URI.open(openblas_uri) do |rf|
-    File.open("#{openblas_dir}/tmp/openblas.tgz", 'wb') { |sf| sf.write(rf.read) }
-  end
-
-  next false if openblas_key != Digest::SHA1.file("#{openblas_dir}/tmp/openblas.tgz").to_s
-
-  puts 'Unpacking OpenBLAS tar.gz file.'
-  Gem::Package::TarReader.new(Zlib::GzipReader.open("#{openblas_dir}/tmp/openblas.tgz")) do |tar|
-    tar.each do |entry|
-      next unless entry.file?
-
-      filename = "#{openblas_dir}/tmp/#{entry.full_name}"
-      next if filename == File.dirname(filename)
-
-      FileUtils.mkdir_p("#{openblas_dir}/tmp/#{File.dirname(entry.full_name)}")
-      File.open(filename, 'wb') { |f| f.write(entry.read) }
-      File.chmod(entry.header.mode, filename)
-    end
-  end
-
-  Dir.chdir("#{openblas_dir}/tmp/OpenBLAS-#{openblas_ver}") do
-    puts 'Building OpenBLAS. This could take a while...'
-    mkstdout, _mkstderr, mkstatus = Open3.capture3("make -j#{Etc.nprocessors}")
-    File.open("#{openblas_dir}/tmp/openblas.log", 'w') { |f| f.puts(mkstdout) }
-    unless mkstatus.success?
-      puts 'Failed to build OpenBLAS.'
-      puts 'Check the make-openblas.log file for more details:'
-      puts "#{openblas_dir}/tmp/openblas.log"
-      break false
-    end
-
-    puts 'Installing OpenBLAS.'
-    insstdout, _insstderr, insstatus = Open3.capture3("make install PREFIX=#{openblas_dir}")
-    File.open("#{openblas_dir}/tmp/openblas.log", 'a') { |f| f.puts(insstdout) }
-    unless insstatus.success?
-      puts 'Failed to install OpenBLAS.'
-      puts 'Check the make-openblas.log file for more details:'
-      puts "#{openblas_dir}/tmp/openblas.log"
-      break false
-    end
-
-    FileUtils.touch("#{openblas_dir}/installed_#{openblas_ver}")
-    true
-  end
-end