torch-rb 0.12.2 → 0.13.0

data/ext/torch/backends.cpp

@@ -14,4 +14,8 @@ void init_backends(Rice::Module& m) {
   Rice::define_module_under(rb_mBackends, "MKL")
     .add_handler<torch::Error>(handle_error)
     .define_singleton_function("available?", &torch::hasMKL);
+
+  Rice::define_module_under(rb_mBackends, "MPS")
+    .add_handler<torch::Error>(handle_error)
+    .define_singleton_function("available?", &torch::hasMPS);
 }

data/ext/torch/tensor.cpp

@@ -35,17 +35,17 @@ std::vector<TensorIndex> index_vector(Array a) {
     if (obj.is_instance_of(rb_cInteger)) {
       indices.push_back(Rice::detail::From_Ruby<int64_t>().convert(obj.value()));
     } else if (obj.is_instance_of(rb_cRange)) {
-      torch::optional<int64_t> start_index = torch::nullopt;
-      torch::optional<int64_t> stop_index = torch::nullopt;
+      torch::optional<c10::SymInt> start_index = torch::nullopt;
+      torch::optional<c10::SymInt> stop_index = torch::nullopt;
 
       Object begin = obj.call("begin");
       if (!begin.is_nil()) {
-        start_index = Rice::detail::From_Ruby<int64_t>().convert(begin.value());
+        start_index = c10::SymInt(Rice::detail::From_Ruby<int64_t>().convert(begin.value()));
       }
 
       Object end = obj.call("end");
       if (!end.is_nil()) {
-        stop_index = Rice::detail::From_Ruby<int64_t>().convert(end.value());
+        stop_index = c10::SymInt(Rice::detail::From_Ruby<int64_t>().convert(end.value()));
       }
 
       Object exclude_end = obj.call("exclude_end?");

data/ext/torch/utils.h

@@ -6,7 +6,7 @@
 #include <rice/stl.hpp>
 
 static_assert(
-  TORCH_VERSION_MAJOR == 1 && TORCH_VERSION_MINOR == 13,
+  TORCH_VERSION_MAJOR == 2 && TORCH_VERSION_MINOR == 0,
   "Incompatible LibTorch version"
 );
 

data/lib/torch/version.rb

@@ -1,3 +1,3 @@
 module Torch
-  VERSION = "0.12.2"
+  VERSION = "0.13.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: torch-rb
 version: !ruby/object:Gem::Version
-  version: 0.12.2
+  version: 0.13.0
 platform: ruby
 authors:
 - Andrew Kane
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-01-31 00:00:00.000000000 Z
+date: 2023-04-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rice
@@ -230,14 +230,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '2.7'
+      version: '3'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.1
+rubygems_version: 3.4.10
 signing_key:
 specification_version: 4
 summary: Deep learning for Ruby, powered by LibTorch