schihei/llama2.c
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix random_f32

Browse Source
This commit is contained in:
aegkmq
2023-07-28 00:35:59 +09:00
parent 1bdf5af743
commit 71200f3092
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
run.c
+1 -1
View File
@@ -345,7 +345,7 @@ unsigned int random_u32() {
return (rng_seed * 0x2545F4914F6CDD1Dull) >> 32; return (rng_seed * 0x2545F4914F6CDD1Dull) >> 32;
} }
float random_f32() { float random_f32() {
return (random_u32() >> 9) * (1.0f / 8388608.0f); return (random_u32() >> 8) / 16777216.0f;
} }
int sample(float* probabilities, int n) { int sample(float* probabilities, int n) {
// sample index from probabilities, they must sum to 1 // sample index from probabilities, they must sum to 1