Function test_float32_capture

python/tests/kernel/test_kernel_float.py:180–191

Test that we can capture float lists inside kernel functions.

()

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

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

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

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


def test_float32_definition():