fix pointer assignment error in softmax_cuda_kernel.cu

ad7f0cb5 · Shenggan · 9924e7be · ad7f0cb5
Commit ad7f0cb5 authored Jul 27, 2022 by Shenggan
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

fastfold/model/fastnn/kernel/cuda_native/csrc/softmax_cuda_kernel.cu ...del/fastnn/kernel/cuda_native/csrc/softmax_cuda_kernel.cu +1 -1

No files found.
--- a/fastfold/model/fastnn/kernel/cuda_native/csrc/softmax_cuda_kernel.cu
+++ b/fastfold/model/fastnn/kernel/cuda_native/csrc/softmax_cuda_kernel.cu
@@ -561,7 +561,7 @@ __global__ void fastfold_softmax_scale_mask_grad(T *d_output, T *output, T *d_in
                row_d_input[lane_id * cols_per_thread + i] =
                    static_cast<T>(scale * ((dy_buf[i] - warp_sum) * y_buf[i]));
            } else {
-                row_d_input = 0;
+                row_d_input[lane_id * cols_per_thread + i] = 0;
            }
        }
    }