Function test_np_complex64_use

python/tests/kernel/test_kernel_complex.py:510–579

Test that we can use complex numbers inside kernel functions.

()

def test_np_complex64_use():
    """Test that we can use complex numbers inside kernel functions."""

    # Use a complex inside np in a kernel (sin)
    @cudaq.kernel
    def complex_np_use_real() -> np.float32:
        v = np.sin(np.complex64((np.pi / 2. + 1) + 1j))
        return v.real

    t = np.sin(np.complex64((np.pi / 2. + 1) + 1j)).real
    assert is_close(t, complex_np_use_real())

    @cudaq.kernel
    def complex_np_use_imag() -> np.float32:
        v = np.sin(np.complex64((np.pi / 2. + 1) + 1j))
        return v.imag

    t = np.sin(np.complex64((np.pi / 2. + 1) + 1j)).imag
    assert is_close(t, complex_np_use_imag())

    # Use a complex inside np in a kernel (cos)
    @cudaq.kernel
    def complex_np_use_real() -> np.float32:
        v = np.cos(np.complex64((np.pi / 2. + 1) + 1j))
        return v.real

    t = np.cos(np.complex64((np.pi / 2. + 1) + 1j)).real
    assert is_close(t, complex_np_use_real())

    @cudaq.kernel
    def complex_np_use_imag() -> np.float32:
        v = np.cos(np.complex64((np.pi / 2. + 1) + 1j))
        return v.imag

    t = np.cos(np.complex64((np.pi / 2. + 1) + 1j)).imag
    assert is_close(t, complex_np_use_imag())

    # Use a complex inside np in a kernel (sqrt)
    @cudaq.kernel
    def complex_np_use_real() -> np.float32:
        v = np.sqrt(np.complex64((np.pi / 2. + 1) + 1j))
        return v.real

    t = np.sqrt(np.complex64((np.pi / 2. + 1) + 1j)).real
    assert is_close(t, complex_np_use_real())

    @cudaq.kernel
    def complex_np_use_imag() -> np.float32:
        v = np.sqrt(np.complex64((np.pi / 2. + 1) + 1j))
        return v.imag

    t = np.sqrt(np.complex64((np.pi / 2. + 1) + 1j)).imag
    assert is_close(t, complex_np_use_imag())

    # Use a complex inside np in a kernel (exp)
    @cudaq.kernel
    def complex_np_use_real() -> np.float32:
        v = np.exp(np.complex64((np.pi / 2. + 1) + 1j))