Skip to content

Commit 51a45c4

Browse files
authored
patch Linear.forward
1 parent 5eb15ef commit 51a45c4

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

ggml_extend.hpp

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1300,7 +1300,9 @@ class Linear : public UnaryBlock {
13001300
if (bias) {
13011301
b = params["bias"];
13021302
}
1303-
return ggml_nn_linear(ctx, x, w, b);
1303+
auto out = ggml_nn_linear(ctx, x, w, b);
1304+
print_ggml_tensor(out, true, "imatrix");
1305+
return out;
13041306
}
13051307
};
13061308

0 commit comments

Comments
 (0)