Function fill_torch_cuda_uniform

src/framework/sampling/torch_random.cpp:277–284

}

void fill_torch_cuda_uniform(float * output, size_t count, uint64_t seed, uint64_t start_index) {
    if (output == nullptr && count != 0) {
        throw std::invalid_argument("torch CUDA uniform output pointer is null");
    }
    for (size_t index = 0; index < count; ++index) {
        output[index] = torch_cuda_uniform_element(seed, start_index + static_cast<uint64_t>(index));
    }
}

std::vector<float> generate_torch_cuda_uniform(size_t count, uint64_t seed, uint64_t start_index) {
    std::vector<float> output(count);