torch-rb 0.12.2 → 0.13.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8160298f6299869f699201dd9217785da5ac7d2562f05b12a2df2d4537143886
-  data.tar.gz: cb902f45cb72adaaa2008e181af2af0844e926ef62f1d2ccaa67251cb70c5799
+  metadata.gz: 311e86910351cc5050fb92146e1f6c9ad018a9609d0f5ca0ef51253f1b813b22
+  data.tar.gz: afc18a5142abecba2fdd17f669ecdf9de5b90f35e8c79a637c4e960427a08439
 SHA512:
-  metadata.gz: c5077c40cb32414ac7d230430fdcf1cd93f199a81100179a1aa26185d6216c049316bbf35d79968442fd4db5e18414991946a366e6e7e6260a904f001774faa2
-  data.tar.gz: b4f1ff33b47596953c0654b07f1916a88ba247af72c1d4693d9b737802a790256d3232c0419c3a603fdebd51f47360833265326a1f27f8242604ff05766ea690
+  metadata.gz: 29df29a3dd0f752f4da731b84f0ca40832d49510a76784364d607977c8e90e32074f7e17ea07167e79330a97ef5f3378b19331d326301a3cc5a0e76180384ac6
+  data.tar.gz: 50e96247102c02cb6f5b0ddc470947f06dfb4efcc3a675ee174a334c4263c1dfe9f3be9c5bfb249239b8b4abce79f6e6dc625833081ec101180ba9a757f3421c

data/CHANGELOG.md

@@ -1,3 +1,12 @@
+## 0.13.1 (2023-05-03)
+
+- Fixed error with Rice 4.1
+
+## 0.13.0 (2023-04-13)
+
+- Updated LibTorch to 2.0.0
+- Dropped support for Ruby < 3
+
 ## 0.12.2 (2023-01-30)
 
 - Added experimental support for DataPipes

data/README.md

@@ -8,6 +8,7 @@ Check out:
 - [TorchText](https://github.com/ankane/torchtext) for text and NLP tasks
 - [TorchAudio](https://github.com/ankane/torchaudio) for audio tasks
 - [TorchRec](https://github.com/ankane/torchrec-ruby) for recommendation systems
+- [TorchData](https://github.com/ankane/torchdata-ruby) for data loading
 
 [![Build Status](https://github.com/ankane/torch.rb/workflows/build/badge.svg?branch=master)](https://github.com/ankane/torch.rb/actions)
 
@@ -409,17 +410,12 @@ Here’s the list of compatible versions.
 
 Torch.rb | LibTorch
 --- | ---
-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
-0.6.0-0.7.0 | 1.8.0-1.8.1
-0.5.0-0.5.3 | 1.7.0-1.7.1
-0.3.0-0.4.2 | 1.6.0
-0.2.0-0.2.7 | 1.5.0-1.5.1
-0.1.8 | 1.4.0
-0.1.0-0.1.7 | 1.3.1
+0.13.x | 2.0.x
+0.12.x | 1.13.x
+0.11.x | 1.12.x
+0.10.x | 1.11.x
+0.9.x | 1.10.x
+0.8.x | 1.9.x
 
 ### Homebrew
 
@@ -431,7 +427,11 @@ brew install pytorch
 
 ## 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.
+Deep learning is significantly faster on a GPU.
+
+### Linux
+
+With Linux, install [CUDA](https://developer.nvidia.com/cuda-downloads) and [cuDNN](https://developer.nvidia.com/cudnn) and reinstall the gem.
 
 Check if CUDA is available
 
@@ -453,6 +453,21 @@ ankane/ml-stack:torch-gpu
 
 And leave the other fields in that section blank. Once the notebook is running, you can run the [MNIST example](https://github.com/ankane/ml-stack/blob/master/torch-gpu/MNIST.ipynb).
 
+### Mac
+
+With Apple silicon, check if Metal Performance Shaders (MPS) is available
+
+```ruby
+Torch::Backends::MPS.available?
+```
+
+Move a neural network to a GPU
+
+```ruby
+device = Torch.device("mps")
+net.to(device)
+```
+
 ## History
 
 View the [changelog](https://github.com/ankane/torch.rb/blob/master/CHANGELOG.md)

data/codegen/generate_functions.rb

@@ -156,10 +156,7 @@ def generate_attach_def(name, type, def_method)
   ruby_name = ruby_name.sub(/\Asparse_/, "") if type == "sparse"
   ruby_name = name if name.start_with?("__")
 
-  # 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);"
+  "rb_#{def_method}(m, \"#{ruby_name}\", #{full_name(name, type)}, -1);"
 end
 
 def generate_method_def(name, functions, type, def_method)
@@ -415,6 +412,8 @@ def generate_function_params(function, params, remove_self)
         "memoryformat"
       when "Storage"
         "storage"
+      when "Layout"
+        "layout"
       else
         raise "Unknown type: #{param[:type]} (#{function.name})"
       end
@@ -445,7 +444,7 @@ 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.dispatch_name
+  dispatch = nil # function.dispatch_name
   unless dispatch
     dispatch = function.base_name
     dispatch += "_symint" if function.func.include?("SymInt")
@@ -640,7 +639,7 @@ def signature_type(param)
     when "int"
       "int64_t"
     when "SymInt"
-      "c10::SymInt"
+      "SymInt"
     when "float"
       "double"
     when "str"