Hi,

I have been using scikit-cudablas
(https://github.com/lebedov/scikit-cuda). It rocks!

Does anyone have a 2d matrix multiplication example with DgemmBatched?

Similar to,

>>> a=np.random.randint(0,3,(16,2)); b=np.random.randint(0,4,(2,16))
>>> np.dot(a,b)

I have been using,
https://github.com/lebedov/scikit-cuda/blob/7e7300474286019c917a6c8a4bca59405c64fbce/tests/test_cublas.py#L531
but it has too many dimensions and I keep getting confused by too many
dimensions for DgemmBatched

_______________________________________________
PyCUDA mailing list
PyCUDA@tiker.net
http://lists.tiker.net/listinfo/pycuda

Reply via email to