fix previous commit
This commit is contained in:
parent
e94a07db28
commit
0e413dc267
@ -53,9 +53,6 @@ def cg(a, b, max_iters=10000, err_thresh=1e-6, context=None, queue=None, verbose
|
|||||||
p_step = ops.create_p_step(context)
|
p_step = ops.create_p_step(context)
|
||||||
dot = ops.create_dot(context)
|
dot = ops.create_dot(context)
|
||||||
|
|
||||||
def a_step(E, H, p, events):
|
|
||||||
return a_step_full(E, H, p, inv_dxes, oeps, invm, gpec, gpmc, Pl, Pr, events)
|
|
||||||
|
|
||||||
'''
|
'''
|
||||||
Start the solve
|
Start the solve
|
||||||
'''
|
'''
|
||||||
|
@ -253,6 +253,6 @@ def create_a_csr(context):
|
|||||||
arguments=', '.join((v_out_args, m_args, v_in_args)))
|
arguments=', '.join((v_out_args, m_args, v_in_args)))
|
||||||
|
|
||||||
def spmv(v_out, m, v_in, e):
|
def spmv(v_out, m, v_in, e):
|
||||||
return spmv_kernel(v_out, m.row_ptr, m.col_ind, m.data, v_in, wait_for=e)
|
return [spmv_kernel(v_out, m.row_ptr, m.col_ind, m.data, v_in, wait_for=e)]
|
||||||
|
|
||||||
return spmv
|
return spmv
|
||||||
|
Loading…
Reference in New Issue
Block a user