forked from OSchip/llvm-project
69 lines
2.1 KiB
C
69 lines
2.1 KiB
C
//===--- amdgpu/impl/atmi_runtime.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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
#ifndef INCLUDE_ATMI_RUNTIME_H_
|
|
#define INCLUDE_ATMI_RUNTIME_H_
|
|
|
|
#include "atmi.h"
|
|
#include "hsa_api.h"
|
|
|
|
extern "C" {
|
|
|
|
/** \defgroup module_functions ATMI Module
|
|
* @{
|
|
*/
|
|
|
|
/**
|
|
* @brief Register the ATMI code module from memory on to a specific place
|
|
* (device).
|
|
*
|
|
* @detail Currently, only GPU devices need explicit module registration because
|
|
* of their specific ISAs that require a separate compilation phase. On the
|
|
* other
|
|
* hand, CPU devices execute regular x86 functions that are compiled with the
|
|
* host program.
|
|
*
|
|
* @param[in] module_bytes A memory region that contains the GPU modules
|
|
* targeting ::AMDGCN platform types. Value cannot be NULL.
|
|
*
|
|
* @param[in] module_size Size of module region
|
|
*
|
|
* @param[in] place Denotes the execution place (device) on which the module
|
|
* should be registered and loaded.
|
|
*
|
|
* @param[in] on_deserialized_data Callback run on deserialized code object,
|
|
* before loading it
|
|
*
|
|
* @param[in] cb_state void* passed to on_deserialized_data callback
|
|
*
|
|
* @retval ::HSA_STATUS_SUCCESS The function has executed successfully.
|
|
*
|
|
* @retval ::HSA_STATUS_ERROR The function encountered errors.
|
|
*
|
|
*/
|
|
hsa_status_t atmi_module_register_from_memory_to_place(
|
|
void *module_bytes, size_t module_size, int DeviceId,
|
|
hsa_status_t (*on_deserialized_data)(void *data, size_t size,
|
|
void *cb_state),
|
|
void *cb_state);
|
|
|
|
/** @} */
|
|
|
|
hsa_status_t atmi_memcpy_h2d(hsa_signal_t signal, void *deviceDest,
|
|
const void *hostSrc, size_t size,
|
|
hsa_agent_t agent);
|
|
|
|
hsa_status_t atmi_memcpy_d2h(hsa_signal_t sig, void *hostDest,
|
|
const void *deviceSrc, size_t size,
|
|
hsa_agent_t agent);
|
|
|
|
/** @} */
|
|
|
|
}
|
|
|
|
#endif // INCLUDE_ATMI_RUNTIME_H_
|