huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-19 19:58:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Small optimization.

Browse Source
This commit is contained in:
laurent
2024-03-17 20:15:51 +01:00
parent 75f2aea5fd
commit dffafd1049
1 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
candle-kernels/src/conv.cu
View File

@ -82,13 +82,12 @@ __device__ void col2im1d(
const size_t src_s2 = k_size; const size_t src_s2 = k_size;
T d = 0; T d = 0;
for (size_t k_i = 0; k_i < min(dst_i3 + 1, k_size); ++k_i) { size_t l_in_i = dst_i3 / stride;
const size_t l_in_i_times_stride = dst_i3 - k_i; const size_t k_i = dst_i3 - stride * l_in_i;
const size_t l_in_i = l_in_i_times_stride / stride; for (size_t k_i = 0; k_i < min(dst_i3 + 1, k_size); k_i += stride) {
const size_t src_i = b_i * src_s0 + l_in_i * src_s1 + c_i * src_s2 + k_i; const size_t src_i = b_i * src_s0 + l_in_i * src_s1 + c_i * src_s2 + k_i;
if (l_in_i * stride == l_in_i_times_stride) {
d += src[src_i]; d += src[src_i];
} l_in_i -= 1;
} }
dst[dst_i] = d; dst[dst_i] = d;
} }