diff --git a/vtkm/Swap.h b/vtkm/Swap.h index b870248d0..233266ad2 100644 --- a/vtkm/Swap.h +++ b/vtkm/Swap.h @@ -23,7 +23,11 @@ #include +#ifdef __CUDACC__ +#include +#else #include +#endif namespace vtkm { @@ -33,9 +37,8 @@ namespace vtkm template VTKM_EXEC_CONT void Swap(T& a, T& b) { - T tmp = a; - a = b; - b = tmp; + using namespace thrust; + swap(a, b); } #else template