Skip to content
Snippets Groups Projects
Commit 29181ce4 authored by Charles Villard's avatar Charles Villard Committed by Maxence Naud
Browse files

edit: TensorImpl: try to fix template ?

parent cf94ef7a
No related branches found
No related tags found
2 merge requests!75Update 0.5.1 -> 0.6.0,!40Fix template compilation and warnings
...@@ -21,20 +21,23 @@ ...@@ -21,20 +21,23 @@
namespace Aidge { namespace Aidge {
template <typename SRC_T, typename DST_T> template <typename SRC_T, typename DST_T>
typename std::enable_if<!std::is_same<half_float::half, DST_T>::value && !std::is_same<half_float::half, SRC_T>::value>::type std::enable_if_t<
thrust_copy(const SRC_T* /*srcData*/, DST_T* /*dstData*/, size_t /*size*/); !(std::is_same<half_float::half, DST_T>::value
|| std::is_same<half_float::half, SRC_T>::value)>
thrust_copy(const SRC_T* srcData, DST_T* dstData, size_t size);
template <typename SRC_T, typename DST_T = half_float::half> void
typename std::enable_if<!std::is_same<half_float::half, SRC_T>::value>::type thrust_copy(const half_float::half* srcData, half_float::half* dstData, size_t size);
thrust_copy(const SRC_T *srcData, half_float::half *dstData, size_t size);
template <typename SRC_T = half_float::half, typename DST_T> template <typename DST_T>
typename std::enable_if<!std::is_same<half_float::half, DST_T>::value>::type std::enable_if_t<!std::is_same<half_float::half, DST_T>::value>
thrust_copy(const half_float::half* srcData, DST_T* dstData, size_t size); thrust_copy(const half_float::half* srcData, DST_T* dstData, size_t size);
template <typename SRC_T = half_float::half, typename DST_T = half_float::half> template <typename SRC_T>
void thrust_copy(const half_float::half* srcData, half_float::half* dstData, size_t size); std::enable_if_t<!std::is_same<half_float::half, SRC_T>::value>
thrust_copy(const SRC_T* srcData, half_float::half* dstData, size_t size);
/** /**
* @brief Abstract class for the TensorImpl_cuda class template. * @brief Abstract class for the TensorImpl_cuda class template.
......
This diff is collapsed.
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment