Function test_float32_definition

python/tests/kernel/test_kernel_float.py:194–205

Test that we can define float lists inside kernel functions.

()

def test_float32_definition():
    """Test that we can define float lists inside kernel functions."""

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

    # Define a list of float inside a kernel
    @cudaq.kernel
    def float_vec_definition(i: int) -> np.float32:
        return [np.float32(1.), np.float32(2)][i]

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


def test_float32_use():