Skip to content

Commit c45a5c4

Browse files
committed
fix(ggml-cpu): use dst dims (ne0..ne3) for carry in dup_f16 non-contiguous path
1 parent fbef0fa commit c45a5c4

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

ggml/src/ggml-cpu/ops.cpp

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -219,13 +219,13 @@ static void ggml_compute_forward_dup_f16(
219219

220220
memcpy(dst_ptr, src0_ptr, sizeof(ggml_fp16_t));
221221

222-
if (++i10 == ne00) {
222+
if (++i10 == ne0) {
223223
i10 = 0;
224-
if (++i11 == ne01) {
224+
if (++i11 == ne1) {
225225
i11 = 0;
226-
if (++i12 == ne02) {
226+
if (++i12 == ne2) {
227227
i12 = 0;
228-
if (++i13 == ne03) {
228+
if (++i13 == ne3) {
229229
i13 = 0;
230230
}
231231
}
@@ -518,13 +518,13 @@ static void ggml_compute_forward_dup_bf16(
518518

519519
memcpy(dst_ptr, src0_ptr, sizeof(ggml_bf16_t));
520520

521-
if (++i10 == ne00) {
521+
if (++i10 == ne0) {
522522
i10 = 0;
523-
if (++i11 == ne01) {
523+
if (++i11 == ne1) {
524524
i11 = 0;
525-
if (++i12 == ne02) {
525+
if (++i12 == ne2) {
526526
i12 = 0;
527-
if (++i13 == ne03) {
527+
if (++i13 == ne3) {
528528
i13 = 0;
529529
}
530530
}

0 commit comments

Comments
 (0)