ext/torch/backends.cpp in torch-rb-0.12.2 vs ext/torch/backends.cpp in torch-rb-0.13.0
- old
+ new
@@ -12,6 +12,10 @@
.define_singleton_function("available?", &torch::hasOpenMP);
Rice::define_module_under(rb_mBackends, "MKL")
.add_handler<torch::Error>(handle_error)
.define_singleton_function("available?", &torch::hasMKL);
+
+ Rice::define_module_under(rb_mBackends, "MPS")
+ .add_handler<torch::Error>(handle_error)
+ .define_singleton_function("available?", &torch::hasMPS);
}