include/ck/utility/reduction_common.hpp Source File

• include
• ck
• utility

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:264

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