提交 9432f511 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix last bug.

上级 4c101c36
......@@ -45,7 +45,8 @@ def gemv(alpha, A, x, beta, y):
handle = scikits.cuda.misc._global_cublas_handle
cublas.cublasSgemv(handle, 't', A.shape[1], A.shape[0], alpha,
A.gpudata, A.strides[0], x.gpudata, x.strides[0],
A.gpudata, max(A.strides[0], A.strides[1]),
x.gpudata, x.strides[0],
beta, y.gpudata, y.strides[0])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论