From 8deb5810051bfb050427d442d77383937b89b7b2 Mon Sep 17 00:00:00 2001
From: Liu Yiqun <liuyiqun01@baidu.com>
Date: Thu, 25 Sep 2025 13:59:32 +0800
Subject: [PATCH] Implement a common function to calculate the tolerance
 configs.

---
 graph_net/test_compiler_util.py | 39 +++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 graph_net/test_compiler_util.py

diff --git a/graph_net/test_compiler_util.py b/graph_net/test_compiler_util.py
new file mode 100644
index 000000000..a8520b6b2
--- /dev/null
+++ b/graph_net/test_compiler_util.py
@@ -0,0 +1,39 @@
+import os
+
+
+def tolerance_generator(t):
+    # for float16
+    yield 10 ** (t * 3 / 5), 10**t
+    # for bfloat16
+    yield 10 ** (t * 1.796 / 5), 10**t
+    # yield float32
+    yield 10 ** (t * 5.886 / 5), 10**t
+    # yield float64
+    yield 10 ** (t * 7 / 5), 10 ** (t * 7 / 5)
+
+
+def calculate_tolerance_pair(begin, end):
+    tolerance_pair_list = []
+    for t in range(begin, end + 1):
+        for rtol, atol in tolerance_generator(t):
+            effective_atol = float(f"{atol:.3g}")
+            effective_rtol = float(f"{rtol:.3g}")
+            tolerance_pair_list.append(
+                {
+                    "atol": effective_atol,
+                    "rtol": effective_rtol,
+                }
+            )
+    return tolerance_pair_list
+
+
+def generate_allclose_configs(cmp_all_close_func):
+    tolerance_pair_list = calculate_tolerance_pair(-10, 5)
+
+    cmp_configs = []
+    for pair in tolerance_pair_list:
+        atol, rtol = pair["atol"], pair["rtol"]
+        cmp_configs.append(
+            (f"[all_close_atol_{atol:.2E}_rtol_{rtol:.2E}]", cmp_all_close_func, pair)
+        )
+    return cmp_configs