mirror-github-bspeice/cuburn
1
0
Fork 0
mirror of https://github.com/stevenrobertson/cuburn.git synced 2025-02-05 11:40:04 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix potential bug in binsearch

Browse Source
This commit is contained in:
Steven Robertson 2011-12-09 14:14:00 -05:00
parent d3ee6f36c2
commit b592cda3db
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
cuburn/code/interp.py
View File

@ -296,9 +296,10 @@ __device__
int bitwise_binsearch(const float *hay, float needle) {
int lo = 0;
// TODO: improve efficiency on 64-bit arches
{{for i in range(search_rounds-1, -1, -1)}}
if (needle > hay[lo | {{1 << i}}])
lo |= {{1 << i}};
if (needle > hay[lo + {{1 << i}}])
lo += {{1 << i}};
{{endfor}}
return lo;
}