Function test_float64_use

python/tests/kernel/test_kernel_float.py:119–160

Test that we can use floats inside kernel functions.

()

def test_float64_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.float64:
        return np.sin(np.float64(np.pi / 2 + 1))

    t = np.sin(np.float64(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.float64:
        return np.cos(np.float64(np.pi / 2 + 1))

    t = np.cos(np.float64(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.float64:
        return np.sqrt(np.float64(np.pi / 2 + 1))

    t = np.sqrt(np.float64(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.float64:
        return np.ceil(np.float64(np.pi / 2 + 1))

    t = np.ceil(np.float64(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.float64:
        return np.exp(np.float64(np.pi / 2 + 1))

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


# np.float32