haraldwolff
/
CLBlast
mirror of https://github.com/CNugteren/CLBlast.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Made it more likely (but no guarantees) for amax/amin to return the first index

pull/378/head
Cedric Nugteren 2020-03-08 11:26:49 +01:00
parent 7fab29304c
commit b46853660e
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/kernels/level1/xamax.opencl
View File

@ -64,7 +64,7 @@ void Xamax(const int n,
#else
x = fabs(x);
#endif
if (x >= max) {
if (x > max) {
max = x;
imax = id*x_inc + x_offset;
}
@ -77,7 +77,7 @@ void Xamax(const int n,
// Performs reduction in local memory
for (int s=WGS1/2; s>0; s=s>>1) {
if (lid < s) {
if (maxlm[lid + s] >= maxlm[lid]) {
if (maxlm[lid + s] > maxlm[lid]) {
maxlm[lid] = maxlm[lid + s];
imaxlm[lid] = imaxlm[lid + s];
}
@ -105,7 +105,7 @@ void XamaxEpilogue(const __global singlereal* restrict maxgm,
const int lid = get_local_id(0);
// Performs the first step of the reduction while loading the data
if (maxgm[lid + WGS2] >= maxgm[lid]) {
if (maxgm[lid + WGS2] > maxgm[lid]) {
maxlm[lid] = maxgm[lid + WGS2];
imaxlm[lid] = imaxgm[lid + WGS2];
}
@ -118,7 +118,7 @@ void XamaxEpilogue(const __global singlereal* restrict maxgm,
// Performs reduction in local memory
for (int s=WGS2/2; s>0; s=s>>1) {
if (lid < s) {
if (maxlm[lid + s] >= maxlm[lid]) {
if (maxlm[lid + s] > maxlm[lid]) {
maxlm[lid] = maxlm[lid + s];
imaxlm[lid] = imaxlm[lid + s];
}