mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

another compile fix

Browse Source
This commit is contained in:
AlexeyAB
2018-09-07 15:25:37 +03:00
parent b141f85cab
commit 57e878b4f9
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/im2col_kernels.cu
View File

@ -131,10 +131,11 @@ __global__ void float_to_bit_gpu_kernel(float *src, unsigned char *dst, size_t s
//for (; index < size_aligned; index += blockDim.x*gridDim.x)
{
src_val = src[index];
//if(index < size) src_val = src[index];
//else src_val = 0;
unsigned int bit_mask = __ballot_sync(0xffffffff, src_val > 0);
//src_val = src[index];
if(index < size) src_val = src[index];
else src_val = 0;
//unsigned int bit_mask = __ballot_sync(0xffffffff, src_val > 0);
unsigned int bit_mask = __ballot(src_val > 0);
if (threadIdx.x % WARP_SIZE == 0) ((unsigned int*)dst)[index / 32] = bit_mask;
}
}