torch-rb 0.10.0 → 0.11.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7884d0bd8ffd23e775b3a1ca59e536bc18cd79c6bbe2736a9802a2b1693d40de
-  data.tar.gz: 10849c8a248a70eca7178ca3529fa7428fc3d8e5cfe2ca774b0be6cb5e75eb9b
+  metadata.gz: d2f88c938144476a772fd7c606751e0a6e67a338cb1c37ede9c2011db7fc4579
+  data.tar.gz: f4408457e3bf8c7bf9b42863459248aa592c70e8e3924cdbe72863a979e65106
 SHA512:
-  metadata.gz: d81587eae00527e9d1e4a65b62a686dbdab27eed9401539faddf553d3a0730ad7394b01bac615bdd0474d1a6602cd0a3117e1d7cb3a3f1d5fbf8bd989fa39e59
-  data.tar.gz: d581d07821f103ee69267bc8e6b15d5094d8b8ef004917af947e2570649e7fbe750c9893d40600b62ee970646654fc545a270e7a59adb764be2ba73f1fa18b62
+  metadata.gz: 90ebc506942809b02331f7accce6e93e714a57b5d2f58b06ad0d3c60204b947eba61fe0bf43431a19e27890f5c764cb348ccb1785a2ffccfba7969f4726b6f1f
+  data.tar.gz: 410af7b4934f79aaae94f6c1bde1d17ee8b043f02354b4ea9f139271575f43abb90afeba162badfa01a3ec406612ae6cdcb4a2adfb1067b1634982a590d10cc9

data/CHANGELOG.md

@@ -1,3 +1,17 @@
+## 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
+
 ## 0.10.0 (2022-03-13)
 
 - Updated LibTorch to 1.11.0

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
 
@@ -250,7 +257,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)
@@ -293,7 +300,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]
@@ -304,12 +317,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"
@@ -392,7 +417,7 @@ def generate_function_params(function, params, remove_self)
           else
             "optionalTensor"
           end
-        when "generator", "tensorlist", "intlist"
+        when "generator", "tensorlist"
           func
         when "string"
           "stringViewOptional"
@@ -453,7 +478,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"
@@ -462,13 +491,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
 
@@ -511,7 +546,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