提交 d6697f9f authored 作者: Frederic Bastien's avatar Frederic Bastien

Small fix for OpenCL

上级 82af9d7b
...@@ -75,7 +75,7 @@ KERNEL void k_multi_warp_multinomial( ...@@ -75,7 +75,7 @@ KERNEL void k_multi_warp_multinomial(
) )
{ {
// each thread takes care of one multinomial draw // each thread takes care of one multinomial draw
int n = blockDim.x*blockIdx.x + threadIdx.x; int n = LDIM_0*GID_0 + LID_0;
if (n < nb_multi) if (n < nb_multi)
{ {
float cummul = 0.; float cummul = 0.;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论