netcdf-mpi

unified_allocator.hh
View Options

			/**
			* @file
			*
			* @author Lucas Frérot <lucas.frerot@epfl.ch>
			*
			* @section LICENSE
			*
			* Copyright (©) 2017 EPFL (Ecole Polytechnique Fédérale de
			* Lausanne) Laboratory (LSMS - Laboratoire de Simulation en Mécanique des
			* Solides)
			*
			* Tamaas is free software: you can redistribute it and/or modify it under the
			* terms of the GNU Lesser General Public License as published by the Free
			* Software Foundation, either version 3 of the License, or (at your option) any
			* later version.
			*
			* Tamaas is distributed in the hope that it will be useful, but WITHOUT ANY
			* WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
			* A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
			* details.
			*
			* You should have received a copy of the GNU Lesser General Public License
			* along with Tamaas. If not, see <http://www.gnu.org/licenses/>.
			*
			*/
			/* -------------------------------------------------------------------------- */
			#ifndef __UNIFIED_ALLOCATOR_HH__
			#define __UNIFIED_ALLOCATOR_HH__
			/* -------------------------------------------------------------------------- */
			#include <memory>
			/* -------------------------------------------------------------------------- */

			namespace tamaas {

			template <typename T>
			struct UnifiedAllocator : public std::allocator<T> {
			T * allocate(std::size_t n) {
			T * p = nullptr;
			cudaMallocManaged(p, n * sizeof(T));
			return p;
			}

			void deallocate(T * p, __attribute__((unused)) std::size_t n) {
			cudaFree(p);
			}
			};

			}

			#endif // __UNIFIED_ALLOCATOR_HH__