pycuda索引numpy阵列的错误整数的错误

Question

我正在将第一步转移到Pycuda上进行一些并行计算，并且遇到了一种我不了解的行为。 我从可以在Pycuda官方网站上找到的非常基本的教程开始（一个简单的脚本，将数组的所有元素加倍 https://documen.tician.de/pycuda/tutorial.html ）。代码如下：

import pycuda.driver as cuda
import pycuda.autoinit
from pycuda.compiler import SourceModule
import numpy
a = numpy.random.randn(4,4)
a = a.astype(numpy.float32)
a_gpu = cuda.mem_alloc(a.nbytes)
cuda.memcpy_htod(a_gpu, a)
mod = SourceModule("""
  __global__ void doublify(float *a)
  {
    int idx = threadIdx.x + threadIdx.y*4;
    a[idx] *= 2;
  }
  """)
func = mod.get_function("doublify")
func(a_gpu, block=(4,4,1))
a_doubled = numpy.empty_like(a)
cuda.memcpy_dtoh(a_doubled, a_gpu)
print(a_doubled)
print(a)

非常清楚并且有效。结果是一个示例，

[[-1.9951048  -1.7537887  -1.3228793  -1.1585734 ]
 [-0.96863186 -1.7235669  -0.3331826  -1.1527038 ]
 [ 2.4142797  -0.35531005  1.8844942   3.996446  ]
 [ 1.400629   -2.7957075  -0.78042877  0.13829945]]
[[-0.9975524  -0.87689435 -0.66143966 -0.5792867 ]
 [-0.48431593 -0.86178344 -0.1665913  -0.5763519 ]
 [ 1.2071398  -0.17765503  0.9422471   1.998223  ]
 [ 0.7003145  -1.3978537  -0.39021438  0.06914973]]

但后来我试图修改代码以处理整数数字：

import pycuda.driver as cuda
import pycuda.autoinit
from pycuda.compiler import SourceModule
import numpy

a = numpy.array([[1,2,3,4], [1,2,3,4], [1,2,3,4], [1,2,3,4]])

a_gpu = cuda.mem_alloc(a.nbytes)
cuda.memcpy_htod(a_gpu, a)
mod = SourceModule("""
  __global__ void doublify(int *a)
  {
    int idx = threadIdx.x + threadIdx.y*4;
    a[idx] *= 2;
  }
  """)
func = mod.get_function("doublify")
func(a_gpu, block=(4,4,1))
a_doubled = numpy.empty_like(a)
cuda.memcpy_dtoh(a_doubled, a_gpu)
print(a_doubled)
print(a)

...这是不起作用的。仅将2D阵列的一部分乘以2，其余的是没有变化的。结果：

[[2 4 6 8]
 [2 4 6 8]
 [1 2 3 4]
 [1 2 3 4]]
[[1 2 3 4]
 [1 2 3 4]
 [1 2 3 4]
 [1 2 3 4]]

为什么会发生这种情况？教程和修改的代码有什么区别？

感谢所有人！

Answer 1

好的，我有点解决了使用浮点类型的待，即使我需要与整数一起工作。显然，在为整数分配内存时，存在一些幕后机制，这与Pycuda不符。