diff --git a/include/MemoryProcessing.cuh b/include/MemoryProcessing.cuh index d3fb882f216fbce20824969184d84d39a06e4b38..965117a51a37f9ab5f9f9f74f849150a61d4936e 100644 --- a/include/MemoryProcessing.cuh +++ b/include/MemoryProcessing.cuh @@ -4,6 +4,9 @@ namespace memproc { + template <MemType memtype> + bool realloc(void *&array, const size_t new_size); + template <MemType memtype> bool realloc(void *&array, size_t &allocated_size, const size_t new_size); diff --git a/include/MemoryProcessing.h b/include/MemoryProcessing.h index 409d2921c54cbbcdda0a5c8fc276906c8472b37e..b3db1fece0388f6a471dfbd833fed4e62bd7be8b 100644 --- a/include/MemoryProcessing.h +++ b/include/MemoryProcessing.h @@ -4,6 +4,9 @@ namespace memproc { + template <MemType memtype> + bool alloc(void *&array, const size_t new_size); + template <MemType memtype> bool realloc(void *&array, size_t &allocated_size, const size_t new_size); diff --git a/src/MemoryProcessing.cpp b/src/MemoryProcessing.cpp index 8705f200dacb72ef3e4e5d663036091849005795..ecaa177ca07bdaf316d9c37f62dcafc4866273e0 100644 --- a/src/MemoryProcessing.cpp +++ b/src/MemoryProcessing.cpp @@ -21,6 +21,16 @@ bool memproc::dealloc<MemType::CPU>(void *&array) return true; } +template <> +bool memproc::alloc<MemType::CPU>(void *&array, const size_t new_size) +{ + array = malloc(new_size); + memset(array, 0, new_size); + + return true; +} + + template <> bool memproc::realloc<MemType::CPU>(void *&array, size_t &allocated_size, const size_t new_size) { diff --git a/src/MemoryProcessing.cu b/src/MemoryProcessing.cu index 5447a3aec52a4d4ca2c2439a9fe1d321a141ae1a..9ce6d30cd2e7780c1df278dfc569ab15d546e411 100644 --- a/src/MemoryProcessing.cu +++ b/src/MemoryProcessing.cu @@ -21,6 +21,15 @@ bool memproc::dealloc<MemType::GPU>(void *&array) return true; } +template <> +bool memproc::alloc<MemType::GPU>(void *&array, const size_t new_size) +{ + cudaMalloc ( (void **)&array, new_size); + cudaMemset(array, 0, new_size); + + return true; +} + template <> bool memproc::realloc<MemType::GPU>(void *&array, size_t &allocated_size, const size_t new_size) {