NextCloud/dlib
1
0
Fork 0
mirror of https://github.com/davisking/dlib.git synced 2024-11-01 10:14:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Added more overloads of memcpy() for cuda_data_ptr.

Browse Source
This commit is contained in:
Davis King 2018-06-11 22:38:50 -04:00
parent b23c718640
commit 4f7c01f469
1 changed files with 22 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
dlib/cuda/cuda_data_ptr.h
View File

@ -178,15 +178,31 @@ namespace dlib
}
friend void memcpy(
cuda_data_ptr& src,
const std::vector<T>& dest
cuda_data_ptr& dest,
const std::vector<T>& src
)
{
if (dest.size() != src.size())
dest = cuda_data_ptr<T>(src.size());
if (src.size() != dest.size())
src = cuda_data_ptr<T>(dest.size());
if (src.size() != 0)
memcpy(src.pdata, dest.data());
if (dest.size() != 0)
memcpy(dest.pdata, src.data());
}
friend void memcpy(
cuda_data_ptr& dest,
const float* src
)
{
memcpy(dest.pdata, src);
}
friend void memcpy(
float* dest,
const cuda_data_ptr& src
)
{
memcpy(dest, src.pdata);
}
private: