1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
/* ************************************************************************
* Copyright (C) 2018-2023 Advanced Micro Devices, Inc. All rights reserved.
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell cop-
* ies of the Software, and to permit persons to whom the Software is furnished
* to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IM-
* PLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
* FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
* COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
* IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNE-
* CTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
*
* ************************************************************************ */
#pragma once
#include "pinned_memory_allocator.hpp"
#include <memory>
//!
//! @brief Pseudo-vector subclass which uses host pinned memory.
//! Delete this class once hipHostRegister has been proven equivalent and host_vector
//! is extended to manage pinning/unpinning.
//!
template <typename T>
struct host_pinned_vector : std::vector<T, pinned_memory_allocator<T>>
{
// Inherit constructors
using std::vector<T, pinned_memory_allocator<T>>::vector;
//!
//! @brief Constructor.
//!
host_pinned_vector(size_t n, int64_t inc)
: std::vector<T, pinned_memory_allocator<T>>(n * (inc ? inc : 1),
pinned_memory_allocator<T>())
, m_n(n)
, m_inc(inc ? inc : 1)
{
}
//!
//! @brief Decay into pointer wherever pointer is expected
//!
operator T*()
{
return this->data();
}
//!
//! @brief Decay into constant pointer wherever constant pointer is expected
//!
operator const T*() const
{
return this->data();
}
//!
//! @brief Transfer from a device vector.
//! @param that That device vector.
//! @return the hip error.
//!
hipError_t transfer_from(const device_vector<T>& that)
{
return hipMemcpy(
this->data(), (const T*)that, sizeof(T) * this->size(), hipMemcpyDeviceToHost);
}
//!
//! @brief Returns the length of the vector.
//!
size_t n() const
{
return m_n;
}
//!
//! @brief Returns the increment of the vector.
//!
int64_t inc() const
{
return m_inc;
}
//!
//! @brief Check if memory exists.
//! @return hipSuccess if memory exists, hipErrorOutOfMemory otherwise.
//!
hipError_t memcheck() const
{
return (nullptr != (const T*)this) ? hipSuccess : hipErrorOutOfMemory;
}
private:
size_t m_n{};
int64_t m_inc{};
};
|
