Adds tests for in-place bitwise elementwise funcs

Author: ndgrigorian

dpctl/tests/elementwise/test_bitwise_and.py

@@ -18,6 +18,7 @@
 import pytest
 
 import dpctl.tensor as dpt
+from dpctl.tensor._type_utils import _can_cast
 from dpctl.tests.helper import get_queue_or_skip, skip_if_dtype_not_supported
 
 from .utils import _integral_dtypes
@@ -85,3 +86,58 @@ def test_bitwise_and_bool():
     r_lo = dpt.logical_and(x1[:, dpt.newaxis], x2[dpt.newaxis])
 
     assert dpt.all(dpt.equal(r_bw, r_lo))
+
+
+@pytest.mark.parametrize("dtype", ["?"] + _integral_dtypes)
+def test_bitwise_and_inplace_python_scalar(dtype):
+    q = get_queue_or_skip()
+    skip_if_dtype_not_supported(dtype, q)
+    X = dpt.zeros((10, 10), dtype=dtype, sycl_queue=q)
+    dt_kind = X.dtype.kind
+    if dt_kind == "b":
+        X &= False
+    else:
+        X &= int(0)
+
+
+@pytest.mark.parametrize("op1_dtype", ["?"] + _integral_dtypes)
+@pytest.mark.parametrize("op2_dtype", ["?"] + _integral_dtypes)
+def test_bitwise_and_inplace_dtype_matrix(op1_dtype, op2_dtype):
+    q = get_queue_or_skip()
+    skip_if_dtype_not_supported(op1_dtype, q)
+    skip_if_dtype_not_supported(op2_dtype, q)
+
+    sz = 127
+    ar1 = dpt.ones(sz, dtype=op1_dtype, sycl_queue=q)
+    ar2 = dpt.ones_like(ar1, dtype=op2_dtype, sycl_queue=q)
+
+    dev = q.sycl_device
+    _fp16 = dev.has_aspect_fp16
+    _fp64 = dev.has_aspect_fp64
+    if _can_cast(ar2.dtype, ar1.dtype, _fp16, _fp64):
+        ar1 &= ar2
+        assert dpt.all(ar1 == 1)
+
+        ar3 = dpt.ones(sz, dtype=op1_dtype, sycl_queue=q)[::-1]
+        ar4 = dpt.ones(2 * sz, dtype=op2_dtype, sycl_queue=q)[::2]
+        ar3 &= ar4
+        assert dpt.all(ar3 == 1)
+    else:
+        with pytest.raises(TypeError):
+            ar1 &= ar2
+            dpt.bitwise_and(ar1, ar2, out=ar1)
+
+    # out is second arg
+    ar1 = dpt.ones(sz, dtype=op1_dtype, sycl_queue=q)
+    ar2 = dpt.ones_like(ar1, dtype=op2_dtype, sycl_queue=q)
+    if _can_cast(ar1.dtype, ar2.dtype, _fp16, _fp64):
+        dpt.bitwise_and(ar1, ar2, out=ar2)
+        assert dpt.all(ar2 == 1)
+
+        ar3 = dpt.ones(sz, dtype=op1_dtype, sycl_queue=q)[::-1]
+        ar4 = dpt.ones(2 * sz, dtype=op2_dtype, sycl_queue=q)[::2]
+        dpt.bitwise_and(ar3, ar4, out=ar4)
+        dpt.all(ar4 == 1)
+    else:
+        with pytest.raises(TypeError):
+            dpt.bitwise_and(ar1, ar2, out=ar2)

dpctl/tests/elementwise/test_bitwise_left_shift.py

@@ -18,6 +18,7 @@
 import pytest
 
 import dpctl.tensor as dpt
+from dpctl.tensor._type_utils import _can_cast
 from dpctl.tests.helper import get_queue_or_skip, skip_if_dtype_not_supported
 
 from .utils import _integral_dtypes
@@ -97,3 +98,54 @@ def test_bitwise_left_shift_range(op_dtype):
 
     z = dpt.bitwise_left_shift(x, y)
     assert dpt.all(dpt.equal(z, 0))
