YAKL
YAKL_memory_spaces.h
Go to the documentation of this file.
1 
6 #pragma once
7 
8 __YAKL_NAMESPACE_WRAPPER_BEGIN__
9 namespace yakl {
10 
11  // Labels for memory spaces. Only memDevice and memHost are expected to be used explicitly by the user
13  int constexpr memDevice = 1;
15  int constexpr memHost = 2;
16  #if defined(YAKL_ARCH_CUDA) || defined(YAKL_ARCH_HIP) || defined(YAKL_ARCH_SYCL)
17  int constexpr memDefault = memDevice;
18  #else
19 
20  int constexpr memDefault = memHost;
21  #endif
22 
23 }
24 __YAKL_NAMESPACE_WRAPPER_END__
25 
26 
yakl::memDevice
constexpr int memDevice
Specifies a device memory address space for a yakl::Array object.
Definition: YAKL_memory_spaces.h:13
__YAKL_NAMESPACE_WRAPPER_END__
#define __YAKL_NAMESPACE_WRAPPER_END__
Definition: YAKL.h:20
__YAKL_NAMESPACE_WRAPPER_BEGIN__
#define __YAKL_NAMESPACE_WRAPPER_BEGIN__
Definition: YAKL.h:19
yakl::memDefault
constexpr int memDefault
If the user does not specify a memory space template parameter to yakl::Array, host is the default.
Definition: YAKL_memory_spaces.h:20
yakl
yakl::memHost
constexpr int memHost
Specifies a device memory address space for a yakl::Array object.
Definition: YAKL_memory_spaces.h:15
Generated by   doxygen 1.8.17