@@ -143,10 +143,10 @@ def _accumulate_common(
143143 _manager .add_event_pair (ht_e , acc_ev )
144144 if not (orig_out is None or out is orig_out ):
145145 # Copy the out data from temporary buffer to original memory
146- ht_e_cpy , cpy_ev = ti ._copy_usm_ndarray_into_usm_ndarray (
146+ ht_e_cpy , cpy_e = ti ._copy_usm_ndarray_into_usm_ndarray (
147147 src = out , dst = orig_out , sycl_queue = q , depends = [acc_ev ]
148148 )
149- _manager .add_event_pair (ht_e_cpy , cpy_ev )
149+ _manager .add_event_pair (ht_e_cpy , cpy_e )
150150 out = orig_out
151151 else :
152152 if _dtype_supported (res_dt , res_dt ):
@@ -203,10 +203,10 @@ def _accumulate_common(
203203 depends = [cpy_e ],
204204 )
205205 _manager .add_event_pair (ht_e , acc_ev )
206- ht_e_cpy2 , cpy_ev2 = ti ._copy_usm_ndarray_into_usm_ndarray (
206+ ht_e_cpy2 , cpy_e2 = ti ._copy_usm_ndarray_into_usm_ndarray (
207207 src = tmp_res , dst = out , sycl_queue = q , depends = [acc_ev ]
208208 )
209- _manager .add_event_pair (ht_e_cpy2 , cpy_ev2 )
209+ _manager .add_event_pair (ht_e_cpy2 , cpy_e2 )
210210
211211 if appended_axis :
212212 out = dpt .squeeze (out )
0 commit comments