    def test_getitem_boolean_index(self) -> None:
        """Verify that NotImplementatedError is thrown encountering
        tensor[boolean_mask_tensor]
        """

        class TestModule(torch.nn.Module):
            def forward(self, x: torch.Tensor, mask: torch.Tensor) -> torch.Tensor:
                return x[mask]

        mod = TestModule().cuda()
        x = torch.rand(10, 4).half().cuda()
        mask = (torch.rand((10,)) > 0.5).cuda()
        mod(x, mask)

        self.assertRaises(
            NotImplementedError,
            lambda: self.run_test(mod, [x, mask], expected_ops={}),
        )
Verify that NotImplementatedError is thrown encountering
tensor[boolean_mask_tensor]
fx2ait/fx2ait/test/converters/test_ait_binary_op.py