@@ -143,10 +143,10 @@ def _accumulate_common(
143
143
_manager .add_event_pair (ht_e , acc_ev )
144
144
if not (orig_out is None or out is orig_out ):
145
145
# 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 (
147
147
src = out , dst = orig_out , sycl_queue = q , depends = [acc_ev ]
148
148
)
149
- _manager .add_event_pair (ht_e_cpy , cpy_ev )
149
+ _manager .add_event_pair (ht_e_cpy , cpy_e )
150
150
out = orig_out
151
151
else :
152
152
if _dtype_supported (res_dt , res_dt ):
@@ -203,10 +203,10 @@ def _accumulate_common(
203
203
depends = [cpy_e ],
204
204
)
205
205
_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 (
207
207
src = tmp_res , dst = out , sycl_queue = q , depends = [acc_ev ]
208
208
)
209
- _manager .add_event_pair (ht_e_cpy2 , cpy_ev2 )
209
+ _manager .add_event_pair (ht_e_cpy2 , cpy_e2 )
210
210
211
211
if appended_axis :
212
212
out = dpt .squeeze (out )
0 commit comments