4753a4e311
We introduced the implementation of supporting asynchronous routines with depend objects specified in Version 5.1 of the OpenMP Application Programming Interface. In brief, these routines omp_target_memcpy_async and omp_target_memcpy_rect_async perform asynchronous (nonblocking) memory copies between any combination of host and device pointers. The basic idea is to create the implicit tasks to carry the memory copy calls and handle dependencies specified by depend objects. The implicit tasks are executed via hidden helper thread in OpenMP runtime. Reviewed By: jdoerfert, tianshilei1992 Committed By: jplehr Differential Revision: https://reviews.llvm.org/D136103
158 lines
4.6 KiB
C
158 lines
4.6 KiB
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 _INTEROP_H_
|
|
#define _INTEROP_H_
|
|
|
|
#include "omptarget.h"
|
|
#include <assert.h>
|
|
|
|
#if defined(_WIN32)
|
|
#define __KAI_KMPC_CONVENTION __cdecl
|
|
#ifndef __KMP_IMP
|
|
#define __KMP_IMP __declspec(dllimport)
|
|
#endif
|
|
#else
|
|
#define __KAI_KMPC_CONVENTION
|
|
#ifndef __KMP_IMP
|
|
#define __KMP_IMP
|
|
#endif
|
|
#endif
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
/// TODO: Include the `omp.h` of the current build
|
|
/* OpenMP 5.1 interop */
|
|
typedef intptr_t omp_intptr_t;
|
|
|
|
/* 0..omp_get_num_interop_properties()-1 are reserved for implementation-defined
|
|
* properties */
|
|
typedef enum omp_interop_property {
|
|
omp_ipr_fr_id = -1,
|
|
omp_ipr_fr_name = -2,
|
|
omp_ipr_vendor = -3,
|
|
omp_ipr_vendor_name = -4,
|
|
omp_ipr_device_num = -5,
|
|
omp_ipr_platform = -6,
|
|
omp_ipr_device = -7,
|
|
omp_ipr_device_context = -8,
|
|
omp_ipr_targetsync = -9,
|
|
omp_ipr_first = -9
|
|
} omp_interop_property_t;
|
|
|
|
#define omp_interop_none 0
|
|
|
|
typedef enum omp_interop_rc {
|
|
omp_irc_no_value = 1,
|
|
omp_irc_success = 0,
|
|
omp_irc_empty = -1,
|
|
omp_irc_out_of_range = -2,
|
|
omp_irc_type_int = -3,
|
|
omp_irc_type_ptr = -4,
|
|
omp_irc_type_str = -5,
|
|
omp_irc_other = -6
|
|
} omp_interop_rc_t;
|
|
|
|
typedef enum omp_interop_fr {
|
|
omp_ifr_cuda = 1,
|
|
omp_ifr_cuda_driver = 2,
|
|
omp_ifr_opencl = 3,
|
|
omp_ifr_sycl = 4,
|
|
omp_ifr_hip = 5,
|
|
omp_ifr_level_zero = 6,
|
|
omp_ifr_last = 7
|
|
} omp_interop_fr_t;
|
|
|
|
typedef void *omp_interop_t;
|
|
|
|
/*!
|
|
* The `omp_get_num_interop_properties` routine retrieves the number of
|
|
* implementation-defined properties available for an `omp_interop_t` object.
|
|
*/
|
|
int __KAI_KMPC_CONVENTION omp_get_num_interop_properties(const omp_interop_t);
|
|
/*!
|
|
* The `omp_get_interop_int` routine retrieves an integer property from an
|
|
* `omp_interop_t` object.
|
|
*/
|
|
omp_intptr_t __KAI_KMPC_CONVENTION omp_get_interop_int(const omp_interop_t,
|
|
omp_interop_property_t,
|
|
int *);
|
|
/*!
|
|
* The `omp_get_interop_ptr` routine retrieves a pointer property from an
|
|
* `omp_interop_t` object.
|
|
*/
|
|
void *__KAI_KMPC_CONVENTION omp_get_interop_ptr(const omp_interop_t,
|
|
omp_interop_property_t, int *);
|
|
/*!
|
|
* The `omp_get_interop_str` routine retrieves a string property from an
|
|
* `omp_interop_t` object.
|
|
*/
|
|
const char *__KAI_KMPC_CONVENTION omp_get_interop_str(const omp_interop_t,
|
|
omp_interop_property_t,
|
|
int *);
|
|
/*!
|
|
* The `omp_get_interop_name` routine retrieves a property name from an
|
|
* `omp_interop_t` object.
|
|
*/
|
|
const char *__KAI_KMPC_CONVENTION omp_get_interop_name(const omp_interop_t,
|
|
omp_interop_property_t);
|
|
/*!
|
|
* The `omp_get_interop_type_desc` routine retrieves a description of the type
|
|
* of a property associated with an `omp_interop_t` object.
|
|
*/
|
|
const char *__KAI_KMPC_CONVENTION
|
|
omp_get_interop_type_desc(const omp_interop_t, omp_interop_property_t);
|
|
/*!
|
|
* The `omp_get_interop_rc_desc` routine retrieves a description of the return
|
|
* code associated with an `omp_interop_t` object.
|
|
*/
|
|
extern const char *__KAI_KMPC_CONVENTION
|
|
omp_get_interop_rc_desc(const omp_interop_t, omp_interop_rc_t);
|
|
|
|
typedef enum omp_interop_backend_type_t {
|
|
// reserve 0
|
|
omp_interop_backend_type_cuda_1 = 1,
|
|
} omp_interop_backend_type_t;
|
|
|
|
typedef enum kmp_interop_type_t {
|
|
kmp_interop_type_unknown = -1,
|
|
kmp_interop_type_platform,
|
|
kmp_interop_type_device,
|
|
kmp_interop_type_tasksync,
|
|
} kmp_interop_type_t;
|
|
|
|
typedef enum omp_foreign_runtime_ids {
|
|
cuda = 1,
|
|
cuda_driver = 2,
|
|
opencl = 3,
|
|
sycl = 4,
|
|
hip = 5,
|
|
level_zero = 6,
|
|
} omp_foreign_runtime_ids_t;
|
|
|
|
/// The interop value type, aka. the interop object.
|
|
typedef struct omp_interop_val_t {
|
|
/// Device and interop-type are determined at construction time and fix.
|
|
omp_interop_val_t(intptr_t device_id, kmp_interop_type_t interop_type)
|
|
: interop_type(interop_type), device_id(device_id) {}
|
|
const char *err_str = nullptr;
|
|
__tgt_async_info *async_info = nullptr;
|
|
__tgt_device_info device_info;
|
|
const kmp_interop_type_t interop_type;
|
|
const intptr_t device_id;
|
|
const omp_foreign_runtime_ids_t vendor_id = cuda;
|
|
const intptr_t backend_type_id = omp_interop_backend_type_cuda_1;
|
|
} omp_interop_val_t;
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
#endif
|