ndav-torch-tensor 0.0.4 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '0680ecfdc12ff8c7439ae719364591beeae64666e91c0cc33b47429222848682'
-  data.tar.gz: 3ee2acec3f638e8918828496250a85f7b531750b9bfaa366c1f2d31b2b126f84
+  metadata.gz: ea390bc80d0a532c10da276aac487657743f4e188a93db24424662c22be0e7c4
+  data.tar.gz: 620f47e4a3feda6a59958562a4e4463ba6fd2afa2cb03cb2b2b75846fe80bd7f
 SHA512:
-  metadata.gz: d0aa2feab3aed3844acaea511a61eda04befd736db3a7009bb4627421cfafe486d21e9300e0bf2ea02483215879c461b555c59857ccbaf8eeaba6944db4b576f
-  data.tar.gz: 94eac20b44cf43b958eb817992e3c11fdb4541b0853fb70c720e5ed0d27affc061f7f528c5688c5c43c7ecf6e4b37dba5273b7958e699ae11b33da8e127f66b5
+  metadata.gz: 3c026d68364bc1a9a8511372dfb9ad2f2d83d8cb1887a1c744575cf8121bf3722d96d536b61bf1f996d0fd3c5b79ad3011cedc4ccb2fa58275510cb7ae60f85c
+  data.tar.gz: be8683e7369a631c60f64564f7884c6ee47bdc317885cfdb0b208edf9df66c52fbf0cb8788175ee3be8629bb859d5517d051ef40070731810d125a78edea7ea0

data/lib/ndav/torch/tensor.rb

@@ -31,31 +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!"
-                 :"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
-        [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.4"
+  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.6"
+  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.4
+  version: 0.0.5
 platform: ruby
 authors:
 - Kitaiti Makoto
@@ -15,14 +15,14 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.0.6
+        version: 0.0.7
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.0.6
+        version: 0.0.7
 - !ruby/object:Gem::Dependency
   name: torch-rb
   requirement: !ruby/object:Gem::Requirement