从CUDA中的各种分布生成随机数

我正在玩GPU上的MCMC，并需要针对CUDA编写的各种采样器的实现。

我在StackOverflow上看到的大多数帖子都涉及统一，二项和正常采样。 是否有任何库允许R中的dpqr函数的简单性和多样性（请参阅此页）？

我希望能够从伽玛，正态，二项式以及贝叶斯问题（逆卡方，逆伽马）中使用的逆分布中抽样，并且宁愿不必使用逆概率变换和接受拒绝采样。

---

对于Gamma分配，这是我目前使用的。 GSL功能修改为与CuRAND一起使用。

__device__ double ran_gamma (curandState localState, const double a, const double b){
/* assume a > 0 */

if (a < 1){
    double u = curand_uniform_double(&localState);
    return ran_gamma (localState, 1.0 + a, b) * pow (u, 1.0 / a);
}

{
    double x, v, u;
    double d = a - 1.0 / 3.0;
    double c = (1.0 / 3.0) / sqrt (d);

    while (1){
        do{
            x = curand_normal_double(&localState);
            v = 1.0 + c * x;
        } while (v <= 0);

        v = v * v * v;
        u = curand_uniform_double(&localState);

        if (u < 1 - 0.0331 * x * x * x * x) 
            break;

        if (log (u) < 0.5 * x * x + d * (1 - v + log (v)))
            break;
    }
    return b * d * v;
}
}

链接地址: http://www.djcxy.com/p/24817.html

上一篇: Generating random numbers from various distributions in CUDA

下一篇: PHP: Random number from a normal distribution