CMSUnrollLoop.h

Go to the documentation of this file.

#ifndef FWCore_Utilities_interface_CMSUnrollLoop_h
#define FWCore_Utilities_interface_CMSUnrollLoop_h
 
// convert the macro argument to a null-terminated quoted string
#define STRINGIFY_(ARG) #ARG
#define STRINGIFY(ARG) STRINGIFY_(ARG)
 
#if defined(__CUDA_ARCH__) || defined(__HIP_DEVICE_COMPILE__)
// CUDA or HIP device compiler
 
#define CMS_UNROLL_LOOP _Pragma(STRINGIFY(unroll))
#define CMS_UNROLL_LOOP_COUNT(N) _Pragma(STRINGIFY(unroll N))
#define CMS_UNROLL_LOOP_DISABLE _Pragma(STRINGIFY(unroll 1))
 
#define CMS_DEVICE_UNROLL_LOOP _Pragma(STRINGIFY(unroll))
#define CMS_DEVICE_UNROLL_LOOP_COUNT(N) _Pragma(STRINGIFY(unroll N))
#define CMS_DEVICE_UNROLL_LOOP_DISABLE _Pragma(STRINGIFY(unroll 1))
 
#else  // defined (__CUDA_ARCH__) || defined (__HIP_DEVICE_COMPILE__)
 
// any host compiler
#define CMS_DEVICE_UNROLL_LOOP
#define CMS_DEVICE_UNROLL_LOOP_COUNT(N)
#define CMS_DEVICE_UNROLL_LOOP_DISABLE
 
#if defined(__clang__)
// clang host compiler
 
#define CMS_UNROLL_LOOP _Pragma(STRINGIFY(clang loop unroll(enable)))
#define CMS_UNROLL_LOOP_COUNT(N) _Pragma(STRINGIFY(clang loop unroll_count(N)))
#define CMS_UNROLL_LOOP_DISABLE _Pragma(STRINGIFY(clang loop unroll(disable)))
 
#elif defined(__GNUC__)
// GCC host compiler
 
#define CMS_UNROLL_LOOP _Pragma(STRINGIFY(GCC ivdep))
#define CMS_UNROLL_LOOP_COUNT(N) _Pragma(STRINGIFY(GCC unroll N)) _Pragma(STRINGIFY(GCC ivdep))
#define CMS_UNROLL_LOOP_DISABLE _Pragma(STRINGIFY(GCC unroll 1))
 
#else
// unsupported or unknown compiler
 
#define CMS_UNROLL_LOOP
#define CMS_UNROLL_LOOP_COUNT(N)
#define CMS_UNROLL_LOOP_DISABLE
 
#endif  // defined(__clang__) || defined(__GNUC__) || ...
 
#endif  // defined (__CUDA_ARCH__) || defined (__HIP_DEVICE_COMPILE__)
 
#endif  // FWCore_Utilities_interface_CMSUnrollLoop_h