Method exec

dnn/src/cambricon/rng/opr_impl.cpp:42–51

        : UniformRNG(handle), m_rand_handle(cnnl_handle(handle)) {}

void UniformRNGImpl::exec(_megdnn_tensor_inout dst, _megdnn_workspace workspace) {
    check_exec(dst.layout, workspace.size);
    megdnn_assert(dst.layout.dtype == dtype::Float32(), "only float32 supported");
    m_rand_handle.ensure_seed(cnnl_handle(handle()), m_param.seed);
    auto cnnl_dtype = convert_to_cnnl_datatype(dst.layout.dtype.enumv());
    cnnl_check(cnnlRandGenerateUniform(
            cnnl_handle(this->handle()), m_rand_handle.gen(), cnnl_dtype,
            m_rand_handle.state(), dst.layout.total_nr_elems(), /*min=*/0.f,
            /*max=*/1.f, dst.raw_ptr()));
}

GaussianRNGImpl::GaussianRNGImpl(Handle* handle)
        : GaussianRNG(handle), m_rand_handle(cnnl_handle(handle)) {}