/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/utility/reduction_common.hpp Source File

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/utility/reduction_common.hpp Source File#

Composable Kernel: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/utility/reduction_common.hpp Source File
reduction_common.hpp
Go to the documentation of this file.
1 // SPDX-License-Identifier: MIT
2 // Copyright (c) 2018-2023, Advanced Micro Devices, Inc. All rights reserved.
3 
4 #pragma once
5 
6 #include "ck/utility/reduction_enums.hpp"
7 
8 namespace ck {
9 
10 struct float_equal_one
11 {
12  template <class T>
13  __host__ __device__ inline bool operator()(T x)
14  {
15  return x <= static_cast<T>(1.0f) and x >= static_cast<T>(1.0f);
16  };
17 };
18 
19 struct float_equal_zero
20 {
21  template <class T>
22  __host__ __device__ inline bool operator()(T x)
23  {
24  return x <= static_cast<T>(0.0f) and x >= static_cast<T>(0.0f);
25  };
26 };
27 
28 } // namespace ck
ck
Definition: ck.hpp:267
reduction_enums.hpp
ck::float_equal_one
Definition: reduction_common.hpp:11
ck::float_equal_one::operator()
__host__ __device__ bool operator()(T x)
Definition: reduction_common.hpp:13
ck::float_equal_zero
Definition: reduction_common.hpp:20
ck::float_equal_zero::operator()
__host__ __device__ bool operator()(T x)
Definition: reduction_common.hpp:22