Shuenhoy/vtk-m
1
0
Fork 0
mirror of https://gitlab.kitware.com/vtk/vtk-m synced 2024-10-08 11:29:02 +00:00
Code Issues 2 Actions Packages Projects Releases Wiki Activity

vtkm/Swap works with hip

Browse Source
This commit is contained in:
Robert Maynard 2020-10-22 16:48:30 +00:00
parent 093d25f342
commit ff381bf8b4
1 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
vtkm/Swap.h
View File

@ -13,7 +13,7 @@
#include <vtkm/internal/ExportMacros.h>
#ifdef __CUDACC__
#ifdef VTKM_CUDA
#include <thrust/swap.h>
#else
#include <algorithm>
@ -23,13 +23,27 @@ namespace vtkm
{
/// Performs a swap operation. Safe to call from cuda code.
#ifdef __CUDACC__
#if defined(VTKM_CUDA)
template <typename T>
VTKM_EXEC_CONT void Swap(T& a, T& b)
{
using namespace thrust;
swap(a, b);
}
#elif defined(VTKM_HIP)
template <typename T>
__host__ void Swap(T& a, T& b)
{
using namespace std;
swap(a, b);
}
template <typename T>
__device__ void Swap(T& a, T& b)
{
T temp = a;
a = b;
b = temp;
}
#else
template <typename T>
VTKM_EXEC_CONT void Swap(T& a, T& b)