Skip to content

Commit dd771e2

Browse files
authored
Update ggml_extend.hpp
1 parent 58851b5 commit dd771e2

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

ggml_extend.hpp

+4-4
Original file line numberDiff line numberDiff line change
@@ -1313,10 +1313,10 @@ class Linear : public UnaryBlock {
13131313
auto im = ggml_mul_mat(ctx, x, w);
13141314

13151315
auto a = ggml_diag_mask_zero(ctx, im, 0);
1316-
auto b = ggml_transpose(ctx, a);
1317-
auto c = ggml_diag_mask_zero(ctx, b, 0);
1318-
auto d = ggml_transpose(ctx, c);
1319-
auto ims = ggml_sum_rows(ctx, d);
1316+
auto c = ggml_transpose(ctx, a);
1317+
auto d = ggml_diag_mask_zero(ctx, c, 0);
1318+
auto e = ggml_transpose(ctx, d);
1319+
auto ims = ggml_sum_rows(ctx, e);
13201320
auto imv = ggml_reshape_1d(ctx, ims, ne);
13211321

13221322
//

0 commit comments

Comments
 (0)