vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fixing potential NULL dereference in training

Browse source
This commit is contained in:
tpltnt 2023-10-10 18:19:01 +02:00
parent 02d2875def
commit 94496cd49c
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
common/train.cpp
View file

@ -46,6 +46,10 @@ struct random_normal_distribution * init_random_normal_distribution(
int seed, float mean, float std, float min, float max
) {
struct random_normal_distribution * rnd = (struct random_normal_distribution *) malloc(sizeof(struct random_normal_distribution));
if( NULL == rnd) {
fprintf(stderr, "could not allocate memory for normal distribution data\n");
exit(1);
}
rnd->gen = std::mt19937(seed);
rnd->rd = std::normal_distribution<float>{mean, std};
rnd->min = min;