ext/torch/torch.cpp in torch-rb-0.13.0 vs ext/torch/torch.cpp in torch-rb-0.13.1
- old
+ new
@@ -22,10 +22,10 @@
return t.reshape(size);
}
void init_torch(Rice::Module& m) {
- m.add_handler<torch::Error>(handle_error);
+ register_handler<torch::Error>(handle_global_error);
add_torch_functions(m);
m.define_singleton_function(
"grad_enabled?",
[]() {
return torch::GradMode::is_enabled();