derek.holloway
7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
221 lines
14 KiB
C++
221 lines
14 KiB
C++
//===----------------------------------------------------------------------===//
|
|
//
|
|
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
// See https://llvm.org/LICENSE.txt for license information.
|
|
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
// SPDX-FileCopyrightText: Copyright (c) 2024 NVIDIA CORPORATION & AFFILIATES.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
// <cmath>
|
|
|
|
#include <cuda/std/cassert>
|
|
#include <cuda/std/cmath>
|
|
#include <cuda/std/limits>
|
|
#include <cuda/std/type_traits>
|
|
|
|
#include "fp_compare.h"
|
|
#include "test_macros.h"
|
|
|
|
__host__ __device__ void test_isgreater(float val)
|
|
{
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreater((float) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreater((float) 0, (double) 0)), bool>), "");
|
|
#if _CCCL_HAS_LONG_DOUBLE()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreater((float) 0, (long double) 0)), bool>), "");
|
|
#endif // _CCCL_HAS_LONG_DOUBLE()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreater((double) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreater((double) 0, (double) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreater(0, (double) 0)), bool>), "");
|
|
#if _CCCL_HAS_LONG_DOUBLE()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreater((double) 0, (long double) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreater((long double) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreater((long double) 0, (double) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreater((long double) 0, (long double) 0)), bool>), "");
|
|
#endif // _CCCL_HAS_LONG_DOUBLE()
|
|
#if _LIBCUDACXX_HAS_NVFP16()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreater((__half) 0, (__half) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreater((__half) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreater((__half) 0, (double) 0)), bool>), "");
|
|
#endif // _LIBCUDACXX_HAS_NVFP16()
|
|
#if _LIBCUDACXX_HAS_NVBF16()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreater((__nv_bfloat16) 0, (__nv_bfloat16) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreater((__nv_bfloat16) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreater((__nv_bfloat16) 0, (double) 0)), bool>), "");
|
|
#endif // _LIBCUDACXX_HAS_NVBF16()
|
|
assert(cuda::std::isgreater(-1.0, 0.F) == false);
|
|
}
|
|
|
|
__host__ __device__ void test_isgreaterequal(float val)
|
|
{
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreaterequal((float) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreaterequal((float) 0, (double) 0)), bool>), "");
|
|
#if _CCCL_HAS_LONG_DOUBLE()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreaterequal((float) 0, (long double) 0)), bool>), "");
|
|
#endif // _CCCL_HAS_LONG_DOUBLE()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreaterequal((double) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreaterequal((double) 0, (double) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreaterequal(0, (double) 0)), bool>), "");
|
|
#if _CCCL_HAS_LONG_DOUBLE()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreaterequal((double) 0, (long double) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreaterequal((long double) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreaterequal((long double) 0, (double) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreaterequal((long double) 0, (long double) 0)), bool>),
|
|
"");
|
|
#endif // _CCCL_HAS_LONG_DOUBLE()
|
|
#if _LIBCUDACXX_HAS_NVFP16()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreaterequal((__half) 0, (__half) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreaterequal((__half) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreaterequal((__half) 0, (double) 0)), bool>), "");
|
|
#endif // _LIBCUDACXX_HAS_NVFP16()
|
|
#if _LIBCUDACXX_HAS_NVBF16()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreaterequal((__nv_bfloat16) 0, (__nv_bfloat16) 0)), bool>),
|
|
"");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreaterequal((__nv_bfloat16) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isgreaterequal((__nv_bfloat16) 0, (double) 0)), bool>), "");
|
|
#endif // _LIBCUDACXX_HAS_NVBF16()
|
|
assert(cuda::std::isgreaterequal(-1.0, 0.F) == false);
|
|
}
|
|
|
|
__host__ __device__ void test_isless(float val)
|
|
{
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isless((float) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isless((float) 0, (double) 0)), bool>), "");
|
|
#if _CCCL_HAS_LONG_DOUBLE()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isless((float) 0, (long double) 0)), bool>), "");
|
|
#endif // _CCCL_HAS_LONG_DOUBLE()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isless((double) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isless((double) 0, (double) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isless(0, (double) 0)), bool>), "");
|
|
#if _CCCL_HAS_LONG_DOUBLE()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isless((double) 0, (long double) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isless((long double) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isless((long double) 0, (double) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isless((long double) 0, (long double) 0)), bool>), "");
|
|
#endif // _CCCL_HAS_LONG_DOUBLE()
|
|
#if _LIBCUDACXX_HAS_NVFP16()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isless((__half) 0, (__half) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isless((__half) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isless((__half) 0, (double) 0)), bool>), "");
|
|
#endif // _LIBCUDACXX_HAS_NVFP16()
|
|
#if _LIBCUDACXX_HAS_NVBF16()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isless((__nv_bfloat16) 0, (__nv_bfloat16) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isless((__nv_bfloat16) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isless((__nv_bfloat16) 0, (double) 0)), bool>), "");
|
|
#endif // _LIBCUDACXX_HAS_NVBF16()
|
|
assert(cuda::std::isless(-1.0, 0.F) == true);
|
|
}
|
|
|
|
__host__ __device__ void test_islessequal(float val)
|
|
{
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessequal((float) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessequal((float) 0, (double) 0)), bool>), "");
|
|
#if _CCCL_HAS_LONG_DOUBLE()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessequal((float) 0, (long double) 0)), bool>), "");
|
|
#endif // _CCCL_HAS_LONG_DOUBLE()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessequal((double) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessequal((double) 0, (double) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessequal(0, (double) 0)), bool>), "");
|
|
#if _CCCL_HAS_LONG_DOUBLE()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessequal((double) 0, (long double) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessequal((long double) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessequal((long double) 0, (double) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessequal((long double) 0, (long double) 0)), bool>), "");
|
|
#endif // _CCCL_HAS_LONG_DOUBLE()
|
|
#if _LIBCUDACXX_HAS_NVFP16()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessequal((__half) 0, (__half) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessequal((__half) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessequal((__half) 0, (double) 0)), bool>), "");
|
|
#endif // _LIBCUDACXX_HAS_NVFP16()
|
|
#if _LIBCUDACXX_HAS_NVBF16()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessequal((__nv_bfloat16) 0, (__nv_bfloat16) 0)), bool>),
|
|
"");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessequal((__nv_bfloat16) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessequal((__nv_bfloat16) 0, (double) 0)), bool>), "");
|
|
#endif // _LIBCUDACXX_HAS_NVBF16()
|
|
assert(cuda::std::islessequal(-1.0, 0.F) == true);
|
|
}
|
|
|
|
__host__ __device__ void test_islessgreater(float val)
|
|
{
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessgreater((float) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessgreater((float) 0, (double) 0)), bool>), "");
|
|
#if _CCCL_HAS_LONG_DOUBLE()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessgreater((float) 0, (long double) 0)), bool>), "");
|
|
#endif // _CCCL_HAS_LONG_DOUBLE()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessgreater((double) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessgreater((double) 0, (double) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessgreater(0, (double) 0)), bool>), "");
|
|
#if _CCCL_HAS_LONG_DOUBLE()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessgreater((double) 0, (long double) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessgreater((long double) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessgreater((long double) 0, (double) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessgreater((long double) 0, (long double) 0)), bool>), "");
|
|
#endif // _CCCL_HAS_LONG_DOUBLE()
|
|
#if _LIBCUDACXX_HAS_NVFP16()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessgreater((__half) 0, (__half) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessgreater((__half) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessgreater((__half) 0, (double) 0)), bool>), "");
|
|
#endif // _LIBCUDACXX_HAS_NVFP16()
|
|
#if _LIBCUDACXX_HAS_NVBF16()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessgreater((__nv_bfloat16) 0, (__nv_bfloat16) 0)), bool>),
|
|
"");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessgreater((__nv_bfloat16) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::islessgreater((__nv_bfloat16) 0, (double) 0)), bool>), "");
|
|
#endif // _LIBCUDACXX_HAS_NVBF16()
|
|
assert(cuda::std::islessgreater(-1.0, 0.F) == true);
|
|
}
|
|
|
|
__host__ __device__ void test_isunordered(float val)
|
|
{
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isunordered((float) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isunordered((float) 0, (double) 0)), bool>), "");
|
|
#if _CCCL_HAS_LONG_DOUBLE()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isunordered((float) 0, (long double) 0)), bool>), "");
|
|
#endif // _CCCL_HAS_LONG_DOUBLE()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isunordered((double) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isunordered((double) 0, (double) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isunordered(0, (double) 0)), bool>), "");
|
|
#if _CCCL_HAS_LONG_DOUBLE()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isunordered((double) 0, (long double) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isunordered((long double) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isunordered((long double) 0, (double) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isunordered((long double) 0, (long double) 0)), bool>), "");
|
|
#endif // _CCCL_HAS_LONG_DOUBLE()
|
|
#if _LIBCUDACXX_HAS_NVFP16()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isunordered((__half) 0, (__half) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isunordered((__half) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isunordered((__half) 0, (double) 0)), bool>), "");
|
|
#endif // _LIBCUDACXX_HAS_NVFP16()
|
|
#if _LIBCUDACXX_HAS_NVBF16()
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isunordered((__nv_bfloat16) 0, (__nv_bfloat16) 0)), bool>),
|
|
"");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isunordered((__nv_bfloat16) 0, (float) 0)), bool>), "");
|
|
static_assert((cuda::std::is_same_v<decltype(cuda::std::isunordered((__nv_bfloat16) 0, (double) 0)), bool>), "");
|
|
#endif // _LIBCUDACXX_HAS_NVBF16()
|
|
assert(cuda::std::isunordered(-1.0, 0.F) == false);
|
|
}
|
|
|
|
__host__ __device__ void test(float val)
|
|
{
|
|
test_isgreater(val);
|
|
test_isgreaterequal(val);
|
|
test_isless(val);
|
|
test_islessequal(val);
|
|
test_islessgreater(val);
|
|
test_isunordered(val);
|
|
}
|
|
|
|
__global__ void test_global_kernel(float* val)
|
|
{
|
|
test(*val);
|
|
}
|
|
|
|
int main(int, char**)
|
|
{
|
|
volatile float val = 1.0f;
|
|
test(val);
|
|
|
|
return 0;
|
|
}
|