atomicMaxF.h File Reference

#include <alpaka/alpaka.hpp>
#include "FWCore/Utilities/interface/bit_cast.h"

Go to the source code of this file.

Functions
template<typename TAcc , typename = std::enable_if_t<alpaka::isAccelerator<TAcc>>>
ALPAKA_FN_HOST_ACC static ALPAKA_FN_INLINE float	atomicMaxF (const TAcc &acc, float *address, float val)

Function Documentation

atomicMaxF()

template<typename TAcc , typename = std::enable_if_t<alpaka::isAccelerator<TAcc>>>

ALPAKA_FN_HOST_ACC static ALPAKA_FN_INLINE float atomicMaxF ( const TAcc &  acc, float *  address, float  val  )

static

Definition at line 11 of file atomicMaxF.h.

References cms::cudacompat::atomicCAS(), edm::bit_cast(), runTheMatrix::ret, and heppy_batch::val.

Referenced by ALPAKA_ACCELERATOR_NAMESPACE::hcalFastCluster_exotic(), ALPAKA_ACCELERATOR_NAMESPACE::hcalFastCluster_multiSeedIterative(), and ALPAKA_ACCELERATOR_NAMESPACE::hcalFastCluster_multiSeedParallel().

                                                                                                        {
#if defined(__CUDA_ARCH__) or defined(__HIP_DEVICE_COMPILE__)
  // GPU implementation uses __float_as_int / __int_as_float
  int ret = __float_as_int(*address);
  while (val > __int_as_float(ret)) {
    int old = ret;
    if ((ret = atomicCAS((int*)address, old, __float_as_int(val))) == old)
      break;
  }
  return __int_as_float(ret);
#else
  // CPU implementation uses edm::bit_cast
  int ret = edm::bit_cast<int>(*address);
  while (val > edm::bit_cast<float>(ret)) {
    int old = ret;
    if ((ret = alpaka::atomicCas(acc, (int*)address, old, edm::bit_cast<int>(val))) == old)
      break;
  }
  return edm::bit_cast<float>(ret);
#endif  // __CUDA_ARCH__ or __HIP_DEVICE_COMPILE__
}