torch-rb 0.21.0 → 0.22.0

data/ext/torch/ext.cpp

@@ -16,10 +16,14 @@ void init_generator(Rice::Module& m, Rice::Class& rb_cGenerator);
 void init_ivalue(Rice::Module& m, Rice::Class& rb_cIValue);
 void init_random(Rice::Module& m);
 
+VALUE rb_eTorchError = Qnil;
+
 extern "C"
 void Init_ext() {
   auto m = Rice::define_module("Torch");
 
+  rb_eTorchError = Rice::define_class_under(m, "Error", rb_eStandardError);
+
   // need to define certain classes up front to keep Rice happy
   auto rb_cIValue = Rice::define_class_under<torch::IValue>(m, "IValue")
     .define_constructor(Rice::Constructor<torch::IValue>());

data/ext/torch/templates.h

@@ -31,7 +31,7 @@ using torch::nn::init::NonlinearityType;
 
 #define END_HANDLE_TH_ERRORS                                         \
   } catch (const torch::Error& ex) {                                 \
-    rb_raise(rb_eRuntimeError, "%s", ex.what_without_backtrace());   \
+    rb_raise(rb_eTorchError, "%s", ex.what_without_backtrace());   \
   } catch (const Rice::Exception& ex) {                              \
     rb_raise(ex.class_of(), "%s", ex.what());                        \
   } catch (const std::exception& ex) {                               \

data/ext/torch/utils.h

@@ -8,12 +8,14 @@
 #include <rice/stl.hpp>
 
 static_assert(
-  TORCH_VERSION_MAJOR == 2 && TORCH_VERSION_MINOR == 8,
+  TORCH_VERSION_MAJOR == 2 && TORCH_VERSION_MINOR == 9,
   "Incompatible LibTorch version"
 );
 
+extern VALUE rb_eTorchError;
+
 inline void handle_global_error(const torch::Error& ex) {
-  throw Rice::Exception(rb_eRuntimeError, ex.what_without_backtrace());
+  throw Rice::Exception(rb_eTorchError, ex.what_without_backtrace());
 }
 
 // keep THP prefix for now to make it easier to compare code

data/lib/torch/version.rb

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

data/lib/torch.rb

@@ -210,7 +210,6 @@ require_relative "torch/utils/data/tensor_dataset"
 require_relative "torch/hub"
 
 module Torch
-  class Error < StandardError; end
   class NotImplementedYet < StandardError
     def message
       "This feature has not been implemented yet. Consider submitting a PR."

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: torch-rb
 version: !ruby/object:Gem::Version
-  version: 0.21.0
+  version: 0.22.0
 platform: ruby
 authors:
 - Andrew Kane