Skip to content

GitLab

Commit ad7f0cb5 authored by Shenggan's avatar Shenggan
Browse files

fix pointer assignment error in softmax_cuda_kernel.cu

parent 9924e7be
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
View file @ ad7f0cb5
...@@ -561,7 +561,7 @@ __global__ void fastfold_softmax_scale_mask_grad(T *d_output, T *output, T *d_in ...@@ -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] = row_d_input[lane_id * cols_per_thread + i] =
static_cast<T>(scale * ((dy_buf[i] - warp_sum) * y_buf[i])); static_cast<T>(scale * ((dy_buf[i] - warp_sum) * y_buf[i]));
} else { } else {
row_d_input = 0; row_d_input[lane_id * cols_per_thread + i] = 0;
} }
} }
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment