|
|
@@ -6383,7 +6383,7 @@ static void ggml_compute_forward_im2col_3d_f16(
|
|
|
const int64_t iih = ioh*s1 + ikh*d1 - p1;
|
|
|
const int64_t iid = iod*s2 + ikd*d2 - p2;
|
|
|
|
|
|
- if (iid < 0 || iid >= ID || iih < 0 || iih >= IH || iiw < 0 || iiw >= IW || iid < 0 || iid >= ID) {
|
|
|
+ if (iid < 0 || iid >= ID || iih < 0 || iih >= IH || iiw < 0 || iiw >= IW) {
|
|
|
dst_data[iic*KD_KH_KW + ikd * KH_KW + ikh*KW + ikw] = 0;
|
|
|
} else {
|
|
|
const float * const s = (const float *) ((const char *)src_data + iid*nb12 + iih*nb11 + iiw*nb10); // [ID, IH, IW]
|