110 lines
3.6 KiB
C
110 lines
3.6 KiB
C
//===--- 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
|