fix make_fx_tosa and format

llvm · Apr 28, 2024 · dda24af · dda24af
1 parent 69d4cbb
commit dda24af
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/projects/pt1/e2e_testing/xfail_sets.py b/projects/pt1/e2e_testing/xfail_sets.py
@@ -1937,6 +1937,7 @@
     # failed to legalize operation 'torch.operator'
     "ElementwisePreluModule_basic",
     "ElementwisePreluStaticModule_basic",
+    "ElementwiseLogSigmoidModule_basic",
     # Shape Related failures
     "PrimListUnpackNumMismatchModule_basic",
     "ReshapeExpandModule_basic",

diff --git a/projects/pt1/python/torch_mlir_e2e_test/test_suite/elementwise.py b/projects/pt1/python/torch_mlir_e2e_test/test_suite/elementwise.py
@@ -2110,16 +2110,17 @@ def ElementwiseLogitModule_basic(module, tu: TestUtils):
 
 
 class ElementwiseLogSigmoidModule(torch.nn.Module):
-
     def __init__(self):
         super().__init__()
         self.m = torch.nn.LogSigmoid()
 
     @export
-    @annotate_args([
-        None,
-        ([-1, -1], torch.float32, True),
-    ])
+    @annotate_args(
+        [
+            None,
+            ([-1, -1], torch.float32, True),
+        ]
+    )
     def forward(self, a):
         return self.m(a)