31 lines
1.3 KiB
C++
31 lines
1.3 KiB
C++
#pragma once
|
|
|
|
#include <hip/hip_runtime.h>
|
|
#include <string>
|
|
|
|
const static std::string GPU_PREFIX = "hip:";
|
|
|
|
// hipify-perl warning: unsupported HIP identifier: cudaMemoryTypeUnregistered
|
|
#define cudaMemoryTypeUnregistered hipMemoryTypeUnregistered
|
|
|
|
// hipify-perl warning: unsupported HIP identifier: CU_MEM_HANDLE_TYPE_FABRIC
|
|
// Note: HIP does not currently support multi-node (fabric-based) transfers.
|
|
// Use POSIX file descriptor handle type for now, which only supports
|
|
// intra-node (same-machine) memory sharing between processes.
|
|
// TODO: Change to appropriate handle type when HIP adds multi-node support.
|
|
#define CU_MEM_HANDLE_TYPE_FABRIC hipMemHandleTypePosixFileDescriptor
|
|
|
|
// hipify-perl warning: unsupported HIP identifier:
|
|
// CU_DEVICE_ATTRIBUTE_GPU_DIRECT_RDMA_WITH_CUDA_VMM_SUPPORTED
|
|
#define CU_DEVICE_ATTRIBUTE_GPU_DIRECT_RDMA_WITH_CUDA_VMM_SUPPORTED \
|
|
hipDeviceAttributeVirtualMemoryManagementSupported
|
|
|
|
// hipify-perl warning: unsupported HIP identifier: CUmemFabricHandle
|
|
#define CUmemFabricHandle void*
|
|
|
|
// HIP mappings for nvlink_allocator compatibility
|
|
// Map CUDA symbols to closest HIP equivalents
|
|
#define CUDA_ERROR_NOT_PERMITTED hipErrorNotSupported
|
|
#define CU_DEVICE_ATTRIBUTE_HANDLE_TYPE_FABRIC_SUPPORTED \
|
|
hipDeviceAttributeVirtualMemoryManagementSupported
|