Merge pull request #1807 from IntelPython/dont-add-compute-event-into-host-task-list

oleksandr-pavlyk · web-flow · commit 6174a9380d91 · 2024-08-24T16:55:43.000-05:00
diff --git a/dpctl/tensor/libtensor/source/boolean_advanced_indexing.cpp b/dpctl/tensor/libtensor/source/boolean_advanced_indexing.cpp
@@ -370,8 +370,6 @@ py_extract(const dpctl::tensor::usm_ndarray &src,
         host_task_events.push_back(cleanup_tmp_allocations_ev);
     }
 
-    host_task_events.push_back(extract_ev);
-
     sycl::event py_obj_management_host_task_ev = dpctl::utils::keep_args_alive(
         exec_q, {src, cumsum, dst}, host_task_events);
 
@@ -691,8 +689,6 @@ py_place(const dpctl::tensor::usm_ndarray &dst,
         host_task_events.push_back(cleanup_tmp_allocations_ev);
     }
 
-    host_task_events.push_back(place_ev);
-
     sycl::event py_obj_management_host_task_ev = dpctl::utils::keep_args_alive(
         exec_q, {dst, cumsum, rhs}, host_task_events);
 
diff --git a/dpctl/tensor/libtensor/source/linalg_functions/dot.cpp b/dpctl/tensor/libtensor/source/linalg_functions/dot.cpp
@@ -593,7 +593,6 @@ py_dot(const dpctl::tensor::usm_ndarray &x1,
                     });
                 });
             host_task_events.push_back(cleanup_tmp_allocations_ev);
-            host_task_events.push_back(dot_ev);
         }
         else { // if (call_batched)
             using shT = std::vector<py::ssize_t>;
@@ -802,7 +801,6 @@ py_dot(const dpctl::tensor::usm_ndarray &x1,
                     });
                 });
             host_task_events.push_back(cleanup_tmp_allocations_ev);
-            host_task_events.push_back(dot_ev);
         }
     }
     return std::make_pair(
diff --git a/dpctl/tensor/libtensor/source/repeat.cpp b/dpctl/tensor/libtensor/source/repeat.cpp
@@ -356,8 +356,6 @@ py_repeat_by_sequence(const dpctl::tensor::usm_ndarray &src,
         host_task_events.push_back(cleanup_tmp_allocations_ev);
     }
 
-    host_task_events.push_back(repeat_ev);
-
     sycl::event py_obj_management_host_task_ev = dpctl::utils::keep_args_alive(
         exec_q, {src, reps, cumsum, dst}, host_task_events);
 
@@ -504,7 +502,6 @@ py_repeat_by_sequence(const dpctl::tensor::usm_ndarray &src,
             });
         });
     host_task_events.push_back(cleanup_tmp_allocations_ev);
-    host_task_events.push_back(repeat_ev);
 
     sycl::event py_obj_management_host_task_ev = dpctl::utils::keep_args_alive(
         exec_q, {src, reps, cumsum, dst}, host_task_events);
@@ -734,8 +731,6 @@ py_repeat_by_scalar(const dpctl::tensor::usm_ndarray &src,
         host_task_events.push_back(cleanup_tmp_allocations_ev);
     }
 
-    host_task_events.push_back(repeat_ev);
-
     sycl::event py_obj_management_host_task_ev =
         dpctl::utils::keep_args_alive(exec_q, {src, dst}, host_task_events);
 
@@ -846,7 +841,6 @@ py_repeat_by_scalar(const dpctl::tensor::usm_ndarray &src,
         });
 
     host_task_events.push_back(cleanup_tmp_allocations_ev);
-    host_task_events.push_back(repeat_ev);
 
     sycl::event py_obj_management_host_task_ev =
         dpctl::utils::keep_args_alive(exec_q, {src, dst}, host_task_events);

Original file line number	Diff line number	Diff line change
`@@ -370,8 +370,6 @@ py_extract(const dpctl::tensor::usm_ndarray &src,`
`370`	`370`	`host_task_events.push_back(cleanup_tmp_allocations_ev);`
`371`	`371`	`}`
`372`	`372`
`373`		`- host_task_events.push_back(extract_ev);`
`374`		`-`
`375`	`373`	`sycl::event py_obj_management_host_task_ev = dpctl::utils::keep_args_alive(`
`376`	`374`	`exec_q, {src, cumsum, dst}, host_task_events);`
`377`	`375`
`@@ -691,8 +689,6 @@ py_place(const dpctl::tensor::usm_ndarray &dst,`
`691`	`689`	`host_task_events.push_back(cleanup_tmp_allocations_ev);`
`692`	`690`	`}`
`693`	`691`
`694`		`- host_task_events.push_back(place_ev);`
`695`		`-`
`696`	`692`	`sycl::event py_obj_management_host_task_ev = dpctl::utils::keep_args_alive(`
`697`	`693`	`exec_q, {dst, cumsum, rhs}, host_task_events);`
`698`	`694`
Original file line number	Diff line number	Diff line change
`@@ -593,7 +593,6 @@ py_dot(const dpctl::tensor::usm_ndarray &x1,`
`593`	`593`	`});`
`594`	`594`	`});`
`595`	`595`	`host_task_events.push_back(cleanup_tmp_allocations_ev);`
`596`		`- host_task_events.push_back(dot_ev);`
`597`	`596`	`}`
`598`	`597`	`else { // if (call_batched)`
`599`	`598`	`using shT = std::vector<py::ssize_t>;`
`@@ -802,7 +801,6 @@ py_dot(const dpctl::tensor::usm_ndarray &x1,`
`802`	`801`	`});`
`803`	`802`	`});`
`804`	`803`	`host_task_events.push_back(cleanup_tmp_allocations_ev);`
`805`		`- host_task_events.push_back(dot_ev);`
`806`	`804`	`}`
`807`	`805`	`}`
`808`	`806`	`return std::make_pair(`