torch-rb 0.9.0 → 0.10.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '09221364dad232f1b76129fe9dc9407675cc2afbd03bd1339c736d4eec752df7'
-  data.tar.gz: a37a0584aed809009ebd74e7c0da9430481ccabf1ac915d2468ee7511c249588
+  metadata.gz: 7884d0bd8ffd23e775b3a1ca59e536bc18cd79c6bbe2736a9802a2b1693d40de
+  data.tar.gz: 10849c8a248a70eca7178ca3529fa7428fc3d8e5cfe2ca774b0be6cb5e75eb9b
 SHA512:
-  metadata.gz: c220d35971b9ce3e5a7a80f6a5d1ae4324f3524d0e0171c680deced66fe2f29342a46eecb1a4447d84a401a677c7bb1ef910a0c7ee6c925ea4b578b7e5712772
-  data.tar.gz: 807fe2907de1caac92da6dddb0154b7971dda3aa0ee2c53f6b3046732f4bf3c02310e59a6441efa0fdf1ad3d0ddb5dcd7a3c1a946adbfbea73b6b30f10a71487
+  metadata.gz: d81587eae00527e9d1e4a65b62a686dbdab27eed9401539faddf553d3a0730ad7394b01bac615bdd0474d1a6602cd0a3117e1d7cb3a3f1d5fbf8bd989fa39e59
+  data.tar.gz: d581d07821f103ee69267bc8e6b15d5094d8b8ef004917af947e2570649e7fbe750c9893d40600b62ee970646654fc545a270e7a59adb764be2ba73f1fa18b62

data/CHANGELOG.md

@@ -1,3 +1,20 @@
+## 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++
+- Fixed memory issue
+
 ## 0.9.0 (2021-10-23)
 
 - Updated LibTorch to 1.10.0

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)
 
@@ -21,10 +22,10 @@ brew install libtorch
 Add this line to your application’s Gemfile:
 
 ```ruby
-gem 'torch-rb'
+gem "torch-rb"
 ```
 
-It can take a few minutes to compile the extension.
+It can take 5-10 minutes to compile the extension.
 
 ## Getting Started
 
@@ -79,7 +80,7 @@ b = Torch.zeros(2, 3)
 
 Each tensor has four properties
 
-- `dtype` - the data type - `:uint8`, `:int8`, `:int16`, `:int32`, `:int64`, `:float32`, `float64`, or `:bool`
+- `dtype` - the data type - `:uint8`, `:int8`, `:int16`, `:int32`, `:int64`, `:float32`, `:float64`, or `:bool`
 - `layout` - `:strided` (dense) or `:sparse`
 - `device` - the compute device, like CPU or GPU
 - `requires_grad` - whether or not to record gradients
@@ -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