torch-rb 0.11.1 → 0.12.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c4c3936a55fd47a8898d3e678aabc81c158f2f39a89c08af0b36695700bf2043
-  data.tar.gz: a68179c7d7bab7547ac3be1d2369abbd5c3c632954996ca2f7fdd089f815edf7
+  metadata.gz: 24617a6191c4d1e42ff0d5885c0762db75d034ee52817d70004ddc52025890dd
+  data.tar.gz: 7036f3c7fac8a1aac22914ec2811b8771464baeb847052e417b069f3897b3977
 SHA512:
-  metadata.gz: 97551aa27f154cade530e58b4ae92286cd18e432acc99646495a197c89fb719cc991b0399aacaf383a2de04e340c04a9e177c44a0169fce91e19435463df4753
-  data.tar.gz: 8a5a5decf900a4d93aa56eb5e5b3848d5386be8bbc3657c10c201ef321cf0082c6521700378a5e96f795c09882b5b72498d8fed3931153b6e1f7557630f55547
+  metadata.gz: 044166f526bea5fea0c4314abc55dfaf2410cdd5d7bce764ccb78329754fe284675dc0ecb55cf5a4546587221cf2361c8e215f1e9e067668ea5833c445bea961
+  data.tar.gz: e086d0dcc731fb3e810f8fb91d04d95bf3c5efd9cd35f053238be60479d8a47f2126ddad7ab2705d18a26175c6c4126f12be3f19962c2a69f24ad3ee551252b1

data/CHANGELOG.md

@@ -1,3 +1,11 @@
+## 0.12.0 (2022-11-05)
+
+- Updated LibTorch to 1.13.0
+
+## 0.11.2 (2022-09-25)
+
+- Improved LibTorch detection for Homebrew on Mac ARM and Linux
+
 ## 0.11.1 (2022-07-06)
 
 - Fixed error with `stft` method

data/README.md

@@ -13,10 +13,10 @@ Check out:
 
 ## Installation
 
-First, [install LibTorch](#libtorch-installation). For Homebrew, use:
+First, [install LibTorch](#libtorch-installation). With Homebrew, it’s part of the PyTorch package:
 
 ```sh
-brew install libtorch
+brew install pytorch
 ```
 
 Add this line to your application’s Gemfile:
@@ -409,7 +409,8 @@ Here’s the list of compatible versions.
 
 Torch.rb | LibTorch
 --- | ---
-0.11.0+ | 1.12.0+
+0.12.0+ | 1.13.0+
+0.11.0-0.11.2 | 1.12.0-1.12.1
 0.10.0-0.10.2 | 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
@@ -425,23 +426,9 @@ Torch.rb | LibTorch
 You can also use Homebrew.
 
 ```sh
-brew install libtorch
+brew install pytorch
 ```
 
-For Mac ARM, run:
-
-```sh
-bundle config build.torch-rb --with-torch-dir=/opt/homebrew
-```
-
-And for Linux, run:
-
-```sh
-bundle config build.torch-rb --with-torch-dir=/home/linuxbrew/.linuxbrew
-```
-
-Then install the gem.
-
 ## Performance
 
 Deep learning is significantly faster on a GPU. With Linux, install [CUDA](https://developer.nvidia.com/cuda-downloads) and [cuDNN](https://developer.nvidia.com/cudnn) and reinstall the gem.

data/codegen/function.rb

@@ -34,6 +34,10 @@ class Function
     !out_index.nil?
   end
 
+  def dispatch_name
+    definition.dig("dispatch", "CompositeImplicitAutograd")
+  end
+
   private
 
   def parse_func
@@ -77,6 +81,10 @@ class Function
         default = "torch.int64"
       end
 
+      if name == "dtype" && base_name == "randint"
+        default = "None"
+      end
+
       default = nil if definition["cpp_no_default_args"].to_a.include?(name)
 
       params << {

data/codegen/generate_functions.rb

@@ -15,6 +15,7 @@ def generate_functions
   generate_files("linalg", :define_singleton_method, functions[:linalg])
   generate_files("special", :define_singleton_method, functions[:special])
   generate_files("sparse", :define_singleton_method, functions[:sparse])
+  # TODO generate nested
 end
 
 def load_functions
@@ -40,13 +41,14 @@ def skip_functions(functions)
     # not supported yet
     f.func.include?("Dimname") ||
     f.func.include?("ConstQuantizerPtr") ||
-    f.func.include?("SymInt") ||
     # TODO fix LibTorch 1.12 changes
     f.base_name == "histogramdd" ||
     f.base_name == "nested_tensor" ||
     f.base_name == "split_copy" ||
     f.base_name == "split_with_sizes_copy" ||
-    f.base_name == "unbind_copy"
+    f.base_name == "unbind_copy" ||
+    # TODO fix LibTorch 1.13 changes
+    f.base_name == "native_channel_shuffle"
   end
 end
 
@@ -56,6 +58,7 @@ def group_functions(functions)
   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" }
+  nested_functions, other_functions = other_functions.partition { |f| f.python_module == "nested" }
   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") }
@@ -72,7 +75,8 @@ def group_functions(functions)
     linalg: linalg_functions,
     fft: fft_functions,
     special: special_functions,
-    sparse: sparse_functions
+    sparse: sparse_functions,
+    nested: nested_functions
   }
 end
 
@@ -387,6 +391,8 @@ def generate_function_params(function, params, remove_self)
         "scalarlist"
       when /\Aint\[/
         "intlist"
+      when /\ASymInt\[/
+        "symintlist"
       when "float[]"
         "doublelist"
       when "Scalar"
@@ -395,6 +401,8 @@ def generate_function_params(function, params, remove_self)
         "toBool"
       when "int"
         "toInt64"
+      when "SymInt"
+        "toSymInt"
       when "float"
         "toDouble"
       when "ScalarType"
@@ -437,7 +445,12 @@ def generate_dispatch_code(function, def_method, params, opt_index, remove_self)
   # torch::empty sets requires_grad by at::empty doesn't
   # https://github.com/pytorch/pytorch/issues/36455
   prefix = remove_self ? "self." : (opt_index ? "torch::" : "at::")
-  dispatch = function.out? ? "#{function.base_name}_out" : function.base_name
+  dispatch = function.dispatch_name
+  unless dispatch
+    dispatch = function.base_name
+    dispatch += "_symint" if function.func.include?("SymInt")
+    dispatch += "_out" if function.out?
+  end
 
   params = params.map { |v| v[:name] }
   params.reject! { |v| v == "self" } if remove_self
@@ -478,6 +491,8 @@ def generate_dispatch_params(function, params)
         "ScalarList"
       when "int"
         "int64_t"
+      when "SymInt"
+        "c10::SymInt"
       when "float"
         "double"
       when /\Aint\[/
@@ -486,6 +501,12 @@ def generate_dispatch_params(function, params)
         else
           "IntArrayRef"
         end
+      when /\ASymInt\[/
+        if param[:optional]
+          "at::OptionalSymIntArrayRef"
+        else
+          "c10::SymIntArrayRef"
+        end
       when "float[]"
         "ArrayRef<double>"
       when "str"
@@ -506,7 +527,7 @@ def generate_dispatch_params(function, params)
         raise "Unknown type: #{param[:type]} (#{function.name})"
       end
 
-    if param[:optional] && !["Tensor", "Scalar"].include?(param[:type]) && !param[:type].start_with?("int[")
+    if param[:optional] && !["Tensor", "Scalar"].include?(param[:type]) && !param[:type].start_with?("int[") && !param[:type].start_with?("SymInt[")
       type = "c10::optional<#{type}>"
     end
 
@@ -614,8 +635,12 @@ def signature_type(param)
       "DirnameList"
     when /\Aint\[\d*\]\z/
       "IntArrayRef"
+    when /\ASymInt\[\d*\]\z/
+      "SymIntArrayRef"
     when "int"
       "int64_t"
+    when "SymInt"
+      "c10::SymInt"
     when "float"
       "double"
     when "str"