+
+
+@pytest.mark.parametrize("dtype", _integral_dtypes)
+def test_bitwise_left_shift_inplace_python_scalar(dtype):
+    q = get_queue_or_skip()
+    skip_if_dtype_not_supported(dtype, q)
+    X = dpt.zeros((10, 10), dtype=dtype, sycl_queue=q)
+    X <<= int(0)
+
+
+@pytest.mark.parametrize("op1_dtype", _integral_dtypes)
+@pytest.mark.parametrize("op2_dtype", _integral_dtypes)
+def test_bitwise_left_shift_inplace_dtype_matrix(op1_dtype, op2_dtype):
+    q = get_queue_or_skip()
+    skip_if_dtype_not_supported(op1_dtype, q)
+    skip_if_dtype_not_supported(op2_dtype, q)
+
+    sz = 127
+    ar1 = dpt.ones(sz, dtype=op1_dtype, sycl_queue=q)
+    ar2 = dpt.ones_like(ar1, dtype=op2_dtype, sycl_queue=q)
+
+    dev = q.sycl_device
+    _fp16 = dev.has_aspect_fp16
+    _fp64 = dev.has_aspect_fp64
+    if _can_cast(ar2.dtype, ar1.dtype, _fp16, _fp64):
+        ar1 <<= ar2
+        assert dpt.all(ar1 == 2)
+
+        ar3 = dpt.ones(sz, dtype=op1_dtype, sycl_queue=q)[::-1]
+        ar4 = dpt.ones(2 * sz, dtype=op2_dtype, sycl_queue=q)[::2]
+        ar3 <<= ar4
+        assert dpt.all(ar3 == 2)
+    else:
+        with pytest.raises(TypeError):
+            ar1 <<= ar2
+            dpt.bitwise_left_shift(ar1, ar2, out=ar1)
+
+    # out is second arg
+    ar1 = dpt.ones(sz, dtype=op1_dtype, sycl_queue=q)
+    ar2 = dpt.ones_like(ar1, dtype=op2_dtype, sycl_queue=q)
+    if _can_cast(ar1.dtype, ar2.dtype, _fp16, _fp64):
+        dpt.bitwise_left_shift(ar1, ar2, out=ar2)
+        assert dpt.all(ar2 == 2)
+
+        ar3 = dpt.ones(sz, dtype=op1_dtype, sycl_queue=q)[::-1]
+        ar4 = dpt.ones(2 * sz, dtype=op2_dtype, sycl_queue=q)[::2]
+        dpt.bitwise_left_shift(ar3, ar4, out=ar4)
+        dpt.all(ar4 == 2)
+    else:
+        with pytest.raises(TypeError):
+            dpt.bitwise_left_shift(ar1, ar2, out=ar2)

dpctl/tests/elementwise/test_bitwise_or.py

@@ -18,6 +18,7 @@
 import pytest
 
 import dpctl.tensor as dpt
+from dpctl.tensor._type_utils import _can_cast
 from dpctl.tests.helper import get_queue_or_skip, skip_if_dtype_not_supported
 
 from .utils import _integral_dtypes
@@ -85,3 +86,58 @@ def test_bitwise_or_bool():
     r_lo = dpt.logical_or(x1[:, dpt.newaxis], x2[dpt.newaxis])
 
     assert dpt.all(dpt.equal(r_bw, r_lo))
