Commit fb4d6109 authored by Michiel Cottaar's avatar Michiel Cottaar
Browse files

BUG: typo

parent 31b5e756
......@@ -137,13 +137,13 @@ def sparse_mult_prepare(sparse):
"""Returns a function that does the GPU multiplication for the provided sparse matrix.
"""
gpu_sparse = _sparse_mult_to_gpu(sparse)
M = sparse.shape[0]
output = gpuarray.zeros((M, ), dtype='f4' if dtype == 'float' else 'f8')
if len(gpu_sparse[-1]) == 0:
def empty_sparse_mult(vec_in):
return output
return empty_sparse_mult
M = sparse.shape[0]
gpu_func = cached_func(_sparse_mult_mod_vector(gpu_sparse[0].size - 1), 'sparse_matrix_mult')
input = gpuarray.zeros((sparse.shape[1], ), dtype='f4' if dtype == 'float' else 'f8')
stream = drv.Stream()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment