Shuenhoy/vtk-m
1
0
Fork 0
mirror of https://gitlab.kitware.com/vtk/vtk-m synced 2024-09-08 13:23:51 +00:00
Code Issues 2 Actions Packages Projects Releases Wiki Activity

Use thrust::swap for vtkm::Swap in CUDA device code.

Browse Source
This commit is contained in:
Allison Vacanti 2018-06-11 12:27:54 -04:00
parent 7158d56cdc
commit cc631650c5
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
vtkm/Swap.h
View File

@ -23,7 +23,11 @@
#include <vtkm/internal/ExportMacros.h> #include <vtkm/internal/ExportMacros.h>
#ifdef __CUDACC__
#include <thrust/swap.h>
#else
#include <algorithm> #include <algorithm>
#endif
namespace vtkm namespace vtkm
{ {
@ -33,9 +37,8 @@ namespace vtkm
template <typename T> template <typename T>
VTKM_EXEC_CONT void Swap(T& a, T& b) VTKM_EXEC_CONT void Swap(T& a, T& b)
{ {
T tmp = a; using namespace thrust;
a = b; swap(a, b);
b = tmp;
} }
#else #else
template <typename T> template <typename T>