Merge pull request #260 from IntelPython/fix/gpairs_dpnp_allocation

ZzEeKkAa · web-flow · commit 9c7e9a1cb546 · 2023-05-11T17:03:23.000-04:00
Fix gpairs dpnp allocation
diff --git a/dpbench/benchmarks/gpairs/gpairs_dpnp.py b/dpbench/benchmarks/gpairs/gpairs_dpnp.py
@@ -12,7 +12,8 @@ def _gpairs_impl(x1, y1, z1, w1, x2, y2, z2, w2, rbins):
         + np.square(z2 - z1[:, None])
     )
     return np.array(
-        [np.outer(w1, w2)[dm <= rbins[k]].sum() for k in range(len(rbins))]
+        [np.outer(w1, w2)[dm <= rbins[k]].sum() for k in range(len(rbins))],
+        device=x1.device,
     )
 
 

Original file line number	Diff line number	Diff line change
`@@ -12,7 +12,8 @@ def _gpairs_impl(x1, y1, z1, w1, x2, y2, z2, w2, rbins):`
`12`	`12`	`+ np.square(z2 - z1[:, None])`
`13`	`13`	`)`
`14`	`14`	`return np.array(`
`15`		`- [np.outer(w1, w2)[dm <= rbins[k]].sum() for k in range(len(rbins))]`
	`15`	`+ [np.outer(w1, w2)[dm <= rbins[k]].sum() for k in range(len(rbins))],`
	`16`	`+ device=x1.device,`
`16`	`17`	`)`
`17`	`18`
`18`	`19`