import numpy """ """ import pycuda.driver as cuda import pycuda.tools import pycuda.gpuarray as gpuarray import pycuda.autoinit, pycuda.compiler
a=numpy.arange(400) a_gpu=gpuarray.arange(400,dtype=numpy.float32) start=cuda.Event() end=cuda.Event() start.record() gpuarray.sum(a_gpu).get()/a.size end.record() end.synchronize() print "GPU array time: %fs" %(start.time_till(end)*1e-3) start.record() numpy.sum(a)/a.size end.record() end.synchronize() print "numpy array time: %fs" %(start.time_till(end)*1e-3) GPU array time: 0.000377s numpy array time: 0.000001s Efren A. Serra (Contractor) DeVine Consulting, Inc. Naval Research Laboratory Marine Meteorology Division 7 Grace Hopper Ave., STOP 2 Monterey, CA 93943 Code 7542 Office: 831-656-4650 _______________________________________________ PyCUDA mailing list PyCUDA@tiker.net http://lists.tiker.net/listinfo/pycuda