--amend

vtavana · vtavana · commit 8d1f68b6be07 · 2023-05-18T22:08:31.000-05:00
diff --git a/dpctl/tensor/_elementwise_common.py b/dpctl/tensor/_elementwise_common.py
@@ -124,6 +124,7 @@ def __call__(self, x, out=None, order="K"):
                 )
 
         ht, _ = self.unary_fn_(buf, out, sycl_queue=exec_q, depends=[copy_ev])
+        ht_copy_ev.wait()
         ht.wait()
 
         return out

Original file line number	Diff line number	Diff line change
`@@ -124,6 +124,7 @@ def __call__(self, x, out=None, order="K"):`
`124`	`124`	`)`
`125`	`125`
`126`	`126`	`ht, _ = self.unary_fn_(buf, out, sycl_queue=exec_q, depends=[copy_ev])`
	`127`	`+ ht_copy_ev.wait()`
`127`	`128`	`ht.wait()`
`128`	`129`
`129`	`130`	`return out`