Fine tune atol and rtol in test_fp16_exp in backends/xnnpack/test/ops/test_exp.py

Author: mergennachin

backends/xnnpack/test/ops/test_exp.py

@@ -21,8 +21,8 @@ def __init__(self):
         def forward(self, x):
             return torch.exp(x)
 
-    def run_exp_test(self, inputs):
-        (
+    def get_exp_test(self, inputs):
+        return (
             Tester(self.Exp(), inputs)
             .export()
             .check_count({"torch.ops.aten.exp.default": 1})
@@ -31,13 +31,12 @@ def run_exp_test(self, inputs):
             .check_not(["executorch_exir_dialects_edge__ops_aten_exp_default"])
             .to_executorch()
             .serialize()
-            .run_method_and_compare_outputs()
         )
 
     def test_fp16_exp(self):
         inputs = (torch.randn(20).to(torch.float16),)
-        self.run_exp_test(inputs)
+        self.get_exp_test(inputs).run_method_and_compare_outputs(atol=0.01, rtol=0.001)
 
     def test_fp32_exp(self):
         inputs = (torch.randn(20),)
-        self.run_exp_test(inputs)
+        self.get_exp_test(inputs).run_method_and_compare_outputs()