add srand() in speculative.cpp

ggml-org · ggerganov · Mar 4, 2024 · Feb 21, 2024 · Feb 22, 2024 · Feb 22, 2024
commit 6afc1f60e1dac858ca5b45dcd3c528ad0afae1b4
diff --git a/examples/speculative/speculative.cpp b/examples/speculative/speculative.cpp
@@ -41,6 +41,12 @@ int main(int argc, char ** argv) {
    // probability threshold for splitting a draft branch (only for n_seq_dft > 1)
    const float p_split  = params.p_split;

+    if (params.seed >= 0) {
+        srand(params.seed);
+    } else {
+        srand(time(NULL));
+    }
+
 #ifndef LOG_DISABLE_LOGS
    log_set_target(log_filename_generator("speculative", "log"));
    LOG_TEE("Log start\n");