| //===--- amdgpu/dynamic_hsa/hsa_ext_amd.h ------------------------- C++ -*-===// |
| // |
| // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| // See https://llvm.org/LICENSE.txt for license information. |
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| // |
| //===----------------------------------------------------------------------===// |
| // |
| // The parts of the hsa api that are presently in use by the amdgpu plugin |
| // |
| //===----------------------------------------------------------------------===// |
| #ifndef HSA_RUNTIME_EXT_AMD_H_ |
| #define HSA_RUNTIME_EXT_AMD_H_ |
| |
| #include "hsa.h" |
| |
| #ifdef __cplusplus |
| extern "C" { |
| #endif |
| |
| typedef struct hsa_amd_memory_pool_s { |
| uint64_t handle; |
| } hsa_amd_memory_pool_t; |
| |
| typedef enum hsa_amd_memory_pool_global_flag_s { |
| HSA_AMD_MEMORY_POOL_GLOBAL_FLAG_KERNARG_INIT = 1, |
| HSA_AMD_MEMORY_POOL_GLOBAL_FLAG_FINE_GRAINED = 2, |
| HSA_AMD_MEMORY_POOL_GLOBAL_FLAG_COARSE_GRAINED = 4 |
| } hsa_amd_memory_pool_global_flag_t; |
| |
| typedef enum { |
| HSA_AMD_MEMORY_POOL_INFO_GLOBAL_FLAGS = 1, |
| HSA_AMD_MEMORY_POOL_INFO_SIZE = 2, |
| HSA_AMD_MEMORY_POOL_INFO_RUNTIME_ALLOC_ALLOWED = 5, |
| } hsa_amd_memory_pool_info_t; |
| |
| typedef enum { |
| HSA_AMD_AGENT_MEMORY_POOL_INFO_ACCESS = 0, |
| } hsa_amd_agent_memory_pool_info_t; |
| |
| typedef enum { |
| HSA_AMD_MEMORY_POOL_ACCESS_NEVER_ALLOWED = 0, |
| } hsa_amd_memory_pool_access_t; |
| |
| typedef enum hsa_amd_agent_info_s { |
| HSA_AMD_AGENT_INFO_COMPUTE_UNIT_COUNT = 0xA002, |
| } hsa_amd_agent_info_t; |
| |
| hsa_status_t hsa_amd_memory_pool_get_info(hsa_amd_memory_pool_t memory_pool, |
| hsa_amd_memory_pool_info_t attribute, |
| void *value); |
| |
| hsa_status_t hsa_amd_agent_iterate_memory_pools( |
| hsa_agent_t agent, |
| hsa_status_t (*callback)(hsa_amd_memory_pool_t memory_pool, void *data), |
| void *data); |
| |
| hsa_status_t hsa_amd_memory_pool_allocate(hsa_amd_memory_pool_t memory_pool, |
| size_t size, uint32_t flags, |
| void **ptr); |
| |
| hsa_status_t hsa_amd_memory_pool_free(void *ptr); |
| |
| hsa_status_t hsa_amd_memory_async_copy(void *dst, hsa_agent_t dst_agent, |
| const void *src, hsa_agent_t src_agent, |
| size_t size, uint32_t num_dep_signals, |
| const hsa_signal_t *dep_signals, |
| hsa_signal_t completion_signal); |
| |
| hsa_status_t hsa_amd_agent_memory_pool_get_info( |
| hsa_agent_t agent, hsa_amd_memory_pool_t memory_pool, |
| hsa_amd_agent_memory_pool_info_t attribute, void *value); |
| |
| hsa_status_t hsa_amd_agents_allow_access(uint32_t num_agents, |
| const hsa_agent_t *agents, |
| const uint32_t *flags, |
| const void *ptr); |
| |
| hsa_status_t hsa_amd_memory_fill(void *ptr, uint32_t value, size_t count); |
| |
| typedef enum hsa_amd_event_type_s { |
| HSA_AMD_GPU_MEMORY_FAULT_EVENT = 0, |
| } hsa_amd_event_type_t; |
| |
| typedef struct hsa_amd_gpu_memory_fault_info_s { |
| hsa_agent_t agent; |
| uint64_t virtual_address; |
| uint32_t fault_reason_mask; |
| } hsa_amd_gpu_memory_fault_info_t; |
| |
| typedef struct hsa_amd_event_s { |
| hsa_amd_event_type_t event_type; |
| union { |
| hsa_amd_gpu_memory_fault_info_t memory_fault; |
| }; |
| } hsa_amd_event_t; |
| |
| typedef hsa_status_t (*hsa_amd_system_event_callback_t)( |
| const hsa_amd_event_t *event, void *data); |
| |
| hsa_status_t |
| hsa_amd_register_system_event_handler(hsa_amd_system_event_callback_t callback, |
| void *data); |
| |
| #ifdef __cplusplus |
| } |
| #endif |
| |
| #endif |