torch-rb 0.15.0 → 0.17.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5cb7d8bf760b3f2a52976b0fda929cd7227a89ed7475b19a87cd9ce53ab5fe4e
-  data.tar.gz: 9977b9740c8490b4be11682dbd700bfe1175cf1f109e81412a694c4a4fd4a043
+  metadata.gz: cd28e938ac0d61829d538f4a36e9f3377f06257ab08fd3ec8d8e3912194a101a
+  data.tar.gz: 7bbb0c98e783c4a02e4ca46a1835c591fff3630c84d1a44abf1ef8df75c45de7
 SHA512:
-  metadata.gz: fb7283603fff1ad25fd234926abaeaafada2f0def3b5a52051242f3348445a5e4b615680f414d09c72a502c355fade701e79867444dd848c8dcd989f42359f19
-  data.tar.gz: bb41a341f271fb5bce47ad6e1a6ef223b6d7c01165a86de13681d25720c96d5a17de81e549439feedcf482c4e539083e3e04e97a768999438b939bc235fb652d
+  metadata.gz: 38415fe461d01d11d620d2072012ea6a86ad000e5d25920af3d2d4cd05c3b088479cb60a2bd2b6dc013da51e3f744913b3d734fc949ac2708b825c3c041f8c52
+  data.tar.gz: 2c54b30ccfa91f9b7576f7927377c03a9d82b7ff8aaf561e92ad89b2b00a2e7e3e08c268d606e6609cbaa13d2dc2e5e11c8329bf1b3475af7025a21dc8a1921f

data/CHANGELOG.md

@@ -1,3 +1,15 @@
+## 0.17.0 (2024-07-26)
+
+- Updated LibTorch to 2.4.0
+- Added `normalize` method
+- Added support for tensor indexing with arrays
+
+## 0.16.0 (2024-06-12)
+
+- Updated LibTorch to 2.3.0
+- Added `ELU` and `GELU` classes
+- Dropped support for Ruby < 3.1
+
 ## 0.15.0 (2024-02-28)
 
 - Updated LibTorch to 2.2.0

data/README.md

@@ -26,7 +26,7 @@ Add this line to your application’s Gemfile:
 gem "torch-rb"
 ```
 
-It can take 5-10 minutes to compile the extension.
+It can take 5-10 minutes to compile the extension. Windows is not currently supported.
 
 ## Getting Started
 
@@ -410,6 +410,8 @@ Here’s the list of compatible versions.
 
 Torch.rb | LibTorch
 --- | ---
+0.17.x | 2.4.x
+0.16.x | 2.3.x
 0.15.x | 2.2.x
 0.14.x | 2.1.x
 0.13.x | 2.0.x

data/codegen/generate_functions.rb

@@ -155,10 +155,10 @@ def generate_attach_def(name, type, def_method)
     end
 
   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 = ruby_name.delete_prefix("fft_") if type == "fft"
+  ruby_name = ruby_name.delete_prefix("linalg_") if type == "linalg"
+  ruby_name = ruby_name.delete_prefix("special_") if type == "special"
+  ruby_name = ruby_name.delete_prefix("sparse_") if type == "sparse"
   ruby_name = name if name.start_with?("__")
 
   "rb_#{def_method}(m, \"#{ruby_name}\", #{full_name(name, type)}, -1);"
@@ -216,7 +216,7 @@ def add_dispatch(function, def_method)
     out_code = generate_dispatch(function["out"], def_method)
     out_index = function["out"].out_index
 
-    return "if (_r.isNone(#{out_index})) {
+    "if (_r.isNone(#{out_index})) {
     #{indent(base_code)}
   } else {
     #{indent(out_code)}
@@ -439,7 +439,7 @@ def generate_function_params(function, params, remove_self)
         else
           "#{func}Optional"
         end
-      end
+    end
 
     "_r.#{func}(#{param[:position]})"
   end