mirror of
https://github.com/opencv/opencv.git
synced 2024-11-29 22:00:25 +08:00
add compute capability check
This commit is contained in:
parent
f52ce6529b
commit
7c98735c0d
@ -505,7 +505,10 @@ __device void CascadeInvoker<Policy>::detect(Detection* objects, const uint ndet
|
|||||||
#if __CUDA_ARCH__ >= 120
|
#if __CUDA_ARCH__ >= 120
|
||||||
if(__any((confidence + impact <= stages[(st + threadIdx.x)]))) st += 2048;
|
if(__any((confidence + impact <= stages[(st + threadIdx.x)]))) st += 2048;
|
||||||
#endif
|
#endif
|
||||||
|
#if __CUDA_ARCH__ >= 300
|
||||||
impact = __shfl(impact, 31);
|
impact = __shfl(impact, 31);
|
||||||
|
#endif
|
||||||
|
|
||||||
confidence += impact;
|
confidence += impact;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user