Function test_float64_capture

python/tests/kernel/test_kernel_float.py:91–102

Test that we can capture float lists inside kernel functions.

()

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

    f = [np.float64(1.), np.float64(2)]

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

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


def test_float64_definition():