Skip to content

Commit 6d5ba2e

Browse files
0x12CC0x12CC
authored
Merge branch 'main' into cooperative_kernel_functions
2 parents 607ea7b + f15234d commit 6d5ba2e

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

62 files changed

+6025
-662
lines changed

‎include/ur_api.h

Lines changed: 420 additions & 106 deletions
Large diffs are not rendered by default.

‎include/ur_ddi.h

Lines changed: 41 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1856,7 +1856,8 @@ typedef ur_result_t(UR_APICALL *ur_pfnCommandBufferAppendKernelLaunchExp_t)(
18561856
const size_t *,
18571857
uint32_t,
18581858
const ur_exp_command_buffer_sync_point_t *,
1859-
ur_exp_command_buffer_sync_point_t *);
1859+
ur_exp_command_buffer_sync_point_t *,
1860+
ur_exp_command_buffer_command_handle_t *);
18601861

18611862
///////////////////////////////////////////////////////////////////////////////
18621863
/// @brief Function-pointer for urCommandBufferAppendUSMMemcpyExp
@@ -2013,6 +2014,40 @@ typedef ur_result_t(UR_APICALL *ur_pfnCommandBufferEnqueueExp_t)(
20132014
const ur_event_handle_t *,
20142015
ur_event_handle_t *);
20152016

