Как выполнить поэлементное назначение в pycuda/scikits.cuda?

Вот код:

import pycuda.autoinit
import pycuda.gpuarray as gpuarray
import pycuda.driver as drv
import numpy as np

import scikits.cuda.linalg as culinalg
import scikits.cuda.misc as cumisc
culinalg.init()

ag = gpuarray.to_gpu(np.random.rand(1000,1000))
bg = gpuarray.to_gpu(np.zeros((1000,1000))
bg[:,:] = ag

Я получил следующую ошибку:

TypeError: 'GPUArray' object does not support item assignment

Так есть ли способ назначить матрицу другой существующей матрице в pycuda/scikits.cuda?


cuda scikits pycuda
person Xing Shi    schedule 15.11.2014    source источник

Ответы (1)


arrow_upward
1
arrow_downward

Хорошо, это не элегантный способ, а решение: используйте ElementwiseKernel pycuda:

import from pycuda.elementwise import ElementwiseKernel
ele_assign = ElementwiseKernel("double *a,double *b","a[i] = b[i]","ele_assign")
ele_assign(bg,ag)
person Xing Shi    schedule 15.11.2014