@@ -382,8 +382,9 @@ cdef class MT19937:
382
382
cdef fptr_dpnp_rng_normal_c_1out_t func
383
383
cdef c_dpctl.DPCTLSyclEventRef event_ref
384
384
385
+ result_shape = utils._object_to_tuple(size)
385
386
if scale == 0.0 :
386
- return utils.dpnp_descriptor(dpnp.full(size , loc, dtype = dtype))
387
+ return utils.dpnp_descriptor(dpnp.full(result_shape , loc, dtype = dtype))
387
388
388
389
# convert string type names (array.dtype) to C enum DPNPFuncType
389
390
param1_type = dpnp_dtype_to_DPNPFuncType(dtype)
@@ -392,7 +393,6 @@ cdef class MT19937:
392
393
kernel_data = get_dpnp_function_ptr(DPNP_FN_RNG_NORMAL_EXT, param1_type, param1_type)
393
394
394
395
# ceate result array with type given by FPTR data
395
- result_shape = utils._object_to_tuple(size)
396
396
result = utils.create_output_descriptor(result_shape,
397
397
kernel_data.return_type,
398
398
None ,
@@ -418,8 +418,9 @@ cdef class MT19937:
418
418
cdef fptr_dpnp_rng_uniform_c_1out_t func
419
419
cdef c_dpctl.DPCTLSyclEventRef event_ref
420
420
421
+ result_shape = utils._object_to_tuple(size)
421
422
if low == high:
422
- return utils.dpnp_descriptor(dpnp.full(size , low, dtype = dtype))
423
+ return utils.dpnp_descriptor(dpnp.full(result_shape , low, dtype = dtype))
423
424
424
425
# convert string type names (array.dtype) to C enum DPNPFuncType
425
426
param1_type = dpnp_dtype_to_DPNPFuncType(dtype)
@@ -428,7 +429,6 @@ cdef class MT19937:
428
429
kernel_data = get_dpnp_function_ptr(DPNP_FN_RNG_UNIFORM_EXT, param1_type, param1_type)
429
430
430
431
# ceate result array with type given by FPTR data
431
- result_shape = utils._object_to_tuple(size)
432
432
result = utils.create_output_descriptor(result_shape,
433
433
kernel_data.return_type,
434
434
None ,
@@ -991,11 +991,12 @@ cpdef utils.dpnp_descriptor dpnp_rng_negative_binomial(double a, double p, size)
991
991
cdef c_dpctl.DPCTLSyclQueueRef q_ref
992
992
cdef c_dpctl.DPCTLSyclEventRef event_ref
993
993
994
+ result_shape = utils._object_to_tuple(size)
994
995
if p == 0.0 :
995
996
filled_val = numpy.iinfo(dtype).min
996
- return utils.dpnp_descriptor(dpnp.full(size , filled_val, dtype = dtype))
997
+ return utils.dpnp_descriptor(dpnp.full(result_shape , filled_val, dtype = dtype))
997
998
elif p == 1.0 :
998
- return utils.dpnp_descriptor(dpnp.full(size , 0 , dtype = dtype))
999
+ return utils.dpnp_descriptor(dpnp.full(result_shape , 0 , dtype = dtype))
999
1000
else :
1000
1001
# convert string type names (array.dtype) to C enum DPNPFuncType
1001
1002
param1_type = dpnp_dtype_to_DPNPFuncType(dtype)
@@ -1004,7 +1005,6 @@ cpdef utils.dpnp_descriptor dpnp_rng_negative_binomial(double a, double p, size)
1004
1005
kernel_data = get_dpnp_function_ptr(DPNP_FN_RNG_NEGATIVE_BINOMIAL_EXT, param1_type, param1_type)
1005
1006
1006
1007
# ceate result array with type given by FPTR data
1007
- result_shape = utils._object_to_tuple(size)
1008
1008
result = utils.create_output_descriptor(result_shape, kernel_data.return_type, None )
1009
1009
1010
1010
result_sycl_queue = result.get_array().sycl_queue
@@ -1108,8 +1108,9 @@ cpdef utils.dpnp_descriptor dpnp_rng_poisson(double lam, size):
1108
1108
cdef c_dpctl.DPCTLSyclQueueRef q_ref
1109
1109
cdef c_dpctl.DPCTLSyclEventRef event_ref
1110
1110
1111
+ result_shape = utils._object_to_tuple(size)
1111
1112
if lam == 0 :
1112
- return utils.dpnp_descriptor(dpnp.full(size , 0 , dtype = dtype))
1113
+ return utils.dpnp_descriptor(dpnp.full(result_shape , 0 , dtype = dtype))
1113
1114
else :
1114
1115
# convert string type names (array.dtype) to C enum DPNPFuncType
1115
1116
param1_type = dpnp_dtype_to_DPNPFuncType(dtype)
@@ -1118,7 +1119,6 @@ cpdef utils.dpnp_descriptor dpnp_rng_poisson(double lam, size):
1118
1119
kernel_data = get_dpnp_function_ptr(DPNP_FN_RNG_POISSON_EXT, param1_type, param1_type)
1119
1120
1120
1121
# ceate result array with type given by FPTR data
1121
- result_shape = utils._object_to_tuple(size)
1122
1122
result = utils.create_output_descriptor(result_shape, kernel_data.return_type, None )
1123
1123
1124
1124
result_sycl_queue = result.get_array().sycl_queue
@@ -1187,8 +1187,9 @@ cpdef utils.dpnp_descriptor dpnp_rng_rayleigh(double scale, size):
1187
1187
cdef c_dpctl.DPCTLSyclQueueRef q_ref
1188
1188
cdef c_dpctl.DPCTLSyclEventRef event_ref
1189
1189
1190
+ result_shape = utils._object_to_tuple(size)
1190
1191
if scale == 0.0 :
1191
- return utils.dpnp_descriptor(dpnp.full(size , 0.0 , dtype = dtype))
1192
+ return utils.dpnp_descriptor(dpnp.full(result_shape , 0.0 , dtype = dtype))
1192
1193
else :
1193
1194
# convert string type names (array.dtype) to C enum DPNPFuncType
1194
1195
param1_type = dpnp_dtype_to_DPNPFuncType(dtype)
@@ -1197,7 +1198,6 @@ cpdef utils.dpnp_descriptor dpnp_rng_rayleigh(double scale, size):
1197
1198
kernel_data = get_dpnp_function_ptr(DPNP_FN_RNG_RAYLEIGH_EXT, param1_type, param1_type)
1198
1199
1199
1200
# ceate result array with type given by FPTR data
1200
- result_shape = utils._object_to_tuple(size)
1201
1201
result = utils.create_output_descriptor(result_shape, kernel_data.return_type, None )
1202
1202
1203
1203
result_sycl_queue = result.get_array().sycl_queue
@@ -1348,8 +1348,9 @@ cpdef utils.dpnp_descriptor dpnp_rng_standard_gamma(double shape, size):
1348
1348
cdef c_dpctl.DPCTLSyclQueueRef q_ref
1349
1349
cdef c_dpctl.DPCTLSyclEventRef event_ref
1350
1350
1351
+ result_shape = utils._object_to_tuple(size)
1351
1352
if shape == 0.0 :
1352
- return utils.dpnp_descriptor(dpnp.full(size , 0.0 , dtype = dtype))
1353
+ return utils.dpnp_descriptor(dpnp.full(result_shape , 0.0 , dtype = dtype))
1353
1354
else :
1354
1355
# convert string type names (array.dtype) to C enum DPNPFuncType
1355
1356
param1_type = dpnp_dtype_to_DPNPFuncType(dtype)
@@ -1358,7 +1359,6 @@ cpdef utils.dpnp_descriptor dpnp_rng_standard_gamma(double shape, size):
1358
1359
kernel_data = get_dpnp_function_ptr(DPNP_FN_RNG_STANDARD_GAMMA_EXT, param1_type, param1_type)
1359
1360
1360
1361
# ceate result array with type given by FPTR data
1361
- result_shape = utils._object_to_tuple(size)
1362
1362
result = utils.create_output_descriptor(result_shape, kernel_data.return_type, None )
1363
1363
1364
1364
result_sycl_queue = result.get_array().sycl_queue
0 commit comments