ndav-torch-tensor 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 033bbf192b1235b16c2b5fe0f69ca5505764e1473c9c5b4abdf18d81d4b9ec94
-  data.tar.gz: 02bca7f664f5a1fa8ec84a93202ef117f887a62b172305101c4278fc8c6cfd08
+  metadata.gz: 51c99b994468de49aacf7b24673dd7cd4e755b1bc1f5ea858d2aa85093aa299c
+  data.tar.gz: 9c0737070ab1b78c0f9ef3908b989060e6693d555380a6c793a3e4f26015dc57
 SHA512:
-  metadata.gz: 1b9360bf3dd2fce2780775dcb03d19554a7f4a67960555bec1b8e88aa3300453ea515dd827efeaf292d63f703f46b692fa6ae381ac2fb7236fc18aab2f01f7a0
-  data.tar.gz: b88c727b786f43319043d874689f8882cbabd4737292d3e663405b1161a1b7b6b904d89b229543f103cad1786310d2ade8a78b930e50c16e82c262e0070ec07b
+  metadata.gz: 4d03dcd52b4cc3db70cd26222a6c2c0f2ae8ab6f361aa593227860fd93f4a1ff8c1423040e63e37bc929e7a517f7f2210874af71fe030e84b7bf3b74b212a747
+  data.tar.gz: b6cdbdfdac3d49920e23bdbad413d92d97e2343536b9f295d25a5319d1fdf139321e67995f20ac6dec61525e48888c129d351269a36b1d3cea6fcf7cdc6c9a5e

data/README.md

@@ -1,11 +1,30 @@
 NDAV::Torch::Tensor
 ===================
 
-MemoryView and pointer support for [Torch]::Tensor.
+[NDAV][] bridge for [Torch][]::Tensor.
+
+SYNOPSIS
+--------
+
+```ruby
+require "ndav/torch/tensor"
+
+torch = Torch.rand(2, 3)     # => Torch::Tensor
+ndav  = torch.to_ndav        # => NDAV
+torch = ndav.to_torch_tensor # => Torch::Tensor
+
+ndav  = NDAV.from_torch_tensor(torch)  # => NDAV
+torch = Torch::Tensor.from_ndav(ndav)  # => Torch::Tensor
+
+include NDAV::Converter
+ndav  = NDAV(torch)       # => NDAV
+torch = TorchTensor(ndav) # => Torch::NArray
+```
 
 LICENSE
 -------
 
 MIT license. See LICENSE.txt file.
 
+[NDAV]: https://kitaitimakoto.gitlab.io/ndav/
 [Torch]: https://github.com/ankane/torch.rb

data/lib/ndav/torch/tensor.rb

@@ -46,13 +46,25 @@ class NDAV
 
       TYPE_SIZES = TYPE_TO_FORMAT.transform_values {|format| ITEM_SIZES[format]}
 
-      module ToNDAV
-        def to_ndav
+      module MemoryViewable
+        def ndav_descriptor(**)
           format = TYPE_TO_FORMAT[dtype]
-          raise ArgumentError, "unsupported dtype: #{dtype}, currently supported: #{TYPE_TO_FORMAT.keys}" unless format
+          unless format
+            warn "unsupported dtype: #{dtype}, currently supported: #{TYPE_TO_FORMAT.keys}"
+            return false
+          end
+
+          byte_size = numel * element_size
 
-          ptr = Fiddle::Pointer.new(_data_ptr, numel * element_size)
-          NDAV.new(ptr, shape:, format:, lifetime: self)
+          {
+            data: ::Fiddle::Pointer.new(_data_ptr, byte_size),
+            shape:,
+            strides: stride.collect {|s| s * element_size},
+            format:,
+            byte_size:,
+            readonly?: false,
+            sub_offsets: nil
+          }
         end
       end
 

data/ndav-torch-tensor.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name = "ndav-torch-tensor"
-  s.version = "0.0.2"
+  s.version = "0.0.3"
   s.authors = ["Kitaiti Makoto"]
   s.summary = "N-Dimensional Array View for Torch::Tensors"
   s.licenses = ["MIT"]
@@ -8,7 +8,7 @@ Gem::Specification.new do |s|
 
   s.files = Dir.chdir(__dir__) {`git ls-files -z`.split("\x0")}
 
-  s.add_runtime_dependency "ndav", ">= 0.0.3"
+  s.add_runtime_dependency "ndav", ">= 0.0.5"
   s.add_runtime_dependency "torch-rb"
 
   s.add_development_dependency "rake"

data/test/test_torch_tensor.rb

@@ -14,6 +14,15 @@ class TestTorchTensor < Test::Unit::TestCase
     assert_kind_of ::Torch::Tensor, ::Torch.from_ndav(@ndav)
   end
 
+  def test_memory_view
+    Fiddle::MemoryView.export @torch_tensor do |view|
+      assert_equal "s", view.format
+      assert_equal [3], view.shape
+      assert_equal [2], view.strides
+      assert_equal [1, 2, 3].pack("s*"), view.to_s
+    end
+  end
+
   def test_ndav_to_torch_tensor
     assert_kind_of ::Torch::Tensor, @ndav.to_torch_tensor
   end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ndav-torch-tensor
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Kitaiti Makoto
@@ -15,14 +15,14 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.0.3
+        version: 0.0.5
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.0.3
+        version: 0.0.5
 - !ruby/object:Gem::Dependency
   name: torch-rb
   requirement: !ruby/object:Gem::Requirement
@@ -166,7 +166,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: N-Dimensional Array View for Torch::Tensors
 test_files: []