diff --git a/src/tensors/gpu/tensor_operators.cu b/src/tensors/gpu/tensor_operators.cu index 5f8c4c122..6dbded2a4 100644 --- a/src/tensors/gpu/tensor_operators.cu +++ b/src/tensors/gpu/tensor_operators.cu @@ -1,7 +1,5 @@ -# if defined(_MSC_VER) +# if !defined(NPP_MAX_32U) #define NPP_MAX_32U ( 4294967295U ) /**< Maximum 32-bit unsigned integer */ -#else -#include #endif #include "common/types.h" @@ -3548,7 +3546,7 @@ __global__ void HammmingAndSort(const uint32_t *weightHash, if (outIdx != NPP_MAX_32U) { uint32_t prevOutIdx; // Not supported in Maxwells or older -// Not supported in Maxwells or older +// Not supported in Maxwells or older #if __CUDA_ARCH__ >= 600 prevOutIdx = atomicAdd_block(&outIdx, (uint32_t) -1); #else