mkmf-cu 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: dddc0a150502a76ea3ad3bf7a36d15f20b1070c6
+  data.tar.gz: b62362f39b1bf24113b97cd87011a423f733702c
+SHA512:
+  metadata.gz: a6dcba385c2537d8cab74fd10655691c54c3e2f1492d69b1d8c04571622eeca96e714da25fed4a86316cb0246e72e19053db3e8c7bee0aa1b4445961cb0aace6
+  data.tar.gz: 190fc8e035136c97a438c0a1ec343abef2434b515565ea0c16b76cb3725df6471c34e38c82fe4524ce6cffa54eac4ef620bae616f07ed943ed87f92d62a4c320

data/bin/mkmf-cu-nvcc

@@ -0,0 +1,24 @@
+#!/usr/bin/env ruby
+require "mkmf-cu/opt"
+require "open3"
+
+opt, opt_h = build_optparser()
+
+argv = ARGV.map{|e| e.dup }
+
+$stderr.puts '[given command line options]: ' + ARGV.join(' ')
+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
+end
+
+$stderr.puts "nvcc" + s
+Open3.capture2("nvcc " + s)

data/lib/mkmf-cu.rb

@@ -0,0 +1,16 @@
+require "mkmf"
+
+MakeMakefile::CONFIG["CC"]  = "mkmf-cu-nvcc --mkmf-cu-ext=c"
+MakeMakefile::CONFIG["CXX"] = "mkmf-cu-nvcc --mkmf-cu-ext=cxx"
+MakeMakefile::C_EXT << "cu"
+MakeMakefile::SRC_EXT << "cu"
+
+def compile_cu_with_cxx_compiler
+  MakeMakefile::C_EXT.delete("cu")
+  MakeMakefile::CXX_EXT << "cu"
+end
+
+def compile_cu_with_cc_compiler
+  MakeMakefile::CXX_EXT.delete("cu")
+  MakeMakefile::C_EXT << "cu"
+end

data/lib/mkmf-cu/opt.rb

@@ -0,0 +1,132 @@
+require "optparse"
+require "rbconfig"
+
+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"].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(\-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"].each{|op|
+    opt_h[op].each{|e|
+      ret << " --compiler-options " + "#{op}=#{e}"
+    }
+  }
+  return ret
+end
+
+def linker_option(opt_h)
+  ret = ""
+  ["-dynamic", "-bundle", "-shared", "-rdynamic"].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
+  ["-pg", "-g", "-G", "-x", "-I", "-D", "-o", "-c", "-O"].each{|op|
+    opt_h[op].each{|e|
+      case op
+      when "-o", "-c", "-x"
+        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[0] + " " 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

metadata

@@ -0,0 +1,48 @@
+--- !ruby/object:Gem::Specification
+name: mkmf-cu
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Takashi Tamura
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2016-02-24 00:00:00.000000000 Z
+dependencies: []
+description: Write Ruby extention in C/C++ with NVIDIA CUDA. A simple wrapper command
+  for nvcc and a monkey patch for mkmf.
+email: ''
+executables:
+- mkmf-cu-nvcc
+extensions: []
+extra_rdoc_files: []
+files:
+- bin/mkmf-cu-nvcc
+- lib/mkmf-cu.rb
+- lib/mkmf-cu/opt.rb
+homepage: https://github.com/ruby-accel/mkmf-cu
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.5.1
+signing_key: 
+specification_version: 4
+summary: Write Ruby extention in C/C++ with NVIDIA CUDA.
+test_files: []