adding numba numpy version of rambo

IntelPython · Apr 12, 2023 · b975431 · b975431
1 parent e91075c
commit b975431
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/dpbench/benchmarks/rambo/rambo_numba_dpex_n.py b/dpbench/benchmarks/rambo/rambo_numba_dpex_n.py
@@ -0,0 +1,19 @@
+# SPDX-FileCopyrightText: 2022 - 2023 Intel Corporation
+#
+# SPDX-License-Identifier: Apache-2.0
+
+import dpnp as np
+from numba_dpex import dpjit
+
+
+@dpjit
+def rambo(nevts, nout, C1, F1, Q1, output):
+    C = 2.0 * C1 - 1.0
+    S = np.sqrt(1 - np.square(C))
+    F = 2.0 * np.pi * F1
+    Q = -np.log(Q1)
+
+    output[:, :, 0] = Q
+    output[:, :, 1] = Q * S * np.sin(F)
+    output[:, :, 2] = Q * S * np.cos(F)
+    output[:, :, 3] = Q * C