ext/torch/cuda.cpp in torch-rb-0.13.0 vs ext/torch/cuda.cpp in torch-rb-0.13.1
- old
+ new
@@ -4,10 +4,9 @@
#include "utils.h"
void init_cuda(Rice::Module& m) {
Rice::define_module_under(m, "CUDA")
- .add_handler<torch::Error>(handle_error)
.define_singleton_function("available?", &torch::cuda::is_available)
.define_singleton_function("device_count", &torch::cuda::device_count)
.define_singleton_function("manual_seed", &torch::cuda::manual_seed)
.define_singleton_function("manual_seed_all", &torch::cuda::manual_seed_all);
}