Function test_float32_use

python/tests/kernel/test_kernel_float.py:208–249

Test that we can use floats inside kernel functions.

()

def test_float32_use():
    """Test that we can use floats inside kernel functions."""

    # Use a float inside np in a kernel (sin)
    @cudaq.kernel
    def float_np_use() -> np.float32:
        return np.sin(np.float32(np.pi / 2 + 1))

    t = np.sin(np.float32(np.pi / 2 + 1))
    assert is_close(t, float_np_use())

    # Use a float inside np in a kernel (cos)
    @cudaq.kernel
    def float_np_use() -> np.float32:
        return np.cos(np.float32(np.pi / 2 + 1))

    t = np.cos(np.float32(np.pi / 2 + 1))
    assert is_close(t, float_np_use())

    # Use a float inside np in a kernel (sqrt)
    @cudaq.kernel
    def float_np_use() -> np.float32:
        return np.sqrt(np.float32(np.pi / 2 + 1))

    t = np.sqrt(np.float32(np.pi / 2 + 1))
    assert is_close(t, float_np_use())

    # Use a float inside np in a kernel (ceil)
    @cudaq.kernel
    def float_np_use() -> np.float32:
        return np.ceil(np.float32(np.pi / 2 + 1))

    t = np.ceil(np.float32(np.pi / 2 + 1))
    assert is_close(t, float_np_use())

    # Use a float inside np in a kernel (exp)
    @cudaq.kernel
    def float_np_use() -> np.float32:
        return np.exp(np.float32(np.pi / 2 + 1))

    t = np.exp(np.float32(np.pi / 2 + 1))
    assert is_close(t, float_np_use())


def test_float_list_parameter_promotion():