PyCUDA学习:gpuarray与kernel的抽象原型

2018-02-27 11:21:41来源:oschina作者:IOTService人点击

分享
# -*-coding:utf-8 -*-
import pycuda.gpuarray as gpuarray
import pycuda.driver as cuda
import pycuda.autoinit
import numpy
a = numpy.asarray([
[1,2,3,4],[5,6,7,8],[9,10,11,12],[13,14,15,16]
])
# 将数据复制到设备
a_gpu = gpuarray.to_gpu(a.astype(numpy.float32))
# 执行kernel并将数据复制到主机
a_doubled = (2*a_gpu).get()
print (a_doubled)
print (a_gpu)

最新文章

123

最新摄影

闪念基因

微信扫一扫

第七城市微信公众平台