Skip to content

Commit 7d8f2a0

Browse files
densmirnAlexander-Makaryev
densmirn
and
Alexander-Makaryev
authored
Extend backend functions interfaces with queue and events (#1122)
* Extend backend functions interfaces with queue and events Co-authored-by: Alexander-Makaryev <alexander.makaryev@gmail.com>
1 parent 812f004 commit 7d8f2a0

24 files changed

+10475
-2527
lines changed

‎dpnp/backend/include/dpnp_gen_1arg_1type_tbl.hpp

Lines changed: 28 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -47,17 +47,34 @@
4747
/** */ \
4848
/** Function "__name__" executes operator "__operation1__" over each element of the array */ \
4949
/** */ \
50-
/** @param[out] result_out Output array. */ \
51-
/** @param[in] result_size Output array size. */ \
52-
/** @param[in] result_ndim Number of output array dimensions. */ \
53-
/** @param[in] result_shape Output array shape. */ \
54-
/** @param[in] result_strides Output array strides. */ \
55-
/** @param[in] input1_in Input array 1. */ \
56-
/** @param[in] input1_size Input array 1 size. */ \
57-
/** @param[in] input1_ndim Number of input array 1 dimensions. */ \
58-
/** @param[in] input1_shape Input array 1 shape. */ \
59-
/** @param[in] input1_strides Input array 1 strides. */ \
60-
/** @param[in] where Where condition. */ \
50+
/** @param[in] q_ref Reference to SYCL queue. */ \
51+
/** @param[out] result_out Output array. */ \
52+
/** @param[in] result_size Output array size. */ \
53+
/** @param[in] result_ndim Number of output array dimensions. */ \
54+
/** @param[in] result_shape Output array shape. */ \
55+
/** @param[in] result_strides Output array strides. */ \
56+
/** @param[in] input1_in Input array 1. */ \
57+
/** @param[in] input1_size Input array 1 size. */ \
58+
/** @param[in] input1_ndim Number of input array 1 dimensions. */ \
59+
/** @param[in] input1_shape Input array 1 shape. */ \
60+
/** @param[in] input1_strides Input array 1 strides. */ \
61+
/** @param[in] where Where condition. */ \
62+
/** @param[in] dep_event_vec_ref Reference to vector of SYCL events. */ \
63+
template <typename _DataType> \
64+
DPCTLSyclEventRef __name__(DPCTLSyclQueueRef q_ref, \
65+
void* result_out, \
66+
const size_t result_size, \
67+
const size_t result_ndim, \
68+
const shape_elem_type* result_shape, \
69+
const shape_elem_type* result_strides, \
70+
const void* input1_in, \
71+
const size_t input1_size, \
72+
const size_t input1_ndim, \
73+
const shape_elem_type* input1_shape, \
74+
const shape_elem_type* input1_strides, \
75+
const size_t* where, \
76+
const DPCTLEventVectorRef dep_event_vec_ref); \
77+
\
6178
template <typename _DataType> \
6279
void __name__(void* result_out, \
6380
const size_t result_size, \

‎dpnp/backend/include/dpnp_gen_1arg_2type_tbl.hpp

Lines changed: 28 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -47,17 +47,34 @@
4747
/** */ \
4848
/** Function "__name__" executes operator "__operation1__" over each element of the array */ \
4949
/** */ \
50-
/** @param[out] result_out Output array. */ \
51-
/** @param[in] result_size Output array size. */ \
52-
/** @param[in] result_ndim Number of output array dimensions. */ \
53-
/** @param[in] result_shape Output array shape. */ \
54-
/** @param[in] result_strides Output array strides. */ \
55-
/** @param[in] input1_in Input array 1. */ \
56-
/** @param[in] input1_size Input array 1 size. */ \
57-
/** @param[in] input1_ndim Number of input array 1 dimensions. */ \
58-
/** @param[in] input1_shape Input array 1 shape. */ \
59-
/** @param[in] input1_strides Input array 1 strides. */ \
60-
/** @param[in] where Where condition. */ \
50+
/** @param[in] q_ref Reference to SYCL queue. */ \
51+
/** @param[out] result_out Output array. */ \
52+
/** @param[in] result_size Output array size. */ \
53+
/** @param[in] result_ndim Number of output array dimensions. */ \
54+
/** @param[in] result_shape Output array shape. */ \
55+
/** @param[in] result_strides Output array strides. */ \
56+
/** @param[in] input1_in Input array 1. */ \
57+
/** @param[in] input1_size Input array 1 size. */ \
58+
/** @param[in] input1_ndim Number of input array 1 dimensions. */ \
59+
/** @param[in] input1_shape Input array 1 shape. */ \
60+
/** @param[in] input1_strides Input array 1 strides. */ \
61+
/** @param[in] where Where condition. */ \
62+
/** @param[in] dep_event_vec_ref Reference to vector of SYCL events. */ \
63+
template <typename _DataType_input, typename _DataType_output> \
64+
DPCTLSyclEventRef __name__(DPCTLSyclQueueRef q_ref, \
65+
void* result_out, \
66+
const size_t result_size, \
67+
const size_t result_ndim, \
68+
const shape_elem_type* result_shape, \
69+
const shape_elem_type* result_strides, \
70+
const void* input1_in, \
71+
const size_t input1_size, \
72+
const size_t input1_ndim, \
73+
const shape_elem_type* input1_shape, \
74+
const shape_elem_type* input1_strides, \
75+
const size_t* where, \
76+
const DPCTLEventVectorRef dep_event_vec_ref); \
77+
\
6178
template <typename _DataType_input, typename _DataType_output> \
6279
void __name__(void* result_out, \
6380
const size_t result_size, \

‎dpnp/backend/include/dpnp_gen_2arg_1type_tbl.hpp

Lines changed: 38 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -46,22 +46,44 @@
4646
/** */ \
4747
/** Function "__name__" executes operator "__operation__" over corresponding elements of input arrays */ \
4848
/** */ \
49-
/** @param[out] result_out Output array. */ \
50-
/** @param[in] result_size Output array size. */ \
51-
/** @param[in] result_ndim Number of output array dimensions. */ \
52-
/** @param[in] result_shape Output array shape. */ \
53-
/** @param[in] result_strides Output array strides. */ \
54-
/** @param[in] input1_in Input array 1. */ \
55-
/** @param[in] input1_size Input array 1 size. */ \
56-
/** @param[in] input1_ndim Number of input array 1 dimensions. */ \
57-
/** @param[in] input1_shape Input array 1 shape. */ \
58-
/** @param[in] input1_strides Input array 1 strides. */ \
59-
/** @param[in] input2_in Input array 2. */ \
60-
/** @param[in] input2_size Input array 2 size. */ \
61-
/** @param[in] input2_ndim Number of input array 2 dimensions. */ \
62-
/** @param[in] input2_shape Input array 2 shape. */ \
63-
/** @param[in] input2_strides Input array 2 strides. */ \
64-
/** @param[in] where Where condition. */ \
49+
/** @param[in] q_ref Reference to SYCL queue. */ \
50+
/** @param[out] result_out Output array. */ \
51+
/** @param[in] result_size Output array size. */ \
52+
/** @param[in] result_ndim Number of output array dimensions. */ \
53+
/** @param[in] result_shape Output array shape. */ \
54+
/** @param[in] result_strides Output array strides. */ \
55+
/** @param[in] input1_in Input array 1. */ \
56+
/** @param[in] input1_size Input array 1 size. */ \
57+
/** @param[in] input1_ndim Number of input array 1 dimensions. */ \
58+
/** @param[in] input1_shape Input array 1 shape. */ \
59+
/** @param[in] input1_strides Input array 1 strides. */ \
60+
/** @param[in] input2_in Input array 2. */ \
61+
/** @param[in] input2_size Input array 2 size. */ \
62+
/** @param[in] input2_ndim Number of input array 2 dimensions. */ \
63+
/** @param[in] input2_shape Input array 2 shape. */ \
64+
/** @param[in] input2_strides Input array 2 strides. */ \
65+
/** @param[in] where Where condition. */ \
66+
/** @param[in] dep_event_vec_ref Reference to vector of SYCL events. */ \
67+
template <typename _DataType> \
68+
DPCTLSyclEventRef __name__(DPCTLSyclQueueRef q_ref, \
69+
void* result_out, \
70+
const size_t result_size, \
71+
const size_t result_ndim, \
72+
const shape_elem_type* result_shape, \
73+
const shape_elem_type* result_strides, \
74+
const void* input1_in, \
75+
const size_t input1_size, \
76+
const size_t input1_ndim, \
77+
const shape_elem_type* input1_shape, \
78+
const shape_elem_type* input1_strides, \
79+
const void* input2_in, \
80+
const size_t input2_size, \
81+
const size_t input2_ndim, \
82+
const shape_elem_type* input2_shape, \
83+
const shape_elem_type* input2_strides, \
84+
const size_t* where, \
85+
const DPCTLEventVectorRef dep_event_vec_ref); \
86+
\
6587
template <typename _DataType> \
6688
void __name__(void* result_out, \
6789
const size_t result_size, \

0 commit comments

Comments
 (0)