Function test_complex_capture

python/tests/kernel/test_kernel_complex.py:53–73

Test that we can pass complex lists to kernel functions.

()

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

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

    @cudaq.kernel
    def complex_vec_capture_real(i: int) -> float:
        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) -> float:
        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_complex_definition():