Plugins/src/Vecmem.cpp

0001 // This file is part of the ACTS project.
0002 //
0003 // Copyright (C) 2016 CERN for the benefit of the ACTS project
0004 //
0005 // This Source Code Form is subject to the terms of the Mozilla Public
0006 // License, v. 2.0. If a copy of the MPL was not distributed with this
0007 // file, You can obtain one at https://mozilla.org/MPL/2.0/.
0008
0009 #include "ActsPython/Utilities/Helpers.hpp"
0010 #include "ActsPython/Utilities/Macros.hpp"
0011
0012 #include <memory>
0013 #include <string>
0014
0015 #include <vecmem/memory/host_memory_resource.hpp>
0016 #include <vecmem/memory/memory_resource.hpp>
0017
0018 namespace py = pybind11;
0019 using namespace pybind11::literals;
0020
0021 PYBIND11_MODULE(ActsPluginsPythonBindingsVecmem, vecmem) {
0022   {
0023     py::class_<vecmem::memory_resource,
0024                std::shared_ptr<vecmem::memory_resource>>(vecmem,
0025                                                          "MemoryResource");
0026
0027     py::class_<vecmem::host_memory_resource, vecmem::memory_resource,
0028                std::shared_ptr<vecmem::host_memory_resource>>(
0029         vecmem, "HostMemoryResource")
0030         .def(py::init<>());
0031   }
0032 }