cumo 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 6898598e4614bd912dfa4e83e31695d7fc5eb256
+  data.tar.gz: 752d37294a51016c0256bcdab6d324ebc975bdac
+SHA512:
+  metadata.gz: 81279774f60ac5f38a23751f52a42ae01022d9a5c33cf24d905302def0bc75e20237ad76bb986f4bdd43a184eca14ff3d27da46095189c29217d8fd894dad9fc
+  data.tar.gz: 609ff0cbf6c3a28fc4fafde14f90adfc82ff7efcf268071a73605d61615feb21770e5bddfc8252a042657cbb2bc179f839030e4b85b1f4c11475ed9b27e92ef5

data/.gitignore

@@ -0,0 +1,27 @@
+/.bundle/
+/.yardoc
+/Gemfile.lock
+/_yardoc/
+/coverage/
+/doc/
+/pkg/
+/spec/reports/
+/tmp/
+*.bundle
+*.so
+*.o
+*.a
+mkmf.log
+test.rb
+ext/cumo/narray/types/*.c
+ext/cumo/narray/types/*.cu
+ext/cumo/include/cumo/extconf.h
+ext/cumo/Makefile
+ext/cumo/depend
+TAGS
+t.rb
+.ruby-version
+.yardoc/
+*.exe
+.DS_Store
+/vendor/

data/.travis.yml

@@ -0,0 +1,5 @@
+sudo: false
+language: ruby
+rvm:
+  - 2.4.1
+before_install: gem install bundler -v 1.15.4

data/3rd_party/mkmf-cu/.gitignore

@@ -0,0 +1,36 @@
+*.gem
+*.rbc
+/.config
+/coverage/
+/InstalledFiles
+/pkg/
+/spec/reports/
+/spec/examples.txt
+/test/tmp/
+/test/version_tmp/
+/tmp/
+
+## Specific to RubyMotion:
+.dat*
+.repl_history
+build/
+
+## Documentation cache and generated files:
+/.yardoc/
+/_yardoc/
+/doc/
+/rdoc/
+
+## Environment normalization:
+/.bundle/
+/vendor/bundle
+/lib/bundler/man/
+
+# for a library or gem, you might want to ignore these files since the code is
+# intended to run in multiple environments; otherwise, check them in:
+Gemfile.lock
+.ruby-version
+.ruby-gemset
+
+# unless supporting rvm < 1.11.0 or doing something fancy, ignore this:
+.rvmrc

data/3rd_party/mkmf-cu/Gemfile

@@ -0,0 +1,3 @@
+source "http://rubygems.org"
+
+gemspec

data/3rd_party/mkmf-cu/LICENSE

@@ -0,0 +1,21 @@
+The MIT License (MIT)
+
+Copyright (c) 2016 ruby-accel
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

data/3rd_party/mkmf-cu/README.md

@@ -0,0 +1,36 @@
+# About mkmf-cu
+
+mkmf-cu is a gem to build Ruby extensions written in C/C++ with NVIDIA CUDA.
+It consists of a simple wrapper command for nvcc and a monkey patch for mkmf.
+
+## How to use it.
+
+Instead of require "mkmf", just
+```ruby
+require "mkmf-cu"
+```
+
+## How does it work?
+
+By requiring "mkmf-cu", compiler commands defined in mkmf
+will be replaced with mkmf-cu-nvcc, a command included in this gem.
+
+When mkmf-cu-nvcc is called with arguments for gcc or clang,
+it convert them to ones suitable for nvcc and execute nvcc with them.
+
+For example,
+
+    mkmf-cu-nvcc -I. -fno-common -pipe -Os -O2 -Wall -o culib.o -c culib.cu
+
+will execute
+
+    nvcc -I. -O2 -o culib.o -c culib.cu --compiler-options -fno-common --compiler-options -Wall
+
+## Example
+
+* https://github.com/ruby-accel/ruby-cuda-example
+
+## Notice
+
+When the suffix of the name of a file containing CUDA code is not .cu,
+you must add the option "-x cu" to $CFLAGS or $CXXFLAGS in extconf.rb.

data/3rd_party/mkmf-cu/Rakefile

@@ -0,0 +1,11 @@
+require 'bundler/gem_tasks'
+require 'rake/testtask'
+require 'rake/clean'
+
+Rake::TestTask.new do |t|
+  t.libs << 'test'
+end
+desc "Run tests"
+
+task :default => [:test]
+

data/3rd_party/mkmf-cu/bin/mkmf-cu-nvcc

@@ -0,0 +1,4 @@
+#!/usr/bin/env ruby
+require_relative "../lib/mkmf-cu/cli"
+
+MakeMakefileCuda::CLI.new(ARGV).run

data/3rd_party/mkmf-cu/lib/mkmf-cu.rb

@@ -0,0 +1,32 @@
+require "mkmf"
+
+module MakeMakefileCuda
+  BIN_PATH = File.join(File.dirname(__dir__), 'bin', 'mkmf-cu-nvcc')
+
+  class << self
+    # @params [cxx] Treat .cu files as C++ files
+    def install!(cxx: false)
+      MakeMakefile::CONFIG["CC"]  = "#{BIN_PATH} --mkmf-cu-ext=c"
+      MakeMakefile::CONFIG["CXX"] = "#{BIN_PATH} --mkmf-cu-ext=cxx"
+      if cxx
+        MakeMakefile::CXX_EXT << "cu"
+      else
+        MakeMakefile::C_EXT << "cu"
+      end
+      MakeMakefile::SRC_EXT << "cu"
+      @installed = true
+    end
+
+    def installed?
+      !!@installed
+    end
+
+    def uninstall!
+      MakeMakefile::CONFIG["CC"] = RbConfig::CONFIG["CC"]
+      MakeMakefile::CONFIG["CXX"] = RbConfig::CONFIG["CXX"]
+      MakeMakefile::C_EXT.delete("cu")
+      MakeMakefile::CXX_EXT.delete("cu")
+      @installed = false
+    end
+  end
+end

data/3rd_party/mkmf-cu/lib/mkmf-cu/cli.rb

@@ -0,0 +1,80 @@
+require "open3"
+require_relative "nvcc"
+
+module MakeMakefileCuda
+  class CLI
+    attr_reader :argv
+
+    def initialize(argv)
+      @argv = argv.map{|e| e.dup }
+    end
+
+    def run
+      if cu_file?
+        puts "[given options]: #{argv.join(' ')}"
+        run_command!(*nvcc_command)
+      elsif c_file?
+        run_command!(*c_command)
+      elsif cxx_file?
+        run_command!(*cxx_command)
+      else
+        raise 'something wrong'
+      end
+    end
+
+    # private
+
+    def run_command!(*args)
+      puts colorize(:green, args.join(' '))
+      exit system(*args)
+    end
+
+    # TODO(sonots): Make it possible to configure "nvcc" and additional arguments
+    def nvcc_command
+      s = MakeMakefileCuda::Nvcc.generate(argv)
+      ["nvcc " << s << " -arch=sm_35"]
+    end
+
+    def c_command
+      [RbConfig::CONFIG["CC"], *argv[1..-1]]
+    end
+
+    def cxx_command
+      [RbConfig::CONFIG["CXX"], *argv[1..-1]]
+    end
+
+    def src_file
+      argv.last # *.{c,cc,cpp,cu}
+    end
+
+    def mkmf_cu_ext
+      argv.first # --mkmf-cu-ext={c|cxx}
+    end
+
+    def cu_file?
+      src_file.end_with?('.cu')
+    end
+
+    def c_file?
+      !cu_file? and mkmf_cu_ext.end_with?('=c')
+    end
+
+    def cxx_file?
+      !cu_file? and mkmf_cu_ext.end_with?('=cxx')
+    end
+
+    COLOR_CODES = {
+      red: 31,
+      green: 32,
+      yellow: 33,
+      blue: 34,
+      magenta: 35,
+      cyan: 36
+    }
+
+    def colorize(code, str)
+      raise "#{color_code} is not supported" unless COLOR_CODES[code]
+      "\e[#{COLOR_CODES[code]}m#{str}\e[0m"
+    end
+  end
+end

data/3rd_party/mkmf-cu/lib/mkmf-cu/nvcc.rb

@@ -0,0 +1,157 @@
+require "optparse"
+require "rbconfig"
+
+module MakeMakefileCuda
+  class Nvcc
+    # Generate (convert) NVCC arguments from GCC arguments
+    def self.generate(argv)
+      new.generate(argv)
+    end
+
+    # private
+
+    def generate(argv)
+      opt, opt_h = build_optparser()
+
+      parse_ill_short(argv, opt_h)
+      parse_ill_short_with_arg(argv, opt_h)
+
+      argv = opt.parse(argv)
+
+      if opt_h["-c"].size > 0
+        s = generate_compiling_command_line(opt_h)
+      elsif opt_h['-dynamic'].size > 0 or opt_h['-shared'].size > 0
+        s = generate_linking_command_line(argv, opt_h)
+      else
+        raise 'something wrong'
+      end
+    end
+
+    def build_optparser
+      opt = OptionParser.new
+      opt_h = Hash.new{|h, k| h[k] = [] }
+
+      opt.on("--arch arg") {|v| opt_h["-arch"] << v }
+      opt.on("--std arg") {|v| opt_h["-std"] << v }
+      opt.on("--stdlib arg") {|v| opt_h["-stdlib"] << v }
+
+      opt.on("--Wl arg") {|v| opt_h["-Wl"] << v }
+
+      opt.on('--profile') {|v| opt_h["-pg"] << "" }
+      opt.on('-g') {|v| opt_h["-g"] << "" }
+      opt.on('-G', "--device-debug") {|v| opt_h["-G"] << "" }
+
+      opt.on('-I path') {|v| opt_h["-I"] << v }
+      opt.on('-D flag') {|v| opt_h["-D"] << v }
+      opt.on('-W flag') {|v| opt_h["-W"] << v }
+      opt.on('-o output') {|v| opt_h["-o"] << v }
+      opt.on('-c file') {|v| opt_h["-c"] << v }
+      opt.on('-f flag') {|v| opt_h["-f"] << v }
+      opt.on('-l file') {|v| opt_h["-l"] << v }
+      opt.on('-L path') {|v| opt_h["-L"] << v }
+      opt.on('-x pat', "--x pat") {|v| opt_h["-x"] << v }
+      opt.on('-O num'){|v| opt_h["-O"] << v if /[0-9]/ =~ v }
+      opt.on('--mkmf-cu-ext ext'){|v| opt_h["--mkmf-cu-ext"] << v}
+
+      return [opt, opt_h]
+    end
+
+    def parse_ill_short(argv, opt_h)
+      ["-shared", "-rdynamic", "-dynamic", "-bundle",  "-pipe", "-pg", "-ggdb3"].each{|opt|
+        if ind = argv.find_index(opt)
+          opt_h[opt] << ""
+          argv.delete_at(ind)
+        end
+      }
+      ["-arch", "-std", "-stdlib"].each{|opt|
+        if ind = argv.find_index(opt)
+          argv[ind] = "-" + opt
+        end
+      }
+    end
+
+    def parse_ill_short_with_arg(argv, opt_h)  
+      [/\A(\-stdlib)=(.*)/, /\A(\-std)=(.*)/, /\A(\-Wl),(.*)/].each{|reg|
+        argv.each{|e|
+          if reg =~ e
+            e[0..-1] = "-" + $1 + '=' + $2
+          end
+        }
+      }
+    end
+
+    def compiler_option(opt_h)
+      ret = ""
+      ["-f", "-W", "-pipe"].each{|op|
+        opt_h[op].each{|e|
+          ret << " --compiler-options " + "#{op}#{e}"
+        }
+      }
+      ["-stdlib", "-std"].each{|op|
+        opt_h[op].each{|e|
+          ret << " --compiler-options " + "#{op}=#{e}"
+        }
+      }
+      return ret
+    end
+
+    def linker_option(opt_h)
+      ret = " -shared "
+      ["-dynamic", "-bundle"].each{|op|
+        opt_h[op].each{|e|
+          ret << " --linker-options " + op
+        }
+      }
+      opt_h["-Wl"].each{|e|
+        ret << " --linker-options " + e
+      }
+      return ret
+    end
+
+    def compiler_bin(opt_h)
+      if opt_h["--mkmf-cu-ext"][0] == "c"
+        " --compiler-bindir " + RbConfig::CONFIG["CC"]
+      elsif opt_h["--mkmf-cu-ext"][0] == "cxx"
+        " --compiler-bindir " + RbConfig::CONFIG["CXX"]
+      end
+    end
+
+    def generate_compiling_command_line(opt_h)
+      s = ""
+      # options nvcc can uderstatnd
+      ["-std", "-pg", "-g", "-G", "-x", "-I", "-D", "-o", "-c", "-O"].each{|op|
+        opt_h[op].each{|e|
+          case op
+          when "-o", "-c", "-x", "-std"
+            s << " #{op} #{e}"
+          else
+            s << " #{op}#{e}"
+          end
+        }
+      }
+      s << compiler_option(opt_h)
+      s << compiler_bin(opt_h)
+
+      return s
+    end
+
+    def generate_linking_command_line(argv, opt_h)
+      s = ""
+      ["-L", "-l", "-o", "-c", "-O"].each{|op|
+        opt_h[op].each{|e|
+          case op
+          when "-o", "-c"
+            s << " #{op} #{e}"
+            s << " " + argv.join(" ") + " " if op == "-o"
+          else
+            s << " #{op}#{e}"
+          end
+        }
+      }
+      s << compiler_option(opt_h)
+      s << linker_option(opt_h)
+      s << compiler_bin(opt_h)
+      return s
+    end
+  end
+end

data/3rd_party/mkmf-cu/mkmf-cu.gemspec

@@ -0,0 +1,16 @@
+Gem::Specification.new do |s|
+  s.name        = 'mkmf-cu'
+  s.version     = '0.1.2'
+  s.date        = '2016-03-26'
+  s.summary     = "Write Ruby extension in C/C++ with NVIDIA CUDA."
+  s.description =
+  "Write Ruby extension in C/C++ with NVIDIA CUDA. A simple wrapper command for nvcc and a monkey patch for mkmf."
+  s.authors     = ["Takashi Tamura"]
+  s.email       = ''
+  s.files       = ["lib/mkmf-cu.rb", "lib/mkmf-cu/opt.rb", "LICENSE", "README.md"]
+  s.homepage    = 'https://github.com/ruby-accel/mkmf-cu'
+  s.license     = 'MIT'
+  s.executables << "mkmf-cu-nvcc"
+
+  s.add_development_dependency "rake", "~> 10.0"
+end