pytorch-sbs/network/CPP_Cuda_new_preview/PyMultiApp.cpp
2023-01-13 21:33:57 +01:00

15 lines
No EOL
409 B
C++

#include <pybind11/pybind11.h>
#include "MultiApp.h"
namespace py = pybind11;
PYBIND11_MODULE(PyMultiApp, m) {
m.doc() = "MultiApp Module";
py::class_<MultiApp>(m, "MultiApp")
.def(py::init<>())
.def("gpu_occupancy_export", &MultiApp::gpu_occupancy_export)
.def("gpu_occupancy_import", &MultiApp::gpu_occupancy_import)
.def("update_entrypoint", &MultiApp::update_entrypoint);
}