quantize : fail fast on write errors (#3521)

2023-10-07 04:41:52 -04:00 · 2023-10-07 04:41:52 -04:00 · 80735cb7bd
parent bff47ce69b
commit 80735cb7bd
1 changed files with 1 additions and 0 deletions
--- a/llama.cpp
+++ b/llama.cpp
@ -7194,6 +7194,7 @@ static void llama_model_quantize_internal(const std::string & fname_inp, const s
    }

    std::ofstream fout(fname_out, std::ios::binary);
+    fout.exceptions(std::ofstream::failbit); // fail fast on write errors

    const size_t meta_size = gguf_get_meta_size(ctx_out);