device/cuda/cuda_test_fixture.hpp

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 #pragma once
0010
0011 // VecMem include(s).
0012 #include <vecmem/memory/cuda/managed_memory_resource.hpp>
0013
0014 // GoogleTest include(s).
0015 #include <gtest/gtest.h>
0016
0017 namespace detray::test::cuda {
0018
0019 /// CUDA specific test fixture class
0020 template <class base_fixture_t>
0021 class cuda_test_fixture : public base_fixture_t {
0022  public:
0023   /// Constructor, providing CUDA specific infrastructure to the test fixture
0024   cuda_test_fixture() : base_fixture_t(m_resource) {}
0025
0026  protected:
0027   /// Memory resource for all of the tests.
0028   vecmem::cuda::managed_memory_resource m_resource;
0029 };
0030
0031 }  // namespace detray::test::cuda