memory_type.h

 
#ifndef UCS_MEMORY_TYPE_H_
#define UCS_MEMORY_TYPE_H_
 
#include <ucs/sys/compiler_def.h>
 
BEGIN_C_DECLS
 
 
#define UCS_MEMORY_TYPES_CPU_ACCESSIBLE \
    (UCS_BIT(UCS_MEMORY_TYPE_HOST) | UCS_BIT(UCS_MEMORY_TYPE_ROCM_MANAGED) | \
     UCS_BIT(UCS_MEMORY_TYPE_ZE_HOST) | UCS_BIT(UCS_MEMORY_TYPE_ZE_MANAGED))
 
 
#define ucs_memory_type_for_each(_x) \
    for ((_x) = UCS_MEMORY_TYPE_HOST; (_x) < UCS_MEMORY_TYPE_LAST; (_x)++)
 
 
typedef enum ucs_memory_type {
    UCS_MEMORY_TYPE_HOST,          
    UCS_MEMORY_TYPE_CUDA,          
    UCS_MEMORY_TYPE_CUDA_MANAGED,  
    UCS_MEMORY_TYPE_ROCM,          
    UCS_MEMORY_TYPE_ROCM_MANAGED,  
    UCS_MEMORY_TYPE_RDMA,          
    UCS_MEMORY_TYPE_ZE_HOST,       
    UCS_MEMORY_TYPE_ZE_DEVICE,     
    UCS_MEMORY_TYPE_ZE_MANAGED,    
    UCS_MEMORY_TYPE_LAST,
    UCS_MEMORY_TYPE_UNKNOWN = UCS_MEMORY_TYPE_LAST
} ucs_memory_type_t;
 
 
extern const char *ucs_memory_type_names[];
 
extern const char *ucs_memory_type_descs[];
 
 
END_C_DECLS
 
#endif