tensor_stream-opencl 0.1.0

data/lib/tensor_stream/opencl/opencl_template_helper.rb

@@ -0,0 +1,99 @@
+require 'erb'
+class OpenclTemplateHelper
+  def initialize(source)
+    @source = source
+  end
+
+  def generate(args = {})
+    current_scope = binding
+
+    args.each do |k, v|
+      current_scope.local_variable_set(k.to_sym, v)
+    end
+
+    ERB.new(@source, nil, '%').result(current_scope)
+  end
+
+  def floating_point?(dtype)
+    TensorStream::Ops::FLOATING_POINT_TYPES.include?(dtype)
+  end
+
+  def render(template, locals = {})
+    filename = File.join(File.dirname(__FILE__), 'kernels', "_#{template}")
+    source = File.read(filename)
+    current_scope = binding
+    locals.each do |k, v|
+      current_scope.local_variable_set(k.to_sym, v)
+    end
+    ERB.new(source, nil, '%').result(current_scope)
+  end
+
+  def dtype_to_c_type(dtype)
+    case dtype.to_s
+    when 'float64'
+      'double'
+    when 'float32', 'float'
+      'float'
+    when 'int32', 'int'
+      'int'
+    when 'int16'
+      'short'
+    when 'uint8'
+      'uchar'
+    when 'boolean'
+      'uchar'
+    else
+      raise "unknown dtype #{dtype}"
+    end
+  end
+
+  def min_value_for(dtype)
+    case dtype.to_s
+    when 'float64'
+      'DBL_MIN'
+    when 'float32', 'float'
+      'FLT_MIN'
+    when 'int32', 'int'
+      'INT_MIN'
+    when 'int16'
+      'SHRT_MIN'
+    when 'int8'
+      '0'
+    when 'boolean'
+      '0'
+    else
+      raise "unknown dtype #{dtype}"
+    end
+  end
+
+  def operator_to_c(op)
+    case op
+    when 'less'
+      '<'
+    when 'less_equal'
+      '<='
+    when 'equal'
+      '=='
+    when 'greater'
+      '>'
+    when 'greater_equal'
+      '>='
+    when 'not_equal'
+      '!='
+    when 'logical_and'
+      '&&'
+    when 'div'
+      '/'
+    when 'add'
+      '+'
+    when 'sub'
+      '-'
+    when 'mul'
+      '*'
+    when 'mod'
+      '%'
+    else
+      raise "unsupported op #{op}"
+    end
+  end
+end

data/lib/tensor_stream/opencl/version.rb

@@ -0,0 +1,5 @@
+module TensorStream
+  module Opencl
+    VERSION = "0.1.0"
+  end
+end

data/tensor_stream-opencl.gemspec

@@ -0,0 +1,40 @@
+
+lib = File.expand_path("../lib", __FILE__)
+$LOAD_PATH.unshift(lib) unless $LOAD_PATH.include?(lib)
+require "tensor_stream/opencl/version"
+
+Gem::Specification.new do |spec|
+  spec.name          = "tensor_stream-opencl"
+  spec.version       = TensorStream::Opencl::VERSION
+  spec.authors       = ["Joseph Dayo"]
+  spec.email         = ["joseph.dayo@gmail.com"]
+
+  spec.summary       = %q{OpenCL evaluator for tensor_stream}
+  spec.description   = %q{OpenCL evaluator for tensor_stream}
+  spec.homepage      = "https://github.com/jedld/tensor_stream-opencl"
+  spec.license       = "MIT"
+
+  # Prevent pushing this gem to RubyGems.org. To allow pushes either set the 'allowed_push_host'
+  # to allow pushing to a single host or delete this section to allow pushing to any host.
+  if spec.respond_to?(:metadata)
+    spec.metadata["allowed_push_host"] = "https://rubygems.org"
+  else
+    raise "RubyGems 2.0 or newer is required to protect against " \
+      "public gem pushes."
+  end
+
+  # Specify which files should be added to the gem when it is released.
+  # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
+  spec.files         = Dir.chdir(File.expand_path('..', __FILE__)) do
+    `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
+  end
+  spec.bindir        = "exe"
+  spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency "bundler", "~> 1.16"
+  spec.add_development_dependency "rake", "~> 10.0"
+  spec.add_development_dependency "rspec", "~> 3.0"
+  spec.add_dependency "tensor_stream"
+  spec.add_dependency "opencl_ruby_ffi"
+end

metadata

@@ -0,0 +1,185 @@
+--- !ruby/object:Gem::Specification
+name: tensor_stream-opencl
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Joseph Dayo
+autorequire: 
+bindir: exe
+cert_chain: []
+date: 2018-09-06 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.16'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.16'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '10.0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.0'
+- !ruby/object:Gem::Dependency
+  name: tensor_stream
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: opencl_ruby_ffi
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+description: OpenCL evaluator for tensor_stream
+email:
+- joseph.dayo@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- ".rspec"
+- ".travis.yml"
+- CODE_OF_CONDUCT.md
+- Gemfile
+- Gemfile.lock
+- LICENSE.txt
+- README.md
+- Rakefile
+- bin/console
+- bin/setup
+- lib/tensor_stream/opencl.rb
+- lib/tensor_stream/opencl/kernels/_bool_operand.cl
+- lib/tensor_stream/opencl/kernels/_operand.cl
+- lib/tensor_stream/opencl/kernels/abs.cl
+- lib/tensor_stream/opencl/kernels/acos.cl
+- lib/tensor_stream/opencl/kernels/add.cl
+- lib/tensor_stream/opencl/kernels/apply_adam.cl
+- lib/tensor_stream/opencl/kernels/apply_gradient.cl
+- lib/tensor_stream/opencl/kernels/apply_momentum.cl
+- lib/tensor_stream/opencl/kernels/argmax.cl
+- lib/tensor_stream/opencl/kernels/argmin.cl
+- lib/tensor_stream/opencl/kernels/asin.cl
+- lib/tensor_stream/opencl/kernels/cast.cl
+- lib/tensor_stream/opencl/kernels/ceil.cl
+- lib/tensor_stream/opencl/kernels/cond.cl.erb
+- lib/tensor_stream/opencl/kernels/cos.cl
+- lib/tensor_stream/opencl/kernels/div.cl.erb
+- lib/tensor_stream/opencl/kernels/exp.cl
+- lib/tensor_stream/opencl/kernels/floor.cl
+- lib/tensor_stream/opencl/kernels/floor_div.cl
+- lib/tensor_stream/opencl/kernels/floor_mod.cl
+- lib/tensor_stream/opencl/kernels/gemm.cl
+- lib/tensor_stream/opencl/kernels/log.cl
+- lib/tensor_stream/opencl/kernels/log1p.cl
+- lib/tensor_stream/opencl/kernels/log_softmax.cl
+- lib/tensor_stream/opencl/kernels/max.cl
+- lib/tensor_stream/opencl/kernels/min.cl
+- lib/tensor_stream/opencl/kernels/mod.cl
+- lib/tensor_stream/opencl/kernels/mul.cl
+- lib/tensor_stream/opencl/kernels/negate.cl
+- lib/tensor_stream/opencl/kernels/pack.cl
+- lib/tensor_stream/opencl/kernels/pow.cl
+- lib/tensor_stream/opencl/kernels/real_div.cl
+- lib/tensor_stream/opencl/kernels/reciprocal.cl
+- lib/tensor_stream/opencl/kernels/round.cl
+- lib/tensor_stream/opencl/kernels/sigmoid.cl
+- lib/tensor_stream/opencl/kernels/sigmoid_grad.cl
+- lib/tensor_stream/opencl/kernels/sign.cl
+- lib/tensor_stream/opencl/kernels/sin.cl
+- lib/tensor_stream/opencl/kernels/softmax.cl
+- lib/tensor_stream/opencl/kernels/softmax_cross.cl
+- lib/tensor_stream/opencl/kernels/softmax_cross_grad.cl
+- lib/tensor_stream/opencl/kernels/softmax_grad.cl
+- lib/tensor_stream/opencl/kernels/sqrt.cl
+- lib/tensor_stream/opencl/kernels/square.cl
+- lib/tensor_stream/opencl/kernels/squared_difference.cl
+- lib/tensor_stream/opencl/kernels/sub.cl
+- lib/tensor_stream/opencl/kernels/tan.cl
+- lib/tensor_stream/opencl/kernels/tanh.cl
+- lib/tensor_stream/opencl/kernels/tanh_grad.cl
+- lib/tensor_stream/opencl/kernels/where.cl
+- lib/tensor_stream/opencl/math_ops.rb
+- lib/tensor_stream/opencl/nn_ops.rb
+- lib/tensor_stream/opencl/opencl_buffer.rb
+- lib/tensor_stream/opencl/opencl_device.rb
+- lib/tensor_stream/opencl/opencl_evaluator.rb
+- lib/tensor_stream/opencl/opencl_template_helper.rb
+- lib/tensor_stream/opencl/version.rb
+- tensor_stream-opencl.gemspec
+homepage: https://github.com/jedld/tensor_stream-opencl
+licenses:
+- MIT
+metadata:
+  allowed_push_host: https://rubygems.org
+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.6.10
+signing_key: 
+specification_version: 4
+summary: OpenCL evaluator for tensor_stream
+test_files: []