Function test_float_capture

python/tests/kernel/test_kernel_float.py:35–46

Test that we can capture float lists inside kernel functions.

()

def test_float_capture():
    """Test that we can capture float lists inside kernel functions."""

    f = [1., 2]

    # Capture a list of float
    @cudaq.kernel
    def float_vec_capture(i: int) -> float:
        return f[i]

    for i in range(len(f)):
        assert is_close(f[i], float_vec_capture(i))


def test_float_definition():