Line data    Source code

       1             : //* This file is part of the MOOSE framework
       2             : //* https://www.mooseframework.org
       3             : //*
       4             : //* All rights reserved, see COPYRIGHT for full restrictions
       5             : //* https://github.com/idaholab/moose/blob/master/COPYRIGHT
       6             : //*
       7             : //* Licensed under LGPL 2.1, please see LICENSE for details
       8             : //* https://www.gnu.org/licenses/lgpl-2.1.html
       9             : 
      10             : #include "KokkosMemoryPool.h"
      11             : 
      12             : namespace Moose::Kokkos
      13             : {
      14             : 
      15           0 : MemoryPool::MemoryPool(dof_id_type size, unsigned int ways)
      16             : {
      17           0 :   MemSpace::memory_space space;
      18             : 
      19           0 :   _pools.create(ways);
      20             : 
      21             :   // clang-format off
      22             : 
      23           0 :   for (auto & pool : _pools)
      24           0 :     new (&pool) ::Kokkos::MemoryPool<MemSpace>(space, size / ways);
      25             : 
      26             :   // clang-format on
      27             : 
      28           0 :   _pools.copyToDevice();
      29           0 : }
      30             : 
      31             : } // namespace Moose::Kokkos