Function test_np_complex64_capture

python/tests/kernel/test_kernel_complex.py:444–469

Test that we can pass complex lists to kernel functions.

()

def test_np_complex64_capture():
    """Test that we can pass complex lists to kernel functions."""

    # Capture a list of complex
    c = [
        np.complex64(.70710678 + 1j),
        np.complex64(0. + 2j),
        np.complex64(0.),
        np.complex64(0.70710678)
    ]

    @cudaq.kernel
    def complex_vec_capture_real(i: int) -> np.float32:
        v = c[i]
        return v.real

    for i in range(len(c)):
        assert is_close(c[i].real, complex_vec_capture_real(i))

    @cudaq.kernel
    def complex_vec_capture_imag(i: int) -> np.float32:
        v = c[i]
        return v.imag

    for i in range(len(c)):
        assert is_close(c[i].imag, complex_vec_capture_imag(i))


def test_np_complex64_definition():