Skip to content

Commit b53442e

Browse files
authored
Update ggml_extend.hpp
1 parent 85bd9a0 commit b53442e

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

ggml_extend.hpp

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1300,9 +1300,12 @@ class Linear : public UnaryBlock {
13001300
if (bias) {
13011301
b = params["bias"];
13021302
}
1303-
auto out = ggml_nn_linear(ctx, x, w, b);
1304-
print_ggml_tensor(out, true, "imatrix");
1305-
return out;
1303+
auto act = ggml_nn_linear(ctx, x, w, NULL);
1304+
//
1305+
print_ggml_tensor(x, true, "imatrix x");
1306+
print_ggml_tensor(w, true, "imatrix w");
1307+
print_ggml_tensor(act, true, "imatrix act");
1308+
return ggml_nn_linear(ctx, x, w, b);
13061309
}
13071310
};
13081311

0 commit comments

Comments
 (0)