qwe

Author: vlad-perevezentsev

perf.py

@@ -0,0 +1,48 @@
+import dpnp
+import numpy as np
+from dpnp.dpnp_utils import get_usm_allocations
+from dpnp.tests.helper import generate_random_numpy_array
+import time
+from IPython import get_ipython
+
+ipython = get_ipython()
+if ipython is None:
+    from IPython.terminal.interactiveshell import TerminalInteractiveShell
+    ipython = TerminalInteractiveShell()
+
+# dtypes = ['c8', 'c16']
+dtypes = ['f4', 'f8', 'c8', 'c16']
+
+n = 8192 * 8192 * 10
+
+for dtype in dtypes:
+    print(f"\n=== dtype: {dtype} ===")
+
+    a = generate_random_numpy_array((n,), dtype=dtype, seed_value=81)
+    b = generate_random_numpy_array((n,), dtype=dtype, seed_value=76)
+
+    atol = 1e-5
+    rtol = 1e-3
+    equal_nan = False
+
+    # dpnp arrays on GPU
+    a_dp = dpnp.array(a, device='cpu')
+    b_dp = dpnp.array(b, device='cpu')
+
+    a_dp = a_dp[::-1]
+    b_dp = b_dp[::-1]
+    _, exec_q = get_usm_allocations([a_dp, b_dp])
+
+    # Cold run
+    # _ = np.isclose(a, b, rtol=rtol, atol=atol, equal_nan=equal_nan)
+    _ = dpnp.isclose(a_dp, b_dp, rtol=rtol, atol=atol, equal_nan=equal_nan)
+    # _ = dpnp.isclose(a_dp, b_dp, rtol=rtol, atol=atol, equal_nan=equal_nan, new=True)
+    exec_q.wait()
+
+    # time.sleep(1)
+    # print("NumPy:")
+    # ipython.run_line_magic('timeit', 'np.isclose(a, b, rtol=rtol, atol=atol, equal_nan=equal_nan)')
+
+    time.sleep(1)
+    print("DPNP (GPU, New):")
+    ipython.run_line_magic('timeit', 'dpnp.isclose(a_dp, b_dp, rtol=rtol, atol=atol, equal_nan=equal_nan); exec_q.wait()')

test.py

@@ -0,0 +1,23 @@
+import dpnp
+from dpnp.tests.helper import generate_random_numpy_array
+import numpy
+
+a = generate_random_numpy_array((16,16), 'c8', hermitian=True, seed_value=64, low=-5, high=5)
+a_dp = dpnp.array(a)
+
+B_dp = dpnp.linalg.pinv(a_dp, hermitian=True)
+reconstructed = dpnp.dot(dpnp.dot(a_dp, B_dp), a_dp)
+
+res = dpnp.isclose(reconstructed, a_dp, rtol=1e-03, atol=1e-03)
+
+print(reconstructed[5][15])
+print(a_dp[5][15])
+print(res[5][15])
+
+print("DPNP isclose")
+print(dpnp.isclose(reconstructed[5][15], a_dp[5][15], rtol=1e-03, atol=1e-03))
+
+print("Numpy isclose")
+print(numpy.isclose(dpnp.asnumpy(reconstructed[5][15]), dpnp.asnumpy(a_dp[5][15]), rtol=1e-03, atol=1e-03))
+
+# print(res[])