Add elementwise python e2e tests

Change-Id: I8770a509a395145fae78cbcc93320279a4c25de9
Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>

Author: catcor01

projects/pt1/python/torch_mlir_e2e_test/test_suite/elementwise.py

@@ -4899,6 +4899,33 @@ def ElementwiseBitwiseAndModule_basic(module, tu: TestUtils):
 # ==============================================================================
 
 
+class ElementwiseBitwiseAndBoolModule(torch.nn.Module):
+    def __init__(self):
+        super().__init__()
+
+    @export
+    @annotate_args(
+        [
+            None,
+            ([-1, -1], torch.bool, True),
+            ([-1, -1], torch.bool, True),
+        ]
+    )
+    def forward(self, x, y):
+        return torch.bitwise_and(x, y)
+
+
+@register_test_case(module_factory=lambda: ElementwiseBitwiseAndBoolModule())
+def ElementwiseBitwiseAndBoolModule_basic(module, tu: TestUtils):
+    module.forward(
+        tu.randint(3, 4, low=0, high=2).to(torch.bool),
+        tu.randint(3, 4, low=0, high=2).to(torch.bool),
+    )
+
+
+# ==============================================================================
+
+
 class ElementwiseBitwiseAndStaticShapeModule(torch.nn.Module):
     def __init__(self):
         super().__init__()
@@ -4953,6 +4980,33 @@ def ElementwiseBitwiseOrModule_basic(module, tu: TestUtils):
 # ==============================================================================
 
 
+class ElementwiseBitwiseOrBoolModule(torch.nn.Module):
+    def __init__(self):
+        super().__init__()
+
+    @export
+    @annotate_args(
+        [
+            None,
+            ([-1, -1], torch.bool, True),
+            ([-1, -1], torch.bool, True),
+        ]
+    )
+    def forward(self, x, y):
+        return torch.bitwise_or(x, y)
+
+
+@register_test_case(module_factory=lambda: ElementwiseBitwiseOrBoolModule())
+def ElementwiseBitwiseOrBoolModule_basic(module, tu: TestUtils):
+    module.forward(
+        tu.randint(3, 4, low=0, high=2).to(torch.bool),
+        tu.randint(3, 4, low=0, high=2).to(torch.bool),
+    )
+
+
+# ==============================================================================
+
+
 class ElementwiseBitwiseOrStaticShapeModule(torch.nn.Module):
     def __init__(self):
         super().__init__()
@@ -5102,6 +5156,33 @@ def ElementwiseBitwiseXorModule_basic(module, tu: TestUtils):
 # ==============================================================================
 
 
+class ElementwiseBitwiseXorBoolModule(torch.nn.Module):
+    def __init__(self):
+        super().__init__()
+
+    @export
+    @annotate_args(
+        [
+            None,
+            ([-1, -1], torch.bool, True),
+            ([-1, -1], torch.bool, True),
+        ]
+    )
+    def forward(self, x, y):
+        return torch.bitwise_xor(x, y)
+
+
+@register_test_case(module_factory=lambda: ElementwiseBitwiseXorBoolModule())
+def ElementwiseBitwiseXorBoolModule_basic(module, tu: TestUtils):
+    module.forward(
+        tu.randint(3, 4, low=0, high=2).to(torch.bool),
+        tu.randint(3, 4, low=0, high=2).to(torch.bool),
+    )
+
+
+# ==============================================================================
+
+
 class ElementwiseBitwiseXorStaticShapeModule(torch.nn.Module):
     def __init__(self):
         super().__init__()