torch-rb 0.9.1 → 0.10.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f5224c74f6e74ed04396dfa0414400af5cb20bc5e654320421116723ffcb8e83
-  data.tar.gz: 6a2881ddacb7610a231ebd5a1c24d0f71a2662f16f360102141dfd0893c13346
+  metadata.gz: f4665eec43d85fbf02ce75f4b268dbf001bfad7e3ae1ecace0e9911b651e2cc2
+  data.tar.gz: d11ee1386ce7feeea68333de6c361d8737a7164cfd1626abce3a511deecb2963
 SHA512:
-  metadata.gz: 045432235e1c691ce85fb937a0562d93b1d9bc312fc648d40dafcc24857eeec4f84e6ceab397793171b0046ccfd785a6caeba37902925dcff1f73c760dd57cec
-  data.tar.gz: 2520fa17dcd13be52aaf1256f431d2951f8113f862189b8691f877dcb48ac9f71ac70719600e27d8c8972494c0b7f11c0983c3330fed7ca6aeed552e8500ec22
+  metadata.gz: cf346bc03f36d4fc920151b0554c93c33a59d0fecd35c6f110dc862ad8b35c6b8641d306124505ddd012ce9d903363672e99772cc2bd7b981d962c9d00f08d3e
+  data.tar.gz: 265157846417fdc3c024e0f50d0b0a663d345ca423ad9233a7d0e722bf5134a8e15e1afce30248992155787484e80311e484c64c5787945b5f5a88625115479a

data/CHANGELOG.md

@@ -1,3 +1,19 @@
+## 0.10.1 (2022-04-12)
+
+- Fixed `dtype`, `device`, and `layout` for `new_*` and `like_*` methods
+
+## 0.10.0 (2022-03-13)
+
+- Updated LibTorch to 1.11.0
+- Added `ParameterList`
+
+## 0.9.2 (2022-02-03)
+
+- Added support for setting `nil` gradient
+- Added checks when setting gradient
+- Fixed precision with `Torch.tensor` method
+- Fixed memory issue when creating tensor for `ByteStorage`
+
 ## 0.9.1 (2022-02-02)
 
 - Moved `like` methods to C++

data/README.md

@@ -7,6 +7,7 @@ Check out:
 - [TorchVision](https://github.com/ankane/torchvision) for computer vision tasks
 - [TorchText](https://github.com/ankane/torchtext) for text and NLP tasks
 - [TorchAudio](https://github.com/ankane/torchaudio) for audio tasks
+- [TorchRec](https://github.com/ankane/torchrec-ruby) for recommendation systems
 
 [![Build Status](https://github.com/ankane/torch.rb/workflows/build/badge.svg?branch=master)](https://github.com/ankane/torch.rb/actions)
 
@@ -408,7 +409,8 @@ Here’s the list of compatible versions.
 
 Torch.rb | LibTorch
 --- | ---
-0.9.0+ | 1.10.0+
+0.10.0+ | 1.11.0+
+0.9.0-0.9.2 | 1.10.0-1.10.2
 0.8.0-0.8.3 | 1.9.0-1.9.1
 0.6.0-0.7.0 | 1.8.0-1.8.1
 0.5.0-0.5.3 | 1.7.0-1.7.1

data/codegen/function.rb

@@ -37,7 +37,7 @@ class Function
   private
 
   def parse_func
-    input, output = func.split(/\s*->\s*/)
+    input, _, output = func.rpartition(/\s+->\s+/)
     [generate_params(input), generate_retvals(output)]
   end
 
@@ -52,7 +52,7 @@ class Function
         next
       end
 
-      type, name = i.split(/\s+/)
+      type, _, name = i.rpartition(/\s+/)
 
       if name.include?("=")
         name, default = name.split("=", 2)

data/codegen/generate_functions.rb

@@ -14,6 +14,7 @@ def generate_functions
   generate_files("fft", :define_singleton_method, functions[:fft])
   generate_files("linalg", :define_singleton_method, functions[:linalg])
   generate_files("special", :define_singleton_method, functions[:special])
+  generate_files("sparse", :define_singleton_method, functions[:sparse])
 end
 
 def load_functions
@@ -47,6 +48,7 @@ def group_functions(functions)
   linalg_functions, other_functions = other_functions.partition { |f| f.python_module == "linalg" }
   fft_functions, other_functions = other_functions.partition { |f| f.python_module == "fft" }
   special_functions, other_functions = other_functions.partition { |f| f.python_module == "special" }
+  sparse_functions, other_functions = other_functions.partition { |f| f.python_module == "sparse" }
   unexpected_functions, other_functions = other_functions.partition { |f| f.python_module }
   torch_functions = other_functions.select { |f| f.variants.include?("function") }
   tensor_functions = other_functions.select { |f| f.variants.include?("method") }
@@ -62,7 +64,8 @@ def group_functions(functions)
     nn: nn_functions,
     linalg: linalg_functions,
     fft: fft_functions,
-    special: special_functions
+    special: special_functions,
+    sparse: sparse_functions
   }
 end
 
@@ -136,6 +139,7 @@ def generate_attach_def(name, type, def_method)
   ruby_name = ruby_name.sub(/\Afft_/, "") if type == "fft"
   ruby_name = ruby_name.sub(/\Alinalg_/, "") if type == "linalg"
   ruby_name = ruby_name.sub(/\Aspecial_/, "") if type == "special"
+  ruby_name = ruby_name.sub(/\Asparse_/, "") if type == "sparse"
   ruby_name = name if name.start_with?("__")
 
   # cast for Ruby < 2.7 https://github.com/thisMagpie/fftw/issues/22#issuecomment-49508900
@@ -289,7 +293,13 @@ def split_opt_params(params)
 end
 
 def generate_tensor_options(function, opt_params)
-  code = "\n  const auto options = TensorOptions()"
+  new_function = function.base_name.start_with?("new_")
+  like_function = function.base_name.end_with?("_like")
+
+  code = String.new("")
+  code << "\n  auto self = _r.tensor(0);" if like_function
+  code << "\n  const auto options = TensorOptions()"
+
   order = ["dtype", "device", "layout", "requires_grad", "pin_memory"]
   opt_params.sort_by { |v| order.index(v[:name]) }.each do |opt|
     i = opt[:position]
@@ -300,12 +310,24 @@ def generate_tensor_options(function, opt_params)
         if function.base_name == "arange"
           "dtype(_r.scalartypeOptional(#{i}))"
         else
-          "dtype(_r.scalartype(#{i}))"
+          if new_function || like_function
+            "dtype(_r.scalartypeWithDefault(#{i}, self.scalar_type()))"
+          else
+            "dtype(_r.scalartype(#{i}))"
+          end
         end
       when "device"
-        "device(_r.device(#{i}))"
+        if new_function || like_function
+          "device(_r.deviceWithDefault(#{i}, self.device()))"
+        else
+          "device(_r.device(#{i}))"
+        end
       when "layout"
-        "layout(_r.layoutOptional(#{i}))"
+        if new_function || like_function
+          "layout(_r.layoutWithDefault(#{i}, self.layout()))"
+        else
+          "layout(_r.layoutOptional(#{i}))"
+        end
       when "requires_grad"
         "requires_grad(_r.toBool(#{i}))"
       when "pin_memory"