rms_norm : relax tolerances

Lightning-AI · Nov 4, 2024 · 61ed8b1 · 61ed8b1
1 parent 6341444
commit 61ed8b1
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/thunder/tests/opinfos.py b/thunder/tests/opinfos.py
@@ -7692,6 +7692,12 @@ def rms_norm_error_generator(op, device, **kwargs):
                 dtypes=(datatypes.float16,),
                 devicetypes=(devices.DeviceType.CPU,),
             ),
+            # See issue - https://github.com/Lightning-AI/lightning-thunder/issues/1395
+            DecorateInfo(
+                custom_comparator(partial(assert_close, atol=2e-3, rtol=2e-3)),
+                dtypes=(datatypes.float16),
+                devicetypes=(devices.DeviceType.CUDA),
+            ),
         ),
     )
     nn_ops.append(rms_norm_opinfo)