hey all,
i tried to repeat the ease of use of pycuda for the CUBLAS apis. if
anyone else is looking at CUDA just for multiplying matrices, you might
want to take a look.
http://kered.org/blog/2009-04-13/easy-python-numpy-cuda-cublas/
simple use case:
import numpy
from pycublas import CUBLASMatrix
A = numpy.mat([[1,2,3],[4,5,6]],numpy.float32)
B = numpy.mat([[2,3],[4,5],[6,7]],numpy.float32)
C = (CUBLASMatrix(A)*CUBLASMatrix(B)).np_mat()
print C
gets speeds in the 30x-50x range for larger matrices.
just fyi,
derek
_______________________________________________
PyCuda mailing list
[email protected]
http://tiker.net/mailman/listinfo/pycuda_tiker.net
