Skip to content

Commit 84ac14b

Browse files
authored
Implement strides for 5 bitwise functions part 3 (#992)
* Implement strides for 5 bitwise functions
1 parent 3e54006 commit 84ac14b

File tree

7 files changed

+164
-41
lines changed

7 files changed

+164
-41
lines changed

dpnp/backend/include/dpnp_gen_2arg_1type_tbl.hpp

Lines changed: 37 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -46,20 +46,46 @@
4646
/** */ \
4747
/** Function "__name__" executes operator "__operation__" over corresponding elements of input arrays */ \
4848
/** */ \
49-
/** @param[out] result1 Output array. */ \
50-
/** @param[in] array1 Input array 1. */ \
51-
/** @param[in] size1 Number of elements in @ref array1 */ \
52-
/** @param[in] array2 Input array 2. */ \
53-
/** @param[in] size2 Number of elements in @ref array2 */ \
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. */ \
5465
template <typename _DataType> \
55-
void __name__(void* result1, const void* array1, const size_t size1, const void* array2, const size_t size2);
66+
void __name__(void* result_out, \
67+
const size_t result_size, \
68+
const size_t result_ndim, \
69+
const size_t* result_shape, \
70+
const size_t* result_strides, \
71+
const void* input1_in, \
72+
const size_t input1_size, \
73+
const size_t input1_ndim, \
74+
const size_t* input1_shape, \
75+
const size_t* input1_strides, \
76+
const void* input2_in, \
77+
const size_t input2_size, \
78+
const size_t input2_ndim, \
79+
const size_t* input2_shape, \
80+
const size_t* input2_strides, \
81+
const size_t* where);
5682

5783
#endif
5884

59-
MACRO_2ARG_1TYPE_OP(dpnp_bitwise_and_c, input_elem1& input_elem2)
60-
MACRO_2ARG_1TYPE_OP(dpnp_bitwise_or_c, input_elem1 | input_elem2)
61-
MACRO_2ARG_1TYPE_OP(dpnp_bitwise_xor_c, input_elem1 ^ input_elem2)
62-
MACRO_2ARG_1TYPE_OP(dpnp_left_shift_c, input_elem1 << input_elem2)
63-
MACRO_2ARG_1TYPE_OP(dpnp_right_shift_c, input_elem1 >> input_elem2)
85+
MACRO_2ARG_1TYPE_OP(dpnp_bitwise_and_c, input1_elem & input2_elem)
86+
MACRO_2ARG_1TYPE_OP(dpnp_bitwise_or_c, input1_elem | input2_elem)
87+
MACRO_2ARG_1TYPE_OP(dpnp_bitwise_xor_c, input1_elem ^ input2_elem)
88+
MACRO_2ARG_1TYPE_OP(dpnp_left_shift_c, input1_elem << input2_elem)
89+
MACRO_2ARG_1TYPE_OP(dpnp_right_shift_c, input1_elem >> input2_elem)
6490

6591
#undef MACRO_2ARG_1TYPE_OP

dpnp/backend/include/dpnp_iface.hpp

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -939,14 +939,20 @@ INP_DLLEXPORT void dpnp_invert_c(void* array1_in, void* result, size_t size);
939939
#define MACRO_2ARG_1TYPE_OP(__name__, __operation__) \
940940
template <typename _DataType> \
941941
INP_DLLEXPORT void __name__(void* result_out, \
942+
const size_t result_size, \
943+
const size_t result_ndim, \
944+
const size_t* result_shape, \
945+
const size_t* result_strides, \
942946
const void* input1_in, \
943947
const size_t input1_size, \
948+
const size_t input1_ndim, \
944949
const size_t* input1_shape, \
945-
const size_t input1_shape_ndim, \
950+
const size_t* input1_strides, \
946951
const void* input2_in, \
947952
const size_t input2_size, \
953+
const size_t input2_ndim, \
948954
const size_t* input2_shape, \
949-
const size_t input2_shape_ndim, \
955+
const size_t* input2_strides, \
950956
const size_t* where);
951957

952958
#include <dpnp_gen_2arg_1type_tbl.hpp>

0 commit comments

Comments
 (0)