Tests failing for exp op in xnnpack

Differential Revision: D76283309

Pull Request resolved: #11498

Author: leafs1

backends/xnnpack/partition/config/__init__.py

@@ -26,7 +26,7 @@
     DeQuantizedPerTensorConfig,
     DivConfig,
     # EluConfig,
-    ExpConfig,
+    # ExpConfig, # Bug in XNNPACK with exp op
     FloorConfig,
     GeluConfig,
     HardswishConfig,
@@ -81,7 +81,7 @@
     ClampConfig,
     DivConfig,
     # EluConfig, # Waiting for PyTorch Pin Update
-    ExpConfig,
+    # ExpConfig, # Bug in XNNPACK with exp op
     FloorConfig,
     GeluConfig,
     HardtanhConfig,

backends/xnnpack/partition/configs.py

@@ -59,7 +59,7 @@
     exir_ops.edge.aten.sigmoid.default,
     exir_ops.edge.aten._softmax.default,
     exir_ops.edge.aten.cat.default,
-    exir_ops.edge.aten.exp.default,
+    # exir_ops.edge.aten.exp.default, # Bug in XNNPACK with exp op
     exir_ops.edge.aten.elu.default,
     exir_ops.edge.aten.avg_pool2d.default,
     exir_ops.edge.aten.leaky_relu.default,

backends/xnnpack/test/ops/test_exp.py

@@ -34,10 +34,12 @@ def run_exp_test(self, inputs):
             .run_method_and_compare_outputs()
         )
 
+    @unittest.skip("D76283309, bug to fix")
     def test_fp16_exp(self):
         inputs = (torch.randn(20).to(torch.float16),)
         self.run_exp_test(inputs)
 
+    @unittest.skip("D76283309, bug to fix")
     def test_fp32_exp(self):
         inputs = (torch.randn(20),)
         self.run_exp_test(inputs)