torch-rb 0.22.1 → 0.22.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d15801cc8463969ebf374f668980c55da64203b973d673895a7315ae150387fd
-  data.tar.gz: 1d4d3bd337f0b8645d642f92b74286996381dc1e3dad1f0d652f5e900cf60bcd
+  metadata.gz: 13a292607c0a4ecc21607db5128d779a2dbd349f9372fe7e140576af4b6e9e95
+  data.tar.gz: 5ce06edf45ef0c4e05d3e77b7517c3c15c3400c36d2b264afd6252bbd6de2ab1
 SHA512:
-  metadata.gz: 52572a441989fcdb66840785354d6f5e5d9dfe431ead839ff6f38c9325bc3ec2c02c035b06adb6c3aeb2fa1c76db365ccf1d5f236f47bf97073a799a117ef7fd
-  data.tar.gz: f156700996b6517b81d268477551c197b5119f910e6fe81dab92fea27d633bed82476fee679fe9c32231ef252b0fa19eb311ab670480816324d7074fcf3d8fef
+  metadata.gz: 24e8f8557835301ed533bcd52eaf22a5bad643d267f2cdc8d9f49c8b6306d844d4a12411e1f32bf0512d54b8f3637aed507fa22d0b8d3c6db3633f5a24a23470
+  data.tar.gz: 39f7d61b3f39dde83ff24eeef19996f907e3943017d2af7c16c1f18b22c1538f208be78e990177e70f33a7dffa40d87bdce0e740307fac2f7c8c8bf3e0ffea5e

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+## 0.22.2 (2025-11-04)
+
+- Added support for CUDA 12.9+
+
 ## 0.22.1 (2025-10-26)
 
 - Fixed error with Rice 4.7

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/lib/torch/version.rb

@@ -1,3 +1,3 @@
 module Torch
-  VERSION = "0.22.1"
+  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.1
+  version: 0.22.2
 platform: ruby
 authors:
 - Andrew Kane