Skip to content
Snippets Groups Projects
Commit 565d53ec authored by 数学の武士's avatar 数学の武士
Browse files

.

parent 680664cb
No related branches found
No related tags found
No related merge requests found
...@@ -10,6 +10,9 @@ namespace memproc ...@@ -10,6 +10,9 @@ namespace memproc
template<MemType memtype> template<MemType memtype>
bool dealloc(void *&array, size_t &allocated_size); bool dealloc(void *&array, size_t &allocated_size);
template<MemType memtype>
bool dealloc(void *&array);
template <MemType dst_memtype, MemType src_memtype> template <MemType dst_memtype, MemType src_memtype>
bool memcopy(void *dst, const void* src, const size_t copy_elem_size); bool memcopy(void *dst, const void* src, const size_t copy_elem_size);
} }
\ No newline at end of file
...@@ -10,6 +10,9 @@ namespace memproc ...@@ -10,6 +10,9 @@ namespace memproc
template<MemType memtype> template<MemType memtype>
bool dealloc(void *&array, size_t &allocated_size); bool dealloc(void *&array, size_t &allocated_size);
template<MemType memtype>
bool dealloc(void *&array);
template <MemType dst_memtype, MemType src_memtype> template <MemType dst_memtype, MemType src_memtype>
bool memcopy(void *dst, const void* src, const size_t copy_elem_size); bool memcopy(void *dst, const void* src, const size_t copy_elem_size);
} }
...@@ -14,6 +14,13 @@ bool memproc::dealloc<MemType::CPU>(void *&array, size_t &allocated_size) ...@@ -14,6 +14,13 @@ bool memproc::dealloc<MemType::CPU>(void *&array, size_t &allocated_size)
return true; return true;
} }
template<>
bool memproc::dealloc<MemType::CPU>(void *&array)
{
free(array);
return true;
}
template <> template <>
bool memproc::realloc<MemType::CPU>(void *&array, size_t &allocated_size, const size_t new_size) bool memproc::realloc<MemType::CPU>(void *&array, size_t &allocated_size, const size_t new_size)
{ {
......
...@@ -14,6 +14,13 @@ bool memproc::dealloc<MemType::GPU>(void *&array, size_t &allocated_size) ...@@ -14,6 +14,13 @@ bool memproc::dealloc<MemType::GPU>(void *&array, size_t &allocated_size)
return true; return true;
} }
template<>
bool memproc::dealloc<MemType::GPU>(void *&array)
{
cudaFree(array);
return true;
}
template <> template <>
bool memproc::realloc<MemType::GPU>(void *&array, size_t &allocated_size, const size_t new_size) bool memproc::realloc<MemType::GPU>(void *&array, size_t &allocated_size, const size_t new_size)
{ {
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment