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) | ||||
|     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 | ||||
|     ''' | ||||
|  | ||||
| @ -253,6 +253,6 @@ def create_a_csr(context): | ||||
|                                     arguments=', '.join((v_out_args, m_args, v_in_args))) | ||||
| 
 | ||||
|     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 | ||||
|  | ||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user