torch-rb 0.10.1 → 0.11.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f4665eec43d85fbf02ce75f4b268dbf001bfad7e3ae1ecace0e9911b651e2cc2
-  data.tar.gz: d11ee1386ce7feeea68333de6c361d8737a7164cfd1626abce3a511deecb2963
+  metadata.gz: c4c3936a55fd47a8898d3e678aabc81c158f2f39a89c08af0b36695700bf2043
+  data.tar.gz: a68179c7d7bab7547ac3be1d2369abbd5c3c632954996ca2f7fdd089f815edf7
 SHA512:
-  metadata.gz: cf346bc03f36d4fc920151b0554c93c33a59d0fecd35c6f110dc862ad8b35c6b8641d306124505ddd012ce9d903363672e99772cc2bd7b981d962c9d00f08d3e
-  data.tar.gz: 265157846417fdc3c024e0f50d0b0a663d345ca423ad9233a7d0e722bf5134a8e15e1afce30248992155787484e80311e484c64c5787945b5f5a88625115479a
+  metadata.gz: 97551aa27f154cade530e58b4ae92286cd18e432acc99646495a197c89fb719cc991b0399aacaf383a2de04e340c04a9e177c44a0169fce91e19435463df4753
+  data.tar.gz: 8a5a5decf900a4d93aa56eb5e5b3848d5386be8bbc3657c10c201ef321cf0082c6521700378a5e96f795c09882b5b72498d8fed3931153b6e1f7557630f55547

data/CHANGELOG.md

@@ -1,3 +1,17 @@
+## 0.11.1 (2022-07-06)
+
+- Fixed error with `stft` method
+
+## 0.11.0 (2022-07-06)
+
+- Updated LibTorch to 1.12.0
+- Dropped support for Ruby < 2.7
+
+## 0.10.2 (2022-06-14)
+
+- Improved numeric operations between scalars and tensors
+- Fixed `dtype` of `cumsum` method
+
 ## 0.10.1 (2022-04-12)
 
 - Fixed `dtype`, `device`, and `layout` for `new_*` and `like_*` methods

data/README.md

@@ -409,7 +409,8 @@ Here’s the list of compatible versions.
 
 Torch.rb | LibTorch
 --- | ---
-0.10.0+ | 1.11.0+
+0.11.0+ | 1.12.0+
+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
 0.6.0-0.7.0 | 1.8.0-1.8.1
@@ -421,13 +422,25 @@ Torch.rb | LibTorch
 
 ### Homebrew
 
-For Mac, you can use Homebrew.
+You can also use Homebrew.
 
 ```sh
 brew install libtorch
 ```
 
-Then install the gem (no need for `bundle config`).
+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
 

data/codegen/function.rb

@@ -60,7 +60,7 @@ class Function
 
       optional = false
       if type.include?("?")
-        optional = true unless ["dtype", "device", "layout", "pin_memory"].include?(name)
+        optional = true
         type = type.delete("?")
       end
 

data/codegen/generate_functions.rb

@@ -39,7 +39,14 @@ def skip_functions(functions)
     f.base_name == "index_put" ||
     # not supported yet
     f.func.include?("Dimname") ||
-    f.func.include?("ConstQuantizerPtr")
+    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"
   end
 end
 
@@ -121,8 +128,11 @@ def write_body(type, method_defs, attach_defs)
 end
 
 def write_file(name, contents)
-  path = File.expand_path("../ext/torch", __dir__)
-  File.write(File.join(path, name), contents)
+  path = File.join(File.expand_path("../ext/torch", __dir__), name)
+  # only write if changed to improve compile times in development
+  if !File.exist?(path) || File.read(path) != contents
+    File.write(path, contents)
+  end
 end
 
 def generate_attach_def(name, type, def_method)
@@ -135,14 +145,14 @@ def generate_attach_def(name, type, def_method)
       name
     end
 
-  ruby_name = "_#{ruby_name}" if ["size", "stride", "random!", "stft"].include?(ruby_name)
+  ruby_name = "_#{ruby_name}" if ["size", "stride", "random!"].include?(ruby_name)
   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
+  # cast for Ruby < 3.0 https://github.com/thisMagpie/fftw/issues/22#issuecomment-49508900
   cast = RUBY_VERSION.to_f > 2.7 ? "" : "(VALUE (*)(...)) "
 
   "rb_#{def_method}(m, \"#{ruby_name}\", #{cast}#{full_name(name, type)}, -1);"
@@ -250,7 +260,7 @@ def generate_dispatch(function, def_method)
 
   cpp_params = generate_dispatch_params(function, params)
   if opt_index
-    cpp_params.insert(remove_self ? opt_index + 1 : opt_index, "const TensorOptions & options")
+    cpp_params.insert(remove_self ? opt_index + 1 : opt_index, "TensorOptions options")
   end
 
   retval = generate_dispatch_retval(function)
@@ -410,7 +420,7 @@ def generate_function_params(function, params, remove_self)
           else
             "optionalTensor"
           end
-        when "generator", "tensorlist", "intlist"
+        when "generator", "tensorlist"
           func
         when "string"
           "stringViewOptional"
@@ -471,7 +481,11 @@ def generate_dispatch_params(function, params)
       when "float"
         "double"
       when /\Aint\[/
-        "IntArrayRef"
+        if param[:optional]
+          "at::OptionalIntArrayRef"
+        else
+          "IntArrayRef"
+        end
       when "float[]"
         "ArrayRef<double>"
       when "str"
@@ -480,13 +494,19 @@ def generate_dispatch_params(function, params)
         else
           "std::string"
         end
-      when "Scalar", "bool", "ScalarType", "Layout", "Device", "Storage", "Generator", "MemoryFormat", "Storage"
+      when "Scalar"
+        if param[:optional]
+          "const c10::optional<Scalar> &"
+        else
+          "const Scalar &"
+        end
+      when "bool", "ScalarType", "Layout", "Device", "Storage", "Generator", "MemoryFormat", "Storage"
         param[:type]
       else
         raise "Unknown type: #{param[:type]} (#{function.name})"
       end
 
-    if param[:optional] && param[:type] != "Tensor"
+    if param[:optional] && !["Tensor", "Scalar"].include?(param[:type]) && !param[:type].start_with?("int[")
       type = "c10::optional<#{type}>"
     end
 
@@ -529,7 +549,7 @@ def generate_dispatch_retval(function)
   when ["float", "float"]
     "std::tuple<double,double>"
   else
-    raise "Unknown retvals: #{types}"
+    raise "Unknown retvals: #{types} (#{function.name})"
   end
 end