xref: /aosp_15_r20/external/pytorch/test/cpp_extensions/torch_library.cu (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 #include <torch/extension.h>
2 
logical_and(bool a,bool b)3 bool logical_and(bool a, bool b) { return a && b; }
4 
TORCH_LIBRARY(torch_library,m)5 TORCH_LIBRARY(torch_library, m) {
6   m.def("logical_and", &logical_and);
7 }
8 
9 struct CuaevComputer : torch::CustomClassHolder {};
10 
TORCH_LIBRARY(cuaev,m)11 TORCH_LIBRARY(cuaev, m) {
12   m.class_<CuaevComputer>("CuaevComputer");
13 }
14 
PYBIND11_MODULE(TORCH_EXTENSION_NAME,m)15 PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {}
16