diff --git a/build/darknet/x64/partial.cmd b/build/darknet/x64/partial.cmd index 2d9e022a..bea737ea 100644 --- a/build/darknet/x64/partial.cmd +++ b/build/darknet/x64/partial.cmd @@ -9,6 +9,9 @@ rem darknet.exe partial cfg/tiny-yolo-voc.cfg tiny-yolo-voc.weights tiny-yolo-vo darknet.exe partial cfg/darknet53_448.cfg darknet53_448.weights darknet53.conv.74 74 +darknet.exe partial cfg/darknet53_448_xnor.cfg darknet53_448_xnor.weights darknet53_448_xnor.conv.74 74 + + darknet.exe partial cfg/yolov2-tiny-voc.cfg yolov2-tiny-voc.weights yolov2-tiny-voc.conv.13 13 diff --git a/src/gemm.c b/src/gemm.c index 4948fbef..b110da61 100644 --- a/src/gemm.c +++ b/src/gemm.c @@ -1646,7 +1646,7 @@ void im2col_cpu_custom_bin(float* data_im, __m256 result256 = _mm256_cmp_ps(src256, float_zero256, _CMP_GT_OS); uint16_t mask = _mm256_movemask_ps(result256); // (val > 0) ? 0 : 1 - uint16_t* dst_ptr = &((uint16_t*)data_col)[col_index / 8]; + uint16_t* dst_ptr = (uint16_t*)&((uint8_t*)data_col)[col_index / 8]; *dst_ptr |= (mask << (col_index % 8)); }