Page Menu
Home
c4science
Search
Configure Global Search
Log In
Files
F86885384
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
Wed, Oct 9, 05:07
Size
1 KB
Mime Type
text/x-c++
Expires
Fri, Oct 11, 05:07 (2 d)
Engine
blob
Format
Raw Data
Handle
21496961
Attached To
rTAMAAS tamaas
unified_allocator.hh
View Options
/**
* @file
* @section LICENSE
*
* Copyright (©) 2016-2021 EPFL (École Polytechnique Fédérale de Lausanne),
* Laboratory (LSMS - Laboratoire de Simulation en Mécanique des Solides)
*
* 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 <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
T
*
allocate
(
std
::
size_t
n
)
{
T
*
p
=
nullptr
;
cudaMallocManaged
(
&
p
,
n
*
sizeof
(
T
));
return
p
;
}
/// Free memory
static
void
deallocate
(
T
*
p
,
__attribute__
((
unused
))
std
::
size_t
n
)
{
cudaFree
(
p
);
}
};
}
// namespace tamaas
#endif
// UNIFIED_ALLOCATOR_HH
Event Timeline
Log In to Comment