Skip to content

Commit 66db155

Browse files
committed
Small bugfix in CUDA backprop calculation of dz.
1 parent 9d6391e commit 66db155

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/satnet_cuda.cu

+2-2
Original file line numberDiff line numberDiff line change
@@ -241,8 +241,8 @@ __global__ void mix_backward(float prox_lam, int n, int m, int k, int32_t *is_in
241241
for (int kk=warp; kk<k; kk+=WARP_NUM) {
242242
float val = warpdot(S+i*m, Phi+kk*m, m);
243243
__syncwarp();
244-
if (lane == 0) val1 = val;
245-
if (lane == 1) val2 = val;
244+
if (kk == 0) val1 = val;
245+
if (kk == 1) val2 = val;
246246
__syncwarp();
247247
}
248248
__syncthreads();

0 commit comments

Comments
 (0)