HeterogeneousCore/CUDAUtilities/interface/cuda_assert.h - master - githubmirror/cmssw

HeterogeneousCore/CUDAUtilities/interface/cuda_assert.h

18 строк368 B

Andrea Bocci

Import CUDA interaction layer from Patatrack, part 1

18 янв 2020, 01:00

18 янв 2020, 01:00f143441

// The omission of #include guards is on purpose: it does make sense to #include
// this file multiple times, setting a different value of GPU_DEBUG beforehand.

#ifdef __CUDA_ARCH__
#ifndef GPU_DEBUG
// disable asserts
#ifndef NDEBUG
#define NDEBUG
#endif
#else
// enable asserts
#ifdef NDEBUG
#undef NDEBUG
#endif
#endif
#endif  // __CUDA_ARCH__

#include <cassert>