ndav-torch-tensor 0.0.3 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 51c99b994468de49aacf7b24673dd7cd4e755b1bc1f5ea858d2aa85093aa299c
-  data.tar.gz: 9c0737070ab1b78c0f9ef3908b989060e6693d555380a6c793a3e4f26015dc57
+  metadata.gz: ea390bc80d0a532c10da276aac487657743f4e188a93db24424662c22be0e7c4
+  data.tar.gz: 620f47e4a3feda6a59958562a4e4463ba6fd2afa2cb03cb2b2b75846fe80bd7f
 SHA512:
-  metadata.gz: 4d03dcd52b4cc3db70cd26222a6c2c0f2ae8ab6f361aa593227860fd93f4a1ff8c1423040e63e37bc929e7a517f7f2210874af71fe030e84b7bf3b74b212a747
-  data.tar.gz: b6cdbdfdac3d49920e23bdbad413d92d97e2343536b9f295d25a5319d1fdf139321e67995f20ac6dec61525e48888c129d351269a36b1d3cea6fcf7cdc6c9a5e
+  metadata.gz: 3c026d68364bc1a9a8511372dfb9ad2f2d83d8cb1887a1c744575cf8121bf3722d96d536b61bf1f996d0fd3c5b79ad3011cedc4ccb2fa58275510cb7ae60f85c
+  data.tar.gz: be8683e7369a631c60f64564f7884c6ee47bdc317885cfdb0b208edf9df66c52fbf0cb8788175ee3be8629bb859d5517d051ef40070731810d125a78edea7ea0

data/README.md

@@ -13,6 +13,8 @@ torch = Torch.rand(2, 3)     # => Torch::Tensor
 ndav  = torch.to_ndav        # => NDAV
 torch = ndav.to_torch_tensor # => Torch::Tensor
 
+memory_view = Fiddle::MemoryView.new(torch) # => Fiddle::MemoryView, now it may export a MemoryView
+
 ndav  = NDAV.from_torch_tensor(torch)  # => NDAV
 torch = Torch::Tensor.from_ndav(ndav)  # => Torch::Tensor
 

data/lib/ndav/torch/tensor.rb

@@ -31,20 +31,38 @@ class NDAV
         float64: "d"
       }
 
-      FORMAT_TO_TYPE = ITEM_SIZES.to_h {|format, size|
-        suffix = size * 8
-        type = case format
-               when "s!", "c", "s", "l", "q", "i", "i!", "l!", "q!"
-                 :"int#{suffix}"
-               when "S!", "C", "S", "L", "Q", "I", "I!", "L!", "Q!", "n", "v", "N", "V"
-                 :"uint#{suffix}"
-               when "f", "e", "g", "d", "E", "G"
-                 :"float#{suffix}"
-               end
-        [format, type]
-      }
+      FORMAT_TO_TYPE = Hash.new {|types, format| types[format] = type_from_format(format)}
+
+      class << self
+        def type_from_format(format)
+          item_desc = ITEM_DESCS[format]
+          return unless item_desc.length == 1
 
-      TYPE_SIZES = TYPE_TO_FORMAT.transform_values {|format| ITEM_SIZES[format]}
+          component = item_desc[0]
+          return unless component.repeat == 1
+
+          suffix = component.size * 8
+          case component.format
+          when "c", "s", "l", "q", "i", "j"
+            :"int#{suffix}"
+          when "C", "S", "L", "Q", "I", "J"
+            :"uint#{suffix}"
+          when "f", "d"
+            :"float#{suffix}"
+          else
+            case IO::Buffer::HOST_ENDIAN
+            when IO::Buffer::LITTLE_ENDIAN
+              if ["v", "V", "e", "E"].include? format
+                :"float#{suffix}"
+              end
+            when IO::Buffer::BIG_ENDIAN
+              if ["n", "N", "g", "G"].include? format
+                :"float#{suffix}"
+              end
+            end
+          end
+        end
+      end
 
       module MemoryViewable
         def ndav_descriptor(**)

data/ndav-torch-tensor.gemspec

@@ -1,6 +1,6 @@
 Gem::Specification.new do |s|
   s.name = "ndav-torch-tensor"
-  s.version = "0.0.3"
+  s.version = "0.0.5"
   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.5"
+  s.add_runtime_dependency "ndav", ">= 0.0.7"
   s.add_runtime_dependency "torch-rb"
 
   s.add_development_dependency "rake"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ndav-torch-tensor
 version: !ruby/object:Gem::Version
-  version: 0.0.3
+  version: 0.0.5
 platform: ruby
 authors:
 - Kitaiti Makoto
@@ -15,14 +15,14 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.0.5
+        version: 0.0.7
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.0.5
+        version: 0.0.7
 - !ruby/object:Gem::Dependency
   name: torch-rb
   requirement: !ruby/object:Gem::Requirement