ndav-ort_value 0.0.2 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: eda5c49be2368e38bb4ce8d836942ae2932537aa08aa3946775f57a745c4ce3b
-  data.tar.gz: 2baa6174a4d6797358cdd527507c75f6342ee0a913f006030e74c53dd10dbdb9
+  metadata.gz: 304532f2da38f6bfb85d3cee934fc353485321481f2765cd574c20896243d9cc
+  data.tar.gz: 95fb3f89d47cb3188ceab7f32c7373e0d8245c1e983147b615687ef6701fe8d0
 SHA512:
-  metadata.gz: ed06686bd9ca80142c3c2551dfb232f3e1e189f961ebd2aa606c6cc2aeb48a2a7e1412836279e98cf44ef3d7ecc18d026f4f5f607f3210033f3cfaa3e5a12708
-  data.tar.gz: 3f7087cc91385fa0419562ae65a1b9d7bcd5023dc0292ba298b0f38cab46cf6dfb4bbd02b9cf30ba76a605d46449c2aec0f5f8e3dcf8d78bf44b102956d8dcb5
+  metadata.gz: 61550a33ec5ebb7c5d0cedf8fe992e09d67948bb91dae9ebf4625287fb29438a852eb716c78da069cd3f9e282ab45c1b0e29c39662c2f2baadeaed158665606b
+  data.tar.gz: '0837dfb4b3e1f38112fb87495ed4f57b1f775303a428872c97a03e5f10eabfdd42a9831a4e61a28ac90ad6efeedb525a2c7978e3853fd28e2dc5f9fe1fc0cfb5'

data/README.md

@@ -1,11 +1,32 @@
 OnnxRuntime::OrtValue::NDAV
 ===========================
 
-MemoryView and pointer support for [ONNX Runtime Ruby][].
+[NDAV][] bridge for [ONNX Runtime Ruby][].
+
+SYNOPSIS
+--------
+
+```ruby
+require "ndav/ort_value"
+
+ort_value = OnnxRuntime::OrtValue.from_array([1.0, 2.0, 3.0], element_type: :float) # => OnnxRuntime::OrtValue
+ndav      = ort_value.to_ndav # => NDAV
+ort_value = ndav.to_ort_value # => OnnxRuntime::OrtValue
+
+memory_view = Fiddle::MemoryView.new(ort_value) # => Fiddle::MemoryView, now it may export a MemoryView
+
+ndav      = NDAV.from_ort_value(ort_value)        # => NDAV
+ort_value = OnnxRuntime::OrtValue.from_ndav(ndav) # => OnnxRuntime::OrtValue
+
+include NDAV::Converter
+ndav  = NDAV(ort_value)    # => NDAV
+ort_value = OrtValue(ndav) # => OnnxRuntime::OrtValue
+```
 
 LICENSE
 ------
 
 MIT license. See LICENSE.txt file.
 
+[NDAV]: https://kitaitimakoto.gitlab.io/ndav/
 [ONNX Runtime Ruby]: https://github.com/ankane/onnxruntime-ruby

data/lib/ndav/ort_value.rb

@@ -83,66 +83,42 @@ class NDAV
       end
     end
 
-    module ToNDAV
-      def to_ndav
+    module MemoryViewable
+      def ndav_descriptor(**)
         item_size = TYPE_SIZES[element_type]
-        Utils.unsupported_type("NDAV", element_type) unless item_size
+        unless item_size
+          warn "unsupported type: #{element_type}"
+          return false
+        end
+
         byte_size = shape.reduce(item_size, :*)
 
-        ::NDAV.new(
-          ::Fiddle::Pointer.new(data_ptr.address, byte_size),
-          shape: shape.dup,
+        {
+          data: ::Fiddle::Pointer.new(data_ptr.address, byte_size),
+          shape:,
+          strides: ::NDAV.default_strides(shape:, item_size:),
           format: TYPE_TO_FORMAT[element_type],
-          lifetime: self
-        )
-      end
-    end
-
-    module FromOrtValue
-      def from_ort_value(array)
-        array.to_ndav
-      end
-    end
-
-    module ToOrtValue
-      def to_ort_value
-        ::OnnxRuntime::OrtValue.from_ndav(self)
+          byte_size:,
+          readonly?: true,
+          sub_offsets: nil
+        }
       end
     end
 
     module Converter
-      if defined? ::NDAV::Converter::OrtValue
-        def OrtValue(array, *, **)
-          if array.kind_of? ::NDAV
-            array.to_ort_value
-          elsif array.respond_to? :to_ort_value
-            array.to_ort_value
-          elsif array.respond_to? :to_ndav
-            array.to_ndav(*, **).to_ort_value
-          else
-            super
-          end
-        end
-      else
-        def OrtValue(array, *, **)
-          if array.kind_of? ::NDAV
-            array.to_ort_value
-          elsif array.respond_to? :to_ort_value
-            array.to_ort_value
-          elsif array.respond_to? :to_ndav
-            array.to_ndav(*, **).to_ort_value
-          else
-            ::NDAV.new(array, *, **).to_ort_value
-          end
+      def OrtValue(array, *, **)
+        if array.kind_of? ::OnnxRuntime::OrtValue
+          array
+        elsif array.respond_to? :to_ort_value
+          array.to_ort_value
+        elsif array.respond_to? :to_ndav
+          to_ndav(*, **).to_ort_value
+        else
+          ::NDAV.new(array, *, **).to_ort_value
         end
       end
     end
-
-    ::OnnxRuntime::OrtValue.extend FromNDAV
-    ::OnnxRuntime::OrtValue.include ToNDAV
-    ::NDAV.extend FromOrtValue
-    ::NDAV.include ToOrtValue
-    ::NDAV::Converter.singleton_class.prepend Converter
-    ::NDAV::Converter.prepend Converter
   end
+
+  ::NDAV.register ::OnnxRuntime::OrtValue, OrtValue, name: :ort_value
 end

data/ndav-ort_value.gemspec

@@ -1,13 +1,13 @@
 Gem::Specification.new do |s|
   s.name = "ndav-ort_value"
-  s.version = "0.0.2"
+  s.version = "0.0.4"
   s.authors = ["Kitaiti Makoto"]
   s.summary = "OnnxRuntime extension for MemoryView exporters and pointers"
   s.licenses = ["MIT"]
   s.homepage = "https://gitlab.com/KitaitiMakoto/ndav-ort_value"
 
   s.add_runtime_dependency "onnxruntime"
-  s.add_runtime_dependency "ndav"
+  s.add_runtime_dependency "ndav", ">= 0.0.5"
 
   s.files = Dir.chdir(__dir__) {`git ls-files -z`.split("\x0")}
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ndav-ort_value
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.4
 platform: ruby
 authors:
 - Kitaiti Makoto
@@ -29,14 +29,14 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.0.5
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.0.5
 - !ruby/object:Gem::Dependency
   name: rake
   requirement: !ruby/object:Gem::Requirement
@@ -164,7 +164,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 4.0.6
+rubygems_version: 4.0.16
 specification_version: 4
 summary: OnnxRuntime extension for MemoryView exporters and pointers
 test_files: []