issue/183 添加fp32 tolerance

734c00a6 · Graylatzhou · 46ee5b45 · 734c00a6
Commit 734c00a6 authored Apr 25, 2025 by Graylatzhou
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

test/infiniop/mul.py test/infiniop/mul.py +3 -1

No files found.
--- a/test/infiniop/mul.py
+++ b/test/infiniop/mul.py
@@ -62,9 +62,11 @@ _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
 PROFILE = False
 NUM_PRERUN = 10