From b23348ce25edf2deb182c1ff7e4d71f27e8fe776 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Sun, 22 Sep 2019 00:06:28 +0200
Subject: [PATCH] WIP: Add test_jacobi_fixed_field_size_gpu
---
pystencils_tests/test_jacobi_llvm.py | 32 +++++++++++++++++++++++++++-
1 file changed, 31 insertions(+), 1 deletion(-)
diff --git a/pystencils_tests/test_jacobi_llvm.py b/pystencils_tests/test_jacobi_llvm.py
index 2965cf24..a072248d 100644
--- a/pystencils_tests/test_jacobi_llvm.py
+++ b/pystencils_tests/test_jacobi_llvm.py
@@ -1,6 +1,6 @@
import numpy as np
-from pystencils import Assignment, Field
+from pystencils import Assignment, Field, show_code
from pystencils.llvm import create_kernel, make_python_function
from pystencils.llvm.llvmjit import generate_and_jit
@@ -30,6 +30,32 @@ def test_jacobi_fixed_field_size():
np.testing.assert_almost_equal(error, 0.0)
+def test_jacobi_fixed_field_size_gpu():
+ size = (30, 20)
+
+ src_field_llvm = np.random.rand(*size)
+ src_field_py = np.copy(src_field_llvm)
+ dst_field_llvm = np.zeros(size)
+ dst_field_py = np.zeros(size)
+
+ f = Field.create_from_numpy_array("f", src_field_llvm)
+ d = Field.create_from_numpy_array("d", dst_field_llvm)
+
+ jacobi = Assignment(d[0, 0], (f[1, 0] + f[-1, 0] + f[0, 1] + f[0, -1]) / 4)
+ ast = create_kernel([jacobi], target='gpu')
+ print(show_code(ast))
+
+ for x in range(1, size[0] - 1):
+ for y in range(1, size[1] - 1):
+ dst_field_py[x, y] = 0.25 * (src_field_py[x - 1, y] + src_field_py[x + 1, y] +
+ src_field_py[x, y - 1] + src_field_py[x, y + 1])
+
+ jit = generate_and_jit(ast)
+ jit('kernel', dst_field_llvm, src_field_llvm)
+ error = np.sum(np.abs(dst_field_py - dst_field_llvm))
+ np.testing.assert_almost_equal(error, 0.0)
+
+
def test_jacobi_variable_field_size():
size = (3, 3, 3)
f = Field.create_generic("f", 3)
@@ -52,3 +78,7 @@ def test_jacobi_variable_field_size():
kernel()
error = np.sum(np.abs(dst_field_py - dst_field_llvm))
np.testing.assert_almost_equal(error, 0.0)
+
+
+if __name__ == "__main__":
+ test_jacobi_fixed_field_size_gpu()
--
GitLab