Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 8 additions & 0 deletions src/flcl-cxx.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,10 @@
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

#ifndef KOKKOS_IMPL_PUBLIC_INCLUDE
#define KOKKOS_IMPL_PUBLIC_INCLUDE
#endif

#ifndef FLCL_CXX_HPP
#define FLCL_CXX_HPP

Expand All @@ -58,6 +62,8 @@ typedef struct _flcl_nd_array_t {
namespace flcl {
#ifdef KOKKOS_ENABLE_CUDA
using HostMemorySpace = Kokkos::CudaUVMSpace;
#elif defined KOKKOS_ENABLE_HIP
using HostMemorySpace = Kokkos::Experimental::HIPManagedSpace;
#else
using HostMemorySpace = Kokkos::HostSpace;
#endif
Expand All @@ -75,6 +81,8 @@ namespace flcl {

#ifdef KOKKOS_ENABLE_CUDA
using HostMemorySpace = Kokkos::CudaUVMSpace;
#elif defined KOKKOS_ENABLE_HIP
using HostMemorySpace = Kokkos::Experimental::HIPManagedSpace;
#else
using HostMemorySpace = Kokkos::HostSpace;
#endif
Expand Down