Skip to content

Commit 8e54ac2

Browse files
committed
STRIDES minor fixes part 3
1 parent 0377384 commit 8e54ac2

File tree

2 files changed

+35
-8
lines changed

2 files changed

+35
-8
lines changed

dpnp/backend/include/dpnp_gen_2arg_3type_tbl.hpp

Lines changed: 32 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -47,12 +47,39 @@
4747
/** */ \
4848
/** Function "__name__" executes operator "__operation1__" over corresponding elements of input arrays */ \
4949
/** */ \
50-
/** @param[in] array1 Input array 1. */ \
51-
/** @param[in] array2 Input array 2. */ \
52-
/** @param[out] result1 Output array. */ \
53-
/** @param[in] size Number of elements in the output array. */ \
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. */ \
5466
template <typename _DataType_input1, typename _DataType_input2, typename _DataType_output> \
55-
void __name__(void* array1, void* array2, void* result1, size_t size);
67+
void __name__(void* result_out, \
68+
const size_t result_size, \
69+
const size_t result_ndim, \
70+
const size_t* result_shape, \
71+
const size_t* result_strides, \
72+
const void* input1_in, \
73+
const size_t input1_size, \
74+
const size_t input1_ndim, \
75+
const size_t* input1_shape, \
76+
const size_t* input1_strides, \
77+
const void* input2_in, \
78+
const size_t input2_size, \
79+
const size_t input2_ndim, \
80+
const size_t* input2_shape, \
81+
const size_t* input2_strides, \
82+
const size_t* where)
5683

5784
#endif
5885

dpnp/dpnp_algo/dpnp_algo.pxd

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -239,9 +239,9 @@ ctypedef void(*fptr_1in_1out_t)(void *, void * , size_t)
239239
ctypedef void(*fptr_2in_1out_t)(void * , const void * , const size_t, const long * , const size_t,
240240
const void *, const size_t, const long * , const size_t, const long * )
241241
ctypedef void(*fptr_2in_1out_strides_t)(void *, const size_t, const size_t, const long * , const long * ,
242-
void *, const size_t, const size_t, const long * , const long * ,
243-
void *, const size_t, const size_t, const long * , const long * ,
244-
const long * )
242+
void *, const size_t, const size_t, const long * , const long * ,
243+
void *, const size_t, const size_t, const long * , const long * ,
244+
const long * )
245245
ctypedef void(*fptr_blas_gemm_2in_1out_t)(void *, void * , void * , size_t, size_t, size_t)
246246
ctypedef void(*dpnp_reduction_c_t)(void *, const void * , const size_t*, const size_t, const long*, const size_t, const void * , const long*)
247247

0 commit comments

Comments
 (0)