From 7db8ae166844b0c39cef67f4428ffe6d4257da03 Mon Sep 17 00:00:00 2001
From: YPOI-WX <wangxing2@sensetime.com>
Date: Mon, 15 Apr 2024 18:09:01 +0800
Subject: [PATCH] update atol and rtol for cuda

---
 diopi_test/python/configs/diopi_configs.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/diopi_test/python/configs/diopi_configs.py b/diopi_test/python/configs/diopi_configs.py
index 23fc44124..927e9d7aa 100755
--- 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(