torch-rb 0.16.0 → 0.17.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8442fc0f85d6f2465258a54e5aefbe03d23a7c0e58753e06855bfebd2f4de802
-  data.tar.gz: ac0efb89f9b6d413498bfb1c2e84336aa728047dd013d00fa736449e5be82617
+  metadata.gz: f04aeddd9c66d668f78ed38d831ec66907743fd5be2cd076e9e547d00f1f2aa3
+  data.tar.gz: e76ca5199c4801c0eaa760ed3e41ebb6e521774bba084809656b1ed9e4b96483
 SHA512:
-  metadata.gz: 6830efe74de98fc8a8d23e7308795a60ee60fff72b3f82fa7cb92815f4efe52fdf3637e0821490f5e3e8c2c8731043f52f5aff20cfb01db1340be0962fed18db
-  data.tar.gz: 3e50976e5add37b4158956c76e3c922167911492acda9e171af42ad39d5abe946c36427e545d9fc820a2800e3df0523b0068ce76b804d0c05a6f1e2ad495de01
+  metadata.gz: 683246aa91d19daa624a9d3a89fa5e4898f19fbb45a72ff2bcac0ee3025859f0f8ad26ac153e58150aedf10547cf670e2fe131cffda246546ded1f1344637040
+  data.tar.gz: bd0b955b997f09a274d8b59a21eafa8bf45972b6a20883b9d457a29417aa2c6d413b324abadfb6772c5bdc3aa0e633d3e25d6f23dd986fa603c55c69ae4a5a85

data/CHANGELOG.md

@@ -1,3 +1,14 @@
+## 0.17.1 (2024-08-19)
+
+- Added `persistent` option to `register_buffer` method
+- Added `prefix` and `recurse` options to `named_buffers` method
+
+## 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

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,7 @@ 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

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