Skip to content

GitLab

Commit 734c00a6 authored by Graylatzhou's avatar Graylatzhou
Browse files

issue/183 添加fp32 tolerance

parent 46ee5b45
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
test/infiniop/mul.py
View file @ 734c00a6
...@@ -62,9 +62,11 @@ _TENSOR_DTYPES = [torch.float16, torch.float32] ...@@ -62,9 +62,11 @@ _TENSOR_DTYPES = [torch.float16, torch.float32]
# Tolerance map for different data types # Tolerance map for different data types
_TOLERANCE_MAP = { _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 DEBUG = False
PROFILE = False PROFILE = False
NUM_PRERUN = 10 NUM_PRERUN = 10
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment