Skip to content

Commit ca99a6c

Browse files
committed
llama : fix uninitialized tensors
1 parent 4d45775 commit ca99a6c

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/llama.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2875,6 +2875,7 @@ struct llama_model {
28752875
llama_hparams hparams = {};
28762876
llama_vocab vocab;
28772877

2878+
// TODO: should init all tensors to nullptr
28782879
struct ggml_tensor * tok_embd;
28792880
struct ggml_tensor * type_embd;
28802881
struct ggml_tensor * pos_embd;
@@ -2890,8 +2891,8 @@ struct llama_model {
28902891
// classifier
28912892
struct ggml_tensor * cls;
28922893
struct ggml_tensor * cls_b;
2893-
struct ggml_tensor * cls_out;
2894-
struct ggml_tensor * cls_out_b;
2894+
struct ggml_tensor * cls_out = nullptr;
2895+
struct ggml_tensor * cls_out_b = nullptr;
28952896

28962897
std::vector<llama_layer> layers;
28972898

0 commit comments

Comments
 (0)