update atol and rtol for cuda

DeepLink-org · Apr 15, 2024 · 7db8ae1 · 7db8ae1
1 parent 03906e9
commit 7db8ae1
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/diopi_test/python/configs/diopi_configs.py b/diopi_test/python/configs/diopi_configs.py
@@ -8307,6 +8307,8 @@
 
     'rms_norm_default': dict(
         name=['rms_norm'],
+        atol=1e-4,
+        rtol=1e-4,
         interface=['CustomizedTest'],
         dtype=[np.float16, np.float32, np.float64],
         para=dict(
@@ -8337,6 +8339,8 @@
 
     'rms_norm_with_normalized_shape': dict(
         name=['rms_norm'],
+        atol=1e-4,
+        rtol=1e-4,
         interface=['CustomizedTest'],
         dtype=[np.float16, np.float32, np.float64],
         para=dict(
@@ -8367,6 +8371,8 @@
 
     'rms_norm_with_bias': dict(
         name=['rms_norm'],
+        atol=1e-4,
+        rtol=1e-4,
         interface=['CustomizedTest'],
         dtype=[np.float16, np.float32, np.float64],
         para=dict(