call __signbit for CUDA >= 6.5 implementation

This commit is contained in:
Evan Shelhamer 2014-09-03 17:45:14 -07:00
Родитель 77d91242cd
Коммит 8819f5953b
1 изменённых файлов: 5 добавлений и 0 удалений

Просмотреть файл

@ -329,7 +329,12 @@ void caffe_gpu_powx<double>(const int N, const double* a,
DEFINE_AND_INSTANTIATE_GPU_UNARY_FUNC(sign, y[index] = (Dtype(0) < x[index])
- (x[index] < Dtype(0)));
#if CUDA_VERSION >= 6050
// __signbit to pick up the CUDA function.
DEFINE_AND_INSTANTIATE_GPU_UNARY_FUNC(sgnbit, y[index] = __signbit(x[index]));
#else
DEFINE_AND_INSTANTIATE_GPU_UNARY_FUNC(sgnbit, y[index] = signbit(x[index]));
#endif
__global__ void popc_kernel(const int n, const float* a,
const float* b, uint8_t* y) {