FIX: rayleigh dist (#513)

Author: samir-nasibli

dpnp/backend/kernels/dpnp_krnl_random.cpp

@@ -33,8 +33,9 @@
 #include "dpnp_utils.hpp"
 #include "queue_sycl.hpp"
 
-namespace mkl_rng = oneapi::mkl::rng;
 namespace mkl_blas = oneapi::mkl::blas;
+namespace mkl_rng = oneapi::mkl::rng;
+namespace mkl_vm = oneapi::mkl::vm;
 
 /**
  * Use get/set functions to access/modify this variable
@@ -461,15 +462,23 @@ void dpnp_rng_rayleigh_c(void* result, _DataType scale, size_t size)
         return;
     }
 
-    // set displacement a
-    const _DataType a = (_DataType(0.0));
+    cl::sycl::vector_class<cl::sycl::event> no_deps;
+
+    const _DataType a = 0.0;
+    const _DataType beta = 2.0;
 
     _DataType* result1 = reinterpret_cast<_DataType*>(result);
 
-    mkl_rng::rayleigh<_DataType> distribution(a, scale);
-    // perform generation
+    mkl_rng::exponential<_DataType> distribution(a, beta);;
+
     auto event_out = mkl_rng::generate(distribution, DPNP_RNG_ENGINE, size, result1);
     event_out.wait();
+    event_out = mkl_vm::sqrt(DPNP_QUEUE, size, result1, result1, no_deps, mkl_vm::mode::ha);
+    event_out.wait();
+    // with MKL
+    // event_out = mkl_blas::axpy(DPNP_QUEUE, size, scale, result1, 1, result1, 1);
+    // event_out.wait();
+    for(size_t i = 0; i < size; i++) result1[i] *= scale;
 }
 
 template <typename _DataType>

tests/skipped_tests.tbl

@@ -139,7 +139,6 @@ tests/test_linalg.py::test_norm3[(1, 2)-3-[[[1, 2], [3, 4]], [[5, 6], [7, 8]]]]
 tests/test_linalg.py::test_norm3[(1, 2)-3-[[[1, 0], [3, 0]], [[5, 0], [7, 0]]]]
 tests/test_linalg.py::test_svd[(3,4)-complex128]
 tests/test_linalg.py::test_svd[(5,3)-complex128]
-tests/test_random.py::TestDistributionsRayleigh::test_moments
 tests/third_party/cupy/binary_tests/test_elementwise.py::TestElementwise::test_bitwise_and
 tests/third_party/cupy/binary_tests/test_elementwise.py::TestElementwise::test_bitwise_or
 tests/third_party/cupy/binary_tests/test_elementwise.py::TestElementwise::test_bitwise_xor

tests/skipped_tests_gpu.tbl

@@ -156,7 +156,6 @@ tests/test_linalg.py::test_svd[(16,16)-float64]
 tests/test_linalg.py::test_svd[(16,16)-float32]
 tests/test_linalg.py::test_svd[(16,16)-int64]
 tests/test_linalg.py::test_svd[(16,16)-int32]
-tests/test_random.py::TestDistributionsRayleigh::test_moments
 tests/test_statistics.py::test_median[2-float32]
 tests/test_statistics.py::test_median[2-float64]
 tests/third_party/cupy/binary_tests/test_elementwise.py::TestElementwise::test_bitwise_and