Sha256: a0ec681c2bebe9f5e051ad5a95b5dcdc6713da44760fdac05e48953cd6cc083d

Contents?: true

Size: 1014 Bytes

Versions: 24

Compression:

Stored size: 1014 Bytes

Contents

# dependencies
require "ffi"

# modules
require_relative "onnxruntime/datasets"
require_relative "onnxruntime/inference_session"
require_relative "onnxruntime/model"
require_relative "onnxruntime/utils"
require_relative "onnxruntime/version"

module OnnxRuntime
  class Error < StandardError; end

  class << self
    attr_accessor :ffi_lib
  end
  lib_name =
    if Gem.win_platform?
      "onnxruntime.dll"
    elsif RbConfig::CONFIG["host_os"] =~ /darwin/i
      if RbConfig::CONFIG["host_cpu"] =~ /arm|aarch64/i
        "libonnxruntime.arm64.dylib"
      else
        "libonnxruntime.dylib"
      end
    else
      if RbConfig::CONFIG["host_cpu"] =~ /arm|aarch64/i
        "libonnxruntime.arm64.so"
      else
        "libonnxruntime.so"
      end
    end
  vendor_lib = File.expand_path("../vendor/#{lib_name}", __dir__)
  self.ffi_lib = [vendor_lib]

  def self.lib_version
    FFI.OrtGetApiBase[:GetVersionString].call.read_string
  end

  # friendlier error message
  autoload :FFI, "onnxruntime/ffi"
end

Version data entries

24 entries across 24 versions & 1 rubygems

Version Path
onnxruntime-0.9.2-x86_64-linux lib/onnxruntime.rb
onnxruntime-0.9.2-x86_64-darwin lib/onnxruntime.rb
onnxruntime-0.9.2-x64-mingw lib/onnxruntime.rb
onnxruntime-0.9.2-arm64-darwin lib/onnxruntime.rb
onnxruntime-0.9.2-aarch64-linux lib/onnxruntime.rb
onnxruntime-0.9.2 lib/onnxruntime.rb
onnxruntime-0.9.0-x86_64-linux lib/onnxruntime.rb
onnxruntime-0.9.0-x86_64-darwin lib/onnxruntime.rb
onnxruntime-0.9.0-x64-mingw lib/onnxruntime.rb
onnxruntime-0.9.0-arm64-darwin lib/onnxruntime.rb
onnxruntime-0.9.0-aarch64-linux lib/onnxruntime.rb
onnxruntime-0.9.0 lib/onnxruntime.rb
onnxruntime-0.8.0-x86_64-linux lib/onnxruntime.rb
onnxruntime-0.8.0-x86_64-darwin lib/onnxruntime.rb
onnxruntime-0.8.0-x64-mingw lib/onnxruntime.rb
onnxruntime-0.8.0-arm64-darwin lib/onnxruntime.rb
onnxruntime-0.8.0-aarch64-linux lib/onnxruntime.rb
onnxruntime-0.8.0 lib/onnxruntime.rb
onnxruntime-0.7.7-x86_64-linux lib/onnxruntime.rb
onnxruntime-0.7.7-x86_64-darwin lib/onnxruntime.rb