Merge branch 'main' into tom/hf_llama3_test

Lightning-AI · Nov 4, 2024 · ebd1cef · ebd1cef
2 parents 845d887 + cb02f8a
commit ebd1cef
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)