Skip to content

Commit 0877c30

Browse files
Merge branch 'master' into gold/2021
2 parents b64472f + f185adb commit 0877c30

11 files changed

+562
-13
lines changed

dpnp/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,8 @@
4040
if "numpy" in sys.modules:
4141
warnings.warn("\nDPNP: Module NumPy found. Please load DPNP module before NumPy.\n")
4242

43-
from dpnp.dparray import dparray as ndarray
43+
# from dpnp.dparray import dparray as ndarray
44+
from dpnp.dpnp_array import dpnp_array as ndarray
4445
from dpnp.dpnp_iface import *
4546
from dpnp.dpnp_iface import __all__ as _iface__all__
4647
from dpnp.dpnp_iface_types import *

dpnp/backend/kernels/dpnp_krnl_random.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -974,7 +974,7 @@ void dpnp_rng_shuffle_c(
974974
return;
975975
}
976976

977-
DPNPC_ptr_adapter<char> result1_ptr(result, size, true, true);
977+
DPNPC_ptr_adapter<char> result1_ptr(result, size * itemsize, true, true);
978978
char* result1 = result1_ptr.get_ptr();
979979

980980
size_t uvec_size = high_dim_size - 1;

dpnp/backend/kernels/dpnp_krnl_statistics.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -760,7 +760,7 @@ void func_map_init_statistics(func_map_t& fmap)
760760
fmap[DPNPFuncName::DPNP_FN_CORRELATE][eft_DBL][eft_DBL] = {eft_DBL,
761761
(void*)dpnp_correlate_c<double, double, double>};
762762

763-
fmap[DPNPFuncName::DPNP_FN_COUNT_NONZERO][eft_BLN][eft_BLN] = {eft_BLN, (void*)dpnp_count_nonzero_c<bool, long>};
763+
fmap[DPNPFuncName::DPNP_FN_COUNT_NONZERO][eft_BLN][eft_BLN] = {eft_LNG, (void*)dpnp_count_nonzero_c<bool, long>};
764764
fmap[DPNPFuncName::DPNP_FN_COUNT_NONZERO][eft_INT][eft_INT] = {eft_LNG, (void*)dpnp_count_nonzero_c<int, long>};
765765
fmap[DPNPFuncName::DPNP_FN_COUNT_NONZERO][eft_LNG][eft_LNG] = {eft_LNG, (void*)dpnp_count_nonzero_c<long, long>};
766766
fmap[DPNPFuncName::DPNP_FN_COUNT_NONZERO][eft_FLT][eft_FLT] = {eft_LNG, (void*)dpnp_count_nonzero_c<float, long>};

dpnp/dpnp_algo/dpnp_algo_linearalgebra.pyx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -305,6 +305,6 @@ cpdef utils.dpnp_descriptor dpnp_outer(utils.dpnp_descriptor array1, utils.dpnp_
305305

306306
for idx1 in range(array1.size):
307307
for idx2 in range(array2.size):
308-
result.get_pyobj()[idx1 * array2.size + idx2] = array1.get_pyobj()[idx1] * array2.get_pyobj()[idx2]
308+
result.get_pyobj()[numpy.unravel_index(idx1 * array2.size + idx2, result.shape)] = array1.get_pyobj()[numpy.unravel_index(idx1, array1.shape)] * array2.get_pyobj()[numpy.unravel_index(idx2, array2.shape)]
309309

310310
return result

dpnp/dpnp_algo/dpnp_algo_manipulation.pyx

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,8 @@ cpdef utils.dpnp_descriptor dpnp_repeat(utils.dpnp_descriptor array1, repeats, a
148148

149149
cpdef utils.dpnp_descriptor dpnp_reshape(utils.dpnp_descriptor array1, newshape, order=None):
150150
# return dpnp.get_dpnp_descriptor(dpctl.tensor.usm_ndarray(newshape, dtype=numpy.dtype(array1.dtype).name, buffer=array1.get_pyobj()))
151-
return dpnp.get_dpnp_descriptor(dpctl.tensor.reshape(array1.get_pyobj(), newshape))
151+
# return dpnp.get_dpnp_descriptor(dpctl.tensor.reshape(array1.get_pyobj(), newshape))
152+
return dpnp.get_dpnp_descriptor(dpctl.tensor.reshape(array1.get_pyobj()._array_obj, newshape))
152153

153154

154155
cpdef utils.dpnp_descriptor dpnp_transpose(utils.dpnp_descriptor array1, axes=None):

dpnp/dpnp_algo/dpnp_algo_sorting.pyx

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,10 @@ ctypedef void(*fptr_dpnp_searchsorted_t)(void * , const void * , const void * ,
4747

4848

4949
cpdef utils.dpnp_descriptor dpnp_argsort(utils.dpnp_descriptor x1):
50-
return call_fptr_1in_1out(DPNP_FN_ARGSORT, x1, x1.shape)
50+
cdef shape_type_c result_shape = x1.shape
51+
if result_shape == ():
52+
result_shape = (1,)
53+
return call_fptr_1in_1out(DPNP_FN_ARGSORT, x1, result_shape)
5154

5255

5356
cpdef utils.dpnp_descriptor dpnp_partition(utils.dpnp_descriptor arr, int kth, axis=-1, kind='introselect', order=None):

0 commit comments

Comments
 (0)