Skip to content

Commit 5ecbe6e

Browse files
pwilkinggerganov
andauthored
Update ggml/src/ggml-cpu/ops.cpp
Co-authored-by: Georgi Gerganov <ggerganov@gmail.com>
1 parent 61667c3 commit 5ecbe6e

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

ggml/src/ggml-cpu/ops.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2213,9 +2213,10 @@ static void ggml_compute_forward_tri_f32(const ggml_compute_params * params, ggm
22132213
const int64_t i02 = (ir - i03*ne02*ne01)/ne01;
22142214
const int64_t i01 = (ir - i03*ne02*ne01 - i02*ne01);
22152215

2216-
float * dst_ptr = (float *) ((char *) dst->data + i03*nb3 + i02*nb2 + i01*nb1 );
2217-
float * src = (float *) ((char *) src0->data + i03*nb03 + i02*nb02 + i01*nb01 );
2218-
ggml_vec_tri_f32(ne0, i01, dst_ptr, src, keep_org_val, c, ttype);
2216+
float * dst_ptr = (float *) ((char *) dst->data + i03*nb3 + i02*nb2 + i01*nb1);
2217+
float * src_ptr = (float *) ((char *) src0->data + i03*nb03 + i02*nb02 + i01*nb01);
2218+
2219+
ggml_vec_tri_f32(ne0, i01, dst_ptr, src_ptr, keep_org_val, c, ttype);
22192220
}
22202221

22212222
}

0 commit comments

Comments
 (0)