2017+
///////////////////////////////////////////////////////////////////////////////
2018+
/// @brief Function-pointer for urCommandBufferRetainCommandExp
2019+
typedef ur_result_t(UR_APICALL *ur_pfnCommandBufferRetainCommandExp_t)(
2020+
ur_exp_command_buffer_command_handle_t);
2021+
2022+
///////////////////////////////////////////////////////////////////////////////
2023+
/// @brief Function-pointer for urCommandBufferReleaseCommandExp
2024+
typedef ur_result_t(UR_APICALL *ur_pfnCommandBufferReleaseCommandExp_t)(
2025+
ur_exp_command_buffer_command_handle_t);
2026+
2027+
///////////////////////////////////////////////////////////////////////////////
2028+
/// @brief Function-pointer for urCommandBufferUpdateKernelLaunchExp
2029+
typedef ur_result_t(UR_APICALL *ur_pfnCommandBufferUpdateKernelLaunchExp_t)(
2030+
ur_exp_command_buffer_command_handle_t,
2031+
const ur_exp_command_buffer_update_kernel_launch_desc_t *);
2032+
2033+
///////////////////////////////////////////////////////////////////////////////
2034+
/// @brief Function-pointer for urCommandBufferGetInfoExp
2035+
typedef ur_result_t(UR_APICALL *ur_pfnCommandBufferGetInfoExp_t)(
2036+
ur_exp_command_buffer_handle_t,
2037+
ur_exp_command_buffer_info_t,
2038+
size_t,
2039+
void *,
2040+
size_t *);
2041+
2042+
///////////////////////////////////////////////////////////////////////////////
2043+
/// @brief Function-pointer for urCommandBufferCommandGetInfoExp
2044+
typedef ur_result_t(UR_APICALL *ur_pfnCommandBufferCommandGetInfoExp_t)(
2045+
ur_exp_command_buffer_command_handle_t,
2046+
ur_exp_command_buffer_command_info_t,
2047+
size_t,
2048+
void *,
2049+
size_t *);
2050+
20162051
///////////////////////////////////////////////////////////////////////////////
20172052
/// @brief Table of CommandBufferExp functions pointers
20182053
typedef struct ur_command_buffer_exp_dditable_t {
@@ -2033,6 +2068,11 @@ typedef struct ur_command_buffer_exp_dditable_t {
20332068
ur_pfnCommandBufferAppendUSMPrefetchExp_t pfnAppendUSMPrefetchExp;
20342069
ur_pfnCommandBufferAppendUSMAdviseExp_t pfnAppendUSMAdviseExp;
20352070
ur_pfnCommandBufferEnqueueExp_t pfnEnqueueExp;
2071+
ur_pfnCommandBufferRetainCommandExp_t pfnRetainCommandExp;
2072+
ur_pfnCommandBufferReleaseCommandExp_t pfnReleaseCommandExp;
2073+
ur_pfnCommandBufferUpdateKernelLaunchExp_t pfnUpdateKernelLaunchExp;
2074+
ur_pfnCommandBufferGetInfoExp_t pfnGetInfoExp;
2075+
ur_pfnCommandBufferCommandGetInfoExp_t pfnCommandGetInfoExp;
20362076
} ur_command_buffer_exp_dditable_t;
20372077

20382078
///////////////////////////////////////////////////////////////////////////////

‎include/ur_print.h

Lines changed: 96 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -930,6 +930,22 @@ UR_APIEXPORT ur_result_t UR_APICALL urPrintExpInteropMemDesc(const struct ur_exp
930930
/// - `buff_size < out_size`
931931
UR_APIEXPORT ur_result_t UR_APICALL urPrintExpInteropSemaphoreDesc(const struct ur_exp_interop_semaphore_desc_t params, char *buffer, const size_t buff_size, size_t *out_size);
932932

933+
///////////////////////////////////////////////////////////////////////////////
934+
/// @brief Print ur_exp_command_buffer_info_t enum
935+
/// @returns
936+
/// - ::UR_RESULT_SUCCESS
937+
/// - ::UR_RESULT_ERROR_INVALID_SIZE
938+
/// - `buff_size < out_size`
939+
UR_APIEXPORT ur_result_t UR_APICALL urPrintExpCommandBufferInfo(enum ur_exp_command_buffer_info_t value, char *buffer, const size_t buff_size, size_t *out_size);
940+
941+
///////////////////////////////////////////////////////////////////////////////
942+
/// @brief Print ur_exp_command_buffer_command_info_t enum
943+
/// @returns
944+
/// - ::UR_RESULT_SUCCESS
945+
/// - ::UR_RESULT_ERROR_INVALID_SIZE
946+
/// - `buff_size < out_size`
947+
UR_APIEXPORT ur_result_t UR_APICALL urPrintExpCommandBufferCommandInfo(enum ur_exp_command_buffer_command_info_t value, char *buffer, const size_t buff_size, size_t *out_size);
948+
933949
///////////////////////////////////////////////////////////////////////////////
934950
/// @brief Print ur_exp_command_buffer_desc_t struct
935951
/// @returns
@@ -938,6 +954,46 @@ UR_APIEXPORT ur_result_t UR_APICALL urPrintExpInteropSemaphoreDesc(const struct
938954
/// - `buff_size < out_size`
939955
UR_APIEXPORT ur_result_t UR_APICALL urPrintExpCommandBufferDesc(const struct ur_exp_command_buffer_desc_t params, char *buffer, const size_t buff_size, size_t *out_size);
940956

957+
///////////////////////////////////////////////////////////////////////////////
958+
/// @brief Print ur_exp_command_buffer_update_memobj_arg_desc_t struct
959+
/// @returns
960+
/// - ::UR_RESULT_SUCCESS
961+
/// - ::UR_RESULT_ERROR_INVALID_SIZE
962+
/// - `buff_size < out_size`
963+
UR_APIEXPORT ur_result_t UR_APICALL urPrintExpCommandBufferUpdateMemobjArgDesc(const struct ur_exp_command_buffer_update_memobj_arg_desc_t params, char *buffer, const size_t buff_size, size_t *out_size);
964+
965+
///////////////////////////////////////////////////////////////////////////////
966+
/// @brief Print ur_exp_command_buffer_update_pointer_arg_desc_t struct
967+
/// @returns
968+
/// - ::UR_RESULT_SUCCESS
969+
/// - ::UR_RESULT_ERROR_INVALID_SIZE
970+
/// - `buff_size < out_size`
971+
UR_APIEXPORT ur_result_t UR_APICALL urPrintExpCommandBufferUpdatePointerArgDesc(const struct ur_exp_command_buffer_update_pointer_arg_desc_t params, char *buffer, const size_t buff_size, size_t *out_size);
972+
973+
///////////////////////////////////////////////////////////////////////////////
974+
/// @brief Print ur_exp_command_buffer_update_value_arg_desc_t struct
975+
/// @returns
976+
/// - ::UR_RESULT_SUCCESS
977+
/// - ::UR_RESULT_ERROR_INVALID_SIZE
978+
/// - `buff_size < out_size`
979+
UR_APIEXPORT ur_result_t UR_APICALL urPrintExpCommandBufferUpdateValueArgDesc(const struct ur_exp_command_buffer_update_value_arg_desc_t params, char *buffer, const size_t buff_size, size_t *out_size);
980+
981+
///////////////////////////////////////////////////////////////////////////////
982+
/// @brief Print ur_exp_command_buffer_update_exec_info_desc_t struct
983+
/// @returns
984+
/// - ::UR_RESULT_SUCCESS
985+
/// - ::UR_RESULT_ERROR_INVALID_SIZE
986+
/// - `buff_size < out_size`
987+
UR_APIEXPORT ur_result_t UR_APICALL urPrintExpCommandBufferUpdateExecInfoDesc(const struct ur_exp_command_buffer_update_exec_info_desc_t params, char *buffer, const size_t buff_size, size_t *out_size);
988+
989+
///////////////////////////////////////////////////////////////////////////////
990+
/// @brief Print ur_exp_command_buffer_update_kernel_launch_desc_t struct
991+
/// @returns
992+
/// - ::UR_RESULT_SUCCESS
993+
/// - ::UR_RESULT_ERROR_INVALID_SIZE
994+
/// - `buff_size < out_size`
995+
UR_APIEXPORT ur_result_t UR_APICALL urPrintExpCommandBufferUpdateKernelLaunchDesc(const struct ur_exp_command_buffer_update_kernel_launch_desc_t params, char *buffer, const size_t buff_size, size_t *out_size);
996+
941997
///////////////////////////////////////////////////////////////////////////////
942998
/// @brief Print ur_exp_peer_info_t enum
943999
/// @returns
@@ -2250,6 +2306,46 @@ UR_APIEXPORT ur_result_t UR_APICALL urPrintCommandBufferAppendUsmAdviseExpParams
22502306
/// - `buff_size < out_size`
22512307
UR_APIEXPORT ur_result_t UR_APICALL urPrintCommandBufferEnqueueExpParams(const struct ur_command_buffer_enqueue_exp_params_t *params, char *buffer, const size_t buff_size, size_t *out_size);
22522308

2309+
///////////////////////////////////////////////////////////////////////////////
2310+
/// @brief Print ur_command_buffer_retain_command_exp_params_t struct
2311+
/// @returns
2312+
/// - ::UR_RESULT_SUCCESS
2313+
/// - ::UR_RESULT_ERROR_INVALID_SIZE
2314+
/// - `buff_size < out_size`
2315+
UR_APIEXPORT ur_result_t UR_APICALL urPrintCommandBufferRetainCommandExpParams(const struct ur_command_buffer_retain_command_exp_params_t *params, char *buffer, const size_t buff_size, size_t *out_size);
2316+
2317+
///////////////////////////////////////////////////////////////////////////////
2318+
/// @brief Print ur_command_buffer_release_command_exp_params_t struct
2319+
/// @returns
2320+
/// - ::UR_RESULT_SUCCESS
2321+
/// - ::UR_RESULT_ERROR_INVALID_SIZE
2322+
/// - `buff_size < out_size`
2323+
UR_APIEXPORT ur_result_t UR_APICALL urPrintCommandBufferReleaseCommandExpParams(const struct ur_command_buffer_release_command_exp_params_t *params, char *buffer, const size_t buff_size, size_t *out_size);
2324+
2325+
///////////////////////////////////////////////////////////////////////////////
2326+
/// @brief Print ur_command_buffer_update_kernel_launch_exp_params_t struct
2327+
/// @returns
2328+
/// - ::UR_RESULT_SUCCESS
2329+
/// - ::UR_RESULT_ERROR_INVALID_SIZE
2330+
/// - `buff_size < out_size`
2331+
UR_APIEXPORT ur_result_t UR_APICALL urPrintCommandBufferUpdateKernelLaunchExpParams(const struct ur_command_buffer_update_kernel_launch_exp_params_t *params, char *buffer, const size_t buff_size, size_t *out_size);
2332+
2333+
///////////////////////////////////////////////////////////////////////////////
2334+
/// @brief Print ur_command_buffer_get_info_exp_params_t struct
2335+
/// @returns
2336+
/// - ::UR_RESULT_SUCCESS
2337+
/// - ::UR_RESULT_ERROR_INVALID_SIZE
2338+
/// - `buff_size < out_size`
2339+
UR_APIEXPORT ur_result_t UR_APICALL urPrintCommandBufferGetInfoExpParams(const struct ur_command_buffer_get_info_exp_params_t *params, char *buffer, const size_t buff_size, size_t *out_size);
2340+
2341+
///////////////////////////////////////////////////////////////////////////////
2342+
/// @brief Print ur_command_buffer_command_get_info_exp_params_t struct
2343+
/// @returns
2344+
/// - ::UR_RESULT_SUCCESS
2345+
/// - ::UR_RESULT_ERROR_INVALID_SIZE
2346+
/// - `buff_size < out_size`
2347+
UR_APIEXPORT ur_result_t UR_APICALL urPrintCommandBufferCommandGetInfoExpParams(const struct ur_command_buffer_command_get_info_exp_params_t *params, char *buffer, const size_t buff_size, size_t *out_size);
2348+
22532349
///////////////////////////////////////////////////////////////////////////////
22542350
/// @brief Print ur_usm_p2p_enable_peer_access_exp_params_t struct
22552351
/// @returns

0 commit comments

Comments
 (0)