Sha256: ab1f014596d8b2af6de925ce60b7105339f8fa931676aec93ef4bcb05230baba

Contents?: true

Size: 1.03 KB

Versions: 6

Compression:

Stored size: 1.03 KB

Contents

# dependencies
require "ffi"

# modules
require_relative "onnxruntime/datasets"
require_relative "onnxruntime/inference_session"
require_relative "onnxruntime/model"
require_relative "onnxruntime/ort_value"
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

6 entries across 6 versions & 1 rubygems

Version Path
onnxruntime-0.9.3-x86_64-linux lib/onnxruntime.rb
onnxruntime-0.9.3-x86_64-darwin lib/onnxruntime.rb
onnxruntime-0.9.3-x64-mingw lib/onnxruntime.rb
onnxruntime-0.9.3-arm64-darwin lib/onnxruntime.rb
onnxruntime-0.9.3-aarch64-linux lib/onnxruntime.rb
onnxruntime-0.9.3 lib/onnxruntime.rb