DynMem: Replace hardcoded float in template class with template type

[hercules2020/kcf.git] / src / dynmem.hpp

diff --git a/src/dynmem.hpp b/src/dynmem.hpp
index 5a32bd6360148463caca6e34c6873bf2bea5d717..e2c625ca0805aac3046e922ec574f006731e02fd 100644 (file)
--- a/src/dynmem.hpp
+++ b/src/dynmem.hpp
@@ -27,7 +27,7 @@ template <typename T> class DynMem_ {
         CudaSafeCall(cudaHostAlloc(reinterpret_cast<void **>(&ptr_h), size, cudaHostAllocMapped));
         CudaSafeCall(cudaHostGetDevicePointer(reinterpret_cast<void **>(&ptr_d), reinterpret_cast<void *>(ptr_h), 0));
 #else
-        ptr_h = new float[size];
+        ptr_h = new T[size];
 #endif
     }
     DynMem_(DynMem_&& other) {