Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F65460847
unified_allocator.hh
No One
Temporary
Actions
Download File
Edit File
Delete File
View Transforms
Subscribe
Mute Notifications
Award Token
Subscribers
None
File Metadata
Details
File Info
Storage
Attached
Created
Mon, Jun 3, 23:22
Size
1 KB
Mime Type
text/x-c++
Expires
Wed, Jun 5, 23:22 (2 d)
Engine
blob
Format
Raw Data
Handle
18075618
Attached To
rTAMAAS tamaas
unified_allocator.hh
View Options
/*
* SPDX-License-Indentifier: AGPL-3.0-or-later
*
* Copyright (©) 2016-2022 EPFL (École Polytechnique Fédérale de Lausanne),
* Laboratory (LSMS - Laboratoire de Simulation en Mécanique des Solides)
* Copyright (©) 2020-2022 Lucas Frérot
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published
* by the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program 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 Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*
*/
/* -------------------------------------------------------------------------- */
#ifndef UNIFIED_ALLOCATOR_HH
#define UNIFIED_ALLOCATOR_HH
/* -------------------------------------------------------------------------- */
#include "span.hh"
#include <cuda_runtime_api.h>
#include <memory>
/* -------------------------------------------------------------------------- */
namespace
tamaas
{
/// Class allocating [unified
/// memory](http://docs.nvidia.com/cuda/cuda-runtime-api/group__CUDART__MEMORY.html#group__CUDART__MEMORY_1gd228014f19cc0975ebe3e0dd2af6dd1b)
template
<
typename
T
>
struct
UnifiedAllocator
{
/// Allocate memory
static
span
<
T
>
allocate
(
typename
span
<
T
>::
size_type
n
)
noexcept
{
T
*
p
=
nullptr
;
cudaMallocManaged
(
&
p
,
n
*
sizeof
(
T
));
return
p
;
}
/// Free memory
static
void
deallocate
(
span
<
T
>
view
)
noexcept
{
cudaFree
(
p
);
}
};
}
// namespace tamaas
#endif
// UNIFIED_ALLOCATOR_HH
Event Timeline
Log In to Comment