Skip to content

Commit 85bd2b8

Browse files
committed
remove C kernel imports to cython
1 parent 23625a2 commit 85bd2b8

File tree

2 files changed

+6
-26
lines changed

2 files changed

+6
-26
lines changed

dpnp/backend.pxd

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -120,16 +120,6 @@ cdef extern from "backend/backend_iface.hpp":
120120
void custom_elemwise_tan_c[_DataType_input, _DataType_output](void * array1, void * result1, size_t size)
121121
void custom_elemwise_tanh_c[_DataType_input, _DataType_output](void * array1, void * result1, size_t size)
122122

123-
# Mathematical part
124-
void custom_elemwise_add_c[_DataType_input1, _DataType_input2, _DataType_output](void * array1, void * array2, void * result1, size_t size)
125-
void custom_elemwise_arctan2_c[_DataType_input1, _DataType_input2, _DataType_output](void * array1, void * array2, void * result1, size_t size)
126-
void custom_elemwise_divide_c[_DataType_input1, _DataType_input2, _DataType_output](void * array1, void * array2, void * result1, size_t size)
127-
void custom_elemwise_hypot_c[_DataType_input1, _DataType_input2, _DataType_output](void * array1, void * array2, void * result1, size_t size)
128-
void custom_elemwise_multiply_c[_DataType_input1, _DataType_input2, _DataType_output](void * array1, void * array2, void * result1, size_t size)
129-
void custom_elemwise_power_c[_DataType_input1, _DataType_input2, _DataType_output](void * array1, void * array2, void * result1, size_t size)
130-
void custom_elemwise_subtract_c[_DataType_input1, _DataType_input2, _DataType_output](void * array1, void * array2, void * result1, size_t size)
131-
void custom_sum_c[_DataType](void * array, void * result, size_t size)
132-
133123
# array manipulation routines
134124
void custom_elemwise_transpose_c[_DataType](void * array1_in, dparray_shape_type & input_shape, dparray_shape_type & result_shape, dparray_shape_type & permute_axes, void * result1, size_t size)
135125

dpnp/backend_statistics.pyx

Lines changed: 6 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ and the rest of the library
3636
import dpnp
3737
import numpy
3838
import dpnp
39-
from dpnp.dpnp_utils cimport checker_throw_type_error, normalize_axis
39+
from dpnp.dpnp_utils cimport *
4040
from dpnp.backend cimport *
4141

4242

@@ -50,23 +50,13 @@ __all__ += [
5050
]
5151

5252

53-
cpdef dpnp_average(dparray array):
54-
call_type = array.dtype
55-
return_type = numpy.float32 if call_type == numpy.float32 else numpy.float64
56-
cdef dparray result = dparray((1), dtype=call_type)
53+
cpdef dpnp_average(dparray x1):
54+
array_sum = dpnp_sum(x1)
5755

58-
if call_type == numpy.float64:
59-
custom_sum_c[double](array.get_data(), result.get_data(), array.size)
60-
elif call_type == numpy.float32:
61-
custom_sum_c[float](array.get_data(), result.get_data(), array.size)
62-
elif call_type == numpy.int64:
63-
custom_sum_c[long](array.get_data(), result.get_data(), array.size)
64-
elif call_type == numpy.int32:
65-
custom_sum_c[int](array.get_data(), result.get_data(), array.size)
66-
else:
67-
checker_throw_type_error("dpnp_average", call_type)
56+
""" Numpy interface inconsistency """
57+
return_type = numpy.float32 if (x1.dtype == numpy.float32) else numpy.float64
6858

69-
return return_type(result[0] / array.size)
59+
return (return_type(array_sum / x1.size))
7060

7161

7262
cpdef dparray dpnp_cov(dparray array1):

0 commit comments

Comments
 (0)