+
+
+@pytest.mark.parametrize("dtype", ["?"] + _integral_dtypes)
+def test_bitwise_or_inplace_python_scalar(dtype):
+    q = get_queue_or_skip()
+    skip_if_dtype_not_supported(dtype, q)
+    X = dpt.zeros((10, 10), dtype=dtype, sycl_queue=q)
+    dt_kind = X.dtype.kind
+    if dt_kind == "b":
+        X |= False
+    else:
+        X |= int(0)
+
+
+@pytest.mark.parametrize("op1_dtype", ["?"] + _integral_dtypes)
+@pytest.mark.parametrize("op2_dtype", ["?"] + _integral_dtypes)
+def test_bitwise_or_inplace_dtype_matrix(op1_dtype, op2_dtype):
+    q = get_queue_or_skip()
+    skip_if_dtype_not_supported(op1_dtype, q)
+    skip_if_dtype_not_supported(op2_dtype, q)
+
+    sz = 127
+    ar1 = dpt.ones(sz, dtype=op1_dtype, sycl_queue=q)
+    ar2 = dpt.ones_like(ar1, dtype=op2_dtype, sycl_queue=q)
+
+    dev = q.sycl_device
+    _fp16 = dev.has_aspect_fp16
+    _fp64 = dev.has_aspect_fp64
+    if _can_cast(ar2.dtype, ar1.dtype, _fp16, _fp64):
+        ar1 |= ar2
+        assert dpt.all(ar1 == 1)
+
+        ar3 = dpt.ones(sz, dtype=op1_dtype, sycl_queue=q)[::-1]
+        ar4 = dpt.ones(2 * sz, dtype=op2_dtype, sycl_queue=q)[::2]
+        ar3 |= ar4
+        assert dpt.all(ar3 == 1)
+    else:
+        with pytest.raises(TypeError):
+            ar1 |= ar2
+            dpt.bitwise_or(ar1, ar2, out=ar1)
+
+    # out is second arg
+    ar1 = dpt.ones(sz, dtype=op1_dtype, sycl_queue=q)
+    ar2 = dpt.ones_like(ar1, dtype=op2_dtype, sycl_queue=q)
+    if _can_cast(ar1.dtype, ar2.dtype, _fp16, _fp64):
+        dpt.bitwise_or(ar1, ar2, out=ar2)
+        assert dpt.all(ar2 == 1)
+
+        ar3 = dpt.ones(sz, dtype=op1_dtype, sycl_queue=q)[::-1]
+        ar4 = dpt.ones(2 * sz, dtype=op2_dtype, sycl_queue=q)[::2]
+        dpt.bitwise_or(ar3, ar4, out=ar4)
+        dpt.all(ar4 == 1)
+    else:
+        with pytest.raises(TypeError):
+            dpt.bitwise_or(ar1, ar2, out=ar2)

dpctl/tests/elementwise/test_bitwise_right_shift.py

@@ -18,6 +18,7 @@
 import pytest
 
 import dpctl.tensor as dpt
+from dpctl.tensor._type_utils import _can_cast
 from dpctl.tests.helper import get_queue_or_skip, skip_if_dtype_not_supported
 
 from .utils import _integral_dtypes
@@ -97,3 +98,54 @@ def test_bitwise_right_shift_range(op_dtype):
 
     z = dpt.bitwise_right_shift(x, y)
     assert dpt.all(dpt.equal(z, 0))
+
+
+@pytest.mark.parametrize("dtype", _integral_dtypes)
+def test_bitwise_right_shift_inplace_python_scalar(dtype):
+    q = get_queue_or_skip()
+    skip_if_dtype_not_supported(dtype, q)
+    X = dpt.zeros((10, 10), dtype=dtype, sycl_queue=q)
+    X >>= int(0)
+
+
+@pytest.mark.parametrize("op1_dtype", _integral_dtypes)
+@pytest.mark.parametrize("op2_dtype", _integral_dtypes)
+def test_bitwise_right_shift_inplace_dtype_matrix(op1_dtype, op2_dtype):
+    q = get_queue_or_skip()
+    skip_if_dtype_not_supported(op1_dtype, q)
+    skip_if_dtype_not_supported(op2_dtype, q)
+
+    sz = 127
+    ar1 = dpt.ones(sz, dtype=op1_dtype, sycl_queue=q)
+    ar2 = dpt.ones_like(ar1, dtype=op2_dtype, sycl_queue=q)
+
+    dev = q.sycl_device
+    _fp16 = dev.has_aspect_fp16
+    _fp64 = dev.has_aspect_fp64
+    if _can_cast(ar2.dtype, ar1.dtype, _fp16, _fp64):
+        ar1 >>= ar2
+        assert dpt.all(ar1 == 0)
+
+        ar3 = dpt.ones(sz, dtype=op1_dtype, sycl_queue=q)[::-1]
+        ar4 = dpt.ones(2 * sz, dtype=op2_dtype, sycl_queue=q)[::2]
+        ar3 >>= ar4
+        assert dpt.all(ar3 == 0)
+    else:
+        with pytest.raises(TypeError):
+            ar1 >>= ar2
+            dpt.bitwise_right_shift(ar1, ar2, out=ar1)
+
+    # out is second arg
+    ar1 = dpt.ones(sz, dtype=op1_dtype, sycl_queue=q)
+    ar2 = dpt.ones_like(ar1, dtype=op2_dtype, sycl_queue=q)
+    if _can_cast(ar1.dtype, ar2.dtype, _fp16, _fp64):
+        dpt.bitwise_right_shift(ar1, ar2, out=ar2)
+        assert dpt.all(ar2 == 0)
+
+        ar3 = dpt.ones(sz, dtype=op1_dtype, sycl_queue=q)[::-1]
+        ar4 = dpt.ones(2 * sz, dtype=op2_dtype, sycl_queue=q)[::2]
+        dpt.bitwise_right_shift(ar3, ar4, out=ar4)
+        dpt.all(ar4 == 0)
+    else:
+        with pytest.raises(TypeError):
+            dpt.bitwise_right_shift(ar1, ar2, out=ar2)

dpctl/tests/elementwise/test_bitwise_xor.py

@@ -18,6 +18,7 @@
 import pytest
 
 import dpctl.tensor as dpt
+from dpctl.tensor._type_utils import _can_cast
 from dpctl.tests.helper import get_queue_or_skip, skip_if_dtype_not_supported
 
 from .utils import _integral_dtypes
@@ -85,3 +86,58 @@ def test_bitwise_xor_bool():
     r_lo = dpt.logical_xor(x1[:, dpt.newaxis], x2[dpt.newaxis])
 
     assert dpt.all(dpt.equal(r_bw, r_lo))
+
+
+@pytest.mark.parametrize("dtype", ["?"] + _integral_dtypes)
+def test_bitwise_xor_inplace_python_scalar(dtype):
+    q = get_queue_or_skip()
+    skip_if_dtype_not_supported(dtype, q)
+    X = dpt.zeros((10, 10), dtype=dtype, sycl_queue=q)
+    dt_kind = X.dtype.kind
+    if dt_kind == "b":
+        X ^= False
+    else:
+        X ^= int(0)
+
+
+@pytest.mark.parametrize("op1_dtype", ["?"] + _integral_dtypes)
+@pytest.mark.parametrize("op2_dtype", ["?"] + _integral_dtypes)
+def test_bitwise_xor_inplace_dtype_matrix(op1_dtype, op2_dtype):
+    q = get_queue_or_skip()
+    skip_if_dtype_not_supported(op1_dtype, q)
+    skip_if_dtype_not_supported(op2_dtype, q)
+
+    sz = 127
+    ar1 = dpt.ones(sz, dtype=op1_dtype, sycl_queue=q)
+    ar2 = dpt.ones_like(ar1, dtype=op2_dtype, sycl_queue=q)
+
+    dev = q.sycl_device
+    _fp16 = dev.has_aspect_fp16
+    _fp64 = dev.has_aspect_fp64
+    if _can_cast(ar2.dtype, ar1.dtype, _fp16, _fp64):
+        ar1 ^= ar2
+        assert dpt.all(ar1 == 0)
+
+        ar3 = dpt.ones(sz, dtype=op1_dtype, sycl_queue=q)[::-1]
+        ar4 = dpt.ones(2 * sz, dtype=op2_dtype, sycl_queue=q)[::2]
+        ar3 ^= ar4
+        assert dpt.all(ar3 == 0)
+    else:
+        with pytest.raises(TypeError):
+            ar1 ^= ar2
+            dpt.bitwise_xor(ar1, ar2, out=ar1)
+
+    # out is second arg
+    ar1 = dpt.ones(sz, dtype=op1_dtype, sycl_queue=q)
+    ar2 = dpt.ones_like(ar1, dtype=op2_dtype, sycl_queue=q)
+    if _can_cast(ar1.dtype, ar2.dtype, _fp16, _fp64):
+        dpt.bitwise_xor(ar1, ar2, out=ar2)
+        assert dpt.all(ar2 == 0)
+
+        ar3 = dpt.ones(sz, dtype=op1_dtype, sycl_queue=q)[::-1]
+        ar4 = dpt.ones(2 * sz, dtype=op2_dtype, sycl_queue=q)[::2]
+        dpt.bitwise_xor(ar3, ar4, out=ar4)
+        dpt.all(ar4 == 0)
+    else:
+        with pytest.raises(TypeError):
+            dpt.bitwise_xor(ar1, ar2, out=ar2)