changing GPU tol

CEMeNT-PSAAP · May 15, 2024 · ac759b7 · ac759b7
1 parent ba2f569
commit ac759b7
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/test/regression/run.py b/test/regression/run.py
@@ -62,6 +62,11 @@
             print(
                 Fore.YELLOW + "Note: Skipping %s (GPU target)" % name + Style.RESET_ALL
             )
+    tol = 1e-9
+    if (target == "gpu"):
+        tol = 1e-6
+        Fore.CYAN
+        print(Fore.YELLOW + 'Note: tolerance increased to ', tol, '(GPU target)')
 names = temp
 
 # Data for each test
@@ -138,7 +143,7 @@
             b = answer[result_name][:]
 
             # Passed?
-            if np.isclose(a, b).all():
+            if np.isclose(a, b, rtol=tol).all():
                 print(
                     Fore.GREEN
                     + "  {}: Passed".format(score + "/" + result)