pycuda
pycuda copied to clipboard
Provide gpuarray.dot
A simple enhancement that adds A.dot(x)
, versus skcuda.dot(A, x)
. This is most useful with many dots (but a __matmul__
operator would be better).
Tests not run, but this is only a three line change that depends on skcuda.dot
.