Skip to content

Commit 76fbbc7

Browse files
ENH: some optimization for random.triangular (#580)
let queue managing on device
1 parent e2aebdd commit 76fbbc7

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

dpnp/backend/kernels/dpnp_krnl_random.cpp

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -781,8 +781,7 @@ void dpnp_rng_triangular_c(
781781
_DataType ratio, lpr, rpr;
782782

783783
mkl_rng::uniform<_DataType> uniform_distribution(d_zero, d_one);
784-
auto event_out = mkl_rng::generate(uniform_distribution, DPNP_RNG_ENGINE, size, result1);
785-
event_out.wait();
784+
auto event_uniform = mkl_rng::generate(uniform_distribution, DPNP_RNG_ENGINE, size, result1);
786785

787786
{
788787
_DataType wtot, wl, wr;
@@ -826,11 +825,12 @@ void dpnp_rng_triangular_c(
826825
}
827826
};
828827
auto kernel_func = [&](cl::sycl::handler& cgh) {
828+
cgh.depends_on({event_uniform});
829829
cgh.parallel_for<class dpnp_rng_triangular_ration_acceptance_c_kernel<_DataType>>(gws,
830830
kernel_parallel_for_func);
831831
};
832-
event_out = DPNP_QUEUE.submit(kernel_func);
833-
event_out.wait();
832+
auto event_ration_acceptance = DPNP_QUEUE.submit(kernel_func);
833+
event_ration_acceptance.wait();
834834
}
835835

836836
template <typename _DataType>

0 commit comments

Comments
 (0)