issue/200: updata fp16 and fp32 tolerance for add and swiglu in python tests

7e7528cd · Zimin Li · 11e7df93 · 7e7528cd · 7e7528cd
Commit 7e7528cd authored Apr 25, 2025 by Zimin Li
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

test/infiniop/add.py test/infiniop/add.py +2 -1

test/infiniop/swiglu.py test/infiniop/swiglu.py +2 -1

No files found.
--- a/test/infiniop/add.py
+++ b/test/infiniop/add.py
@@ -62,7 +62,8 @@ _TENSOR_DTYPES = [torch.float16, torch.float32]

 # Tolerance map for different data types
 _TOLERANCE_MAP = {
-    torch.float16: {"atol": 1e-4, "rtol": 1e-2},
+    torch.float16: {"atol": 1e-3, "rtol": 1e-3},
+    torch.float32: {"atol": 1e-7, "rtol": 1e-7},
 }

 DEBUG = False

--- a/test/infiniop/swiglu.py
+++ b/test/infiniop/swiglu.py
@@ -61,7 +61,8 @@ _TENSOR_DTYPES = [torch.float16, torch.float32]

 # Tolerance map for different data types
 _TOLERANCE_MAP = {
-    torch.float16: {"atol": 1e-4, "rtol": 1e-2},
+    torch.float16: {"atol": 1e-3, "rtol": 1e-3},
+    torch.float32: {"atol": 2e-7, "rtol": 1e-7},
 }

 DEBUG = False