torch-rb 0.22.0 → 0.22.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c6d84711879346fb2ee772ba127520ab533b5171a05a0609aa5ed96ca244db4d
-  data.tar.gz: 0ad9c5a15d4e645240061fa19f3d56eb40770dfdc3594a1306ffde0c17792d5a
+  metadata.gz: 13a292607c0a4ecc21607db5128d779a2dbd349f9372fe7e140576af4b6e9e95
+  data.tar.gz: 5ce06edf45ef0c4e05d3e77b7517c3c15c3400c36d2b264afd6252bbd6de2ab1
 SHA512:
-  metadata.gz: be84a03edc6bc7c62ac42ada37e104ed4df05f9fb7588730f2b8e5e9f6b2137dd7793b6bb71c85cd46b4ef8020c95fc9fef07aa2d4de86999c7ef683e955ad86
-  data.tar.gz: 669b7cc9d525c3a305d6c1eead7a65d0290da32c40049653d5f4a0c4fcc5d9650234b1f3c6027e91465d950944a746e738daea9ec775b02fdaffe33978feb650
+  metadata.gz: 24e8f8557835301ed533bcd52eaf22a5bad643d267f2cdc8d9f49c8b6306d844d4a12411e1f32bf0512d54b8f3637aed507fa22d0b8d3c6db3633f5a24a23470
+  data.tar.gz: 39f7d61b3f39dde83ff24eeef19996f907e3943017d2af7c16c1f18b22c1538f208be78e990177e70f33a7dffa40d87bdce0e740307fac2f7c8c8bf3e0ffea5e

data/CHANGELOG.md

@@ -1,3 +1,11 @@
+## 0.22.2 (2025-11-04)
+
+- Added support for CUDA 12.9+
+
+## 0.22.1 (2025-10-26)
+
+- Fixed error with Rice 4.7
+
 ## 0.22.0 (2025-10-15)
 
 - Updated LibTorch to 2.9.0

data/ext/torch/extconf.rb

@@ -33,9 +33,11 @@ unless inc && lib
 end
 
 cuda_inc, cuda_lib = dir_config("cuda")
-cuda_inc ||= "/usr/local/cuda/include"
 cuda_lib ||= "/usr/local/cuda/lib64"
 
+cudnn_inc, cudnn_lib = dir_config("cudnn")
+cudnn_lib ||= "/usr/local/cuda/lib"
+
 $LDFLAGS += " -L#{lib}" if Dir.exist?(lib)
 abort "LibTorch not found" unless have_library("torch")
 
@@ -45,6 +47,7 @@ have_library("nnpack")
 with_cuda = false
 if Dir["#{lib}/*torch_cuda*"].any?
   $LDFLAGS += " -L#{cuda_lib}" if Dir.exist?(cuda_lib)
+  $LDFLAGS += " -L#{cudnn_lib}" if Dir.exist?(cudnn_lib) && cudnn_lib != cuda_lib
   with_cuda = have_library("cuda") && have_library("cudnn")
 end
 
@@ -57,10 +60,12 @@ if RbConfig::CONFIG["host_os"] =~ /darwin/i && RbConfig::CONFIG["host_cpu"] =~ /
 end
 $LDFLAGS += ":#{cuda_lib}/stubs:#{cuda_lib}" if with_cuda
 
-# https://github.com/pytorch/pytorch/blob/v1.5.0/torch/utils/cpp_extension.py#L1232-L1238
+# https://github.com/pytorch/pytorch/blob/v2.9.0/torch/utils/cpp_extension.py#L1351-L1364
 $LDFLAGS += " -lc10 -ltorch_cpu -ltorch"
 if with_cuda
-  $LDFLAGS += " -lcuda -lnvrtc -lnvToolsExt -lcudart -lc10_cuda -ltorch_cuda -lcufft -lcurand -lcublas -lcudnn"
+  $LDFLAGS += " -lcuda -lnvrtc"
+  $LDFLAGS += " -lnvToolsExt" if File.exist?("#{cuda_lib}/libnvToolsExt.so")
+  $LDFLAGS += " -lcudart -lc10_cuda -ltorch_cuda -lcufft -lcurand -lcublas -lcudnn"
   # TODO figure out why this is needed
   $LDFLAGS += " -Wl,--no-as-needed,#{lib}/libtorch.so"
 end

data/ext/torch/ivalue.cpp

@@ -55,7 +55,7 @@ void init_ivalue(Rice::Module& m, Rice::Class& rb_cIValue) {
         Rice::Array obj;
         for (auto& elem : list) {
           auto v = torch::IValue{elem};
-          obj.push(Rice::Object(Rice::detail::To_Ruby<torch::IValue>().convert(v)));
+          obj.push(Rice::Object(Rice::detail::To_Ruby<torch::IValue>().convert(v)), false);
         }
         return obj;
       })

data/ext/torch/templates.h

@@ -50,14 +50,25 @@ namespace Rice::detail {
   template<typename T>
   class To_Ruby<c10::complex<T>> {
   public:
+    To_Ruby() = default;
+
+    explicit To_Ruby(Arg* arg) : arg_(arg) { }
+
     VALUE convert(c10::complex<T> const& x) {
       return rb_dbl_complex_new(x.real(), x.imag());
     }
+
+  private:
+    Arg* arg_ = nullptr;
   };
 
   template<typename T>
   class From_Ruby<c10::complex<T>> {
   public:
+    From_Ruby() = default;
+
+    explicit From_Ruby(Arg* arg) : arg_(arg) { }
+
     Convertible is_convertible(VALUE value) { return Convertible::Cast; }
 
     c10::complex<T> convert(VALUE x) {
@@ -65,6 +76,9 @@ namespace Rice::detail {
       VALUE imag = rb_funcall(x, rb_intern("imag"), 0);
       return c10::complex<T>(From_Ruby<T>().convert(real), From_Ruby<T>().convert(imag));
     }
+
+  private:
+    Arg* arg_ = nullptr;
   };
 
   template<>
@@ -75,6 +89,10 @@ namespace Rice::detail {
   template<>
   class From_Ruby<FanModeType> {
   public:
+    From_Ruby() = default;
+
+    explicit From_Ruby(Arg* arg) : arg_(arg) { }
+
     Convertible is_convertible(VALUE value) { return Convertible::Cast; }
 
     FanModeType convert(VALUE x) {
@@ -87,6 +105,9 @@ namespace Rice::detail {
         throw std::runtime_error("Unsupported nonlinearity type: " + s);
       }
     }
+
+  private:
+    Arg* arg_ = nullptr;
   };
 
   template<>
@@ -97,6 +118,10 @@ namespace Rice::detail {
   template<>
   class From_Ruby<NonlinearityType> {
   public:
+    From_Ruby() = default;
+
+    explicit From_Ruby(Arg* arg) : arg_(arg) { }
+
     Convertible is_convertible(VALUE value) { return Convertible::Cast; }
 
     NonlinearityType convert(VALUE x) {
@@ -127,6 +152,9 @@ namespace Rice::detail {
         throw std::runtime_error("Unsupported nonlinearity type: " + s);
       }
     }
+
+  private:
+    Arg* arg_ = nullptr;
   };
 
   template<>
@@ -137,6 +165,10 @@ namespace Rice::detail {
   template<>
   class From_Ruby<Scalar> {
   public:
+    From_Ruby() = default;
+
+    explicit From_Ruby(Arg* arg) : arg_(arg) { }
+
     Convertible is_convertible(VALUE value) { return Convertible::Cast; }
 
     Scalar convert(VALUE x) {
@@ -146,5 +178,8 @@ namespace Rice::detail {
         return torch::Scalar(From_Ruby<double>().convert(x));
       }
     }
+
+  private:
+    Arg* arg_ = nullptr;
   };
 } // namespace Rice::detail

data/ext/torch/tensor.cpp

@@ -20,7 +20,7 @@ Array flat_data(Tensor& tensor) {
 
   Array a;
   for (int i = 0; i < tensor.numel(); i++) {
-    a.push(view[i].item().to<T>());
+    a.push(view[i].item().to<T>(), false);
   }
   return a;
 }
@@ -129,7 +129,7 @@ void init_tensor(Rice::Module& m, Rice::Class& c, Rice::Class& rb_cTensorOptions
       [](Tensor& self) {
         Array a;
         for (auto &size : self.sizes()) {
-          a.push(size);
+          a.push(size, false);
         }
         return a;
       })
@@ -138,7 +138,7 @@ void init_tensor(Rice::Module& m, Rice::Class& c, Rice::Class& rb_cTensorOptions
       [](Tensor& self) {
         Array a;
         for (auto &stride : self.strides()) {
-          a.push(stride);
+          a.push(stride, false);
         }
         return a;
       })

data/lib/torch/version.rb

@@ -1,3 +1,3 @@
 module Torch
-  VERSION = "0.22.0"
+  VERSION = "0.22.2"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: torch-rb
 version: !ruby/object:Gem::Version
-  version: 0.22.0
+  version: 0.22.2
 platform: ruby
 authors:
 - Andrew Kane
@@ -15,14 +15,14 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '4.5'
+        version: '4.7'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '4.5'
+        version: '4.7'
 email: andrew@ankane.org
 executables: []
 extensions: