ENH: misc update; random.weibull (#724)

Author: samir-nasibli

dpnp/backend/kernels/dpnp_krnl_random.cpp

@@ -1505,18 +1505,21 @@ void dpnp_rng_weibull_c(void* result, const double alpha, const size_t size)
     {
         return;
     }
-    _DataType* result1 = reinterpret_cast<_DataType*>(result);
 
-    // set displacement a
-    const _DataType a = (_DataType(0.0));
-
-    // set beta
-    const _DataType beta = (_DataType(1.0));
+    if (alpha == 0)
+    {
+        dpnp_zeros_c<_DataType>(result, size);
+    }
+    else
+    {
+        _DataType* result1 = reinterpret_cast<_DataType*>(result);
+        const _DataType a = (_DataType(0.0));
+        const _DataType beta = (_DataType(1.0));
 
-    mkl_rng::weibull<_DataType> distribution(alpha, a, beta);
-    // perform generation
-    auto event_out = mkl_rng::generate(distribution, DPNP_RNG_ENGINE, size, result1);
-    event_out.wait();
+        mkl_rng::weibull<_DataType> distribution(alpha, a, beta);
+        auto event_out = mkl_rng::generate(distribution, DPNP_RNG_ENGINE, size, result1);
+        event_out.wait();
+    }
 }
 
 template <typename _DataType>

dpnp/random/dpnp_algo_random.pyx

@@ -1133,23 +1133,19 @@ cpdef dparray dpnp_rng_weibull(double a, size):
     cdef DPNPFuncData kernel_data
     cdef fptr_dpnp_rng_weibull_c_1out_t func
 
-    if a == 0.0:
-        result = dparray(size, dtype=dtype)
-        result.fill(0.0)
-    else:
-        # convert string type names (dparray.dtype) to C enum DPNPFuncType
-        param1_type = dpnp_dtype_to_DPNPFuncType(numpy.float64)
+    # convert string type names (dparray.dtype) to C enum DPNPFuncType
+    param1_type = dpnp_dtype_to_DPNPFuncType(numpy.float64)
 
-        # get the FPTR data structure
-        kernel_data = get_dpnp_function_ptr(DPNP_FN_RNG_WEIBULL, param1_type, param1_type)
+    # get the FPTR data structure
+    kernel_data = get_dpnp_function_ptr(DPNP_FN_RNG_WEIBULL, param1_type, param1_type)
 
-        result_type = dpnp_DPNPFuncType_to_dtype(< size_t > kernel_data.return_type)
-        # ceate result array with type given by FPTR data
-        result = dparray(size, dtype=result_type)
+    result_type = dpnp_DPNPFuncType_to_dtype(< size_t > kernel_data.return_type)
+    # ceate result array with type given by FPTR data
+    result = dparray(size, dtype=result_type)
 
-        func = <fptr_dpnp_rng_weibull_c_1out_t > kernel_data.ptr
-        # call FPTR function
-        func(result.get_data(), a, result.size)
+    func = <fptr_dpnp_rng_weibull_c_1out_t > kernel_data.ptr
+    # call FPTR function
+    func(result.get_data(), a, result.size)
 
     return result