Add gpu vs cpu results comparison

DEEPDIP-project · Jun 19, 2024 · 9ab56d2 · 9ab56d2
1 parent 0e9cb49
commit 9ab56d2
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/examples/src/test_derivatives.jl b/examples/src/test_derivatives.jl
@@ -414,6 +414,12 @@ else
     println("[GPU] ERROR: Some methods produce different results")
 end
 
+if all(results[1] ≈ Array(result) for result in results_GPU[2:end])
+    println("[GPU vs CPU] All methods produce the same results")
+else
+    println("[GPU vs CPU] ERROR: Some methods produce different results")
+end
+
 # sort the results by time
 methods_gpu = []
 times_gpu = []