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

Apply suggestions from code review

Browse source 
Co-authored-by: slaren <slarengh@gmail.com>
This commit is contained in:
l3utterfly 2023-08-04 19:14:26 +08:00 • committed by GitHub
parent d4a126cdb5
commit e74d42dfff
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
llama-util.h
View file

@ -151,7 +151,7 @@ struct llama_file {
// llama_context_data // llama_context_data
struct llama_data_context { struct llama_data_context {
virtual void write(const void* src, size_t size) = 0; virtual void write(const void * src, size_t size) = 0;
virtual size_t get_size_written() = 0; virtual size_t get_size_written() = 0;
virtual ~llama_data_context() = default; virtual ~llama_data_context() = default;
}; };
@ -160,9 +160,9 @@ struct llama_data_buffer_context : llama_data_context {
uint8_t* ptr; uint8_t* ptr;
size_t size_written = 0; size_t size_written = 0;
llama_data_buffer_context(uint8_t* p) : ptr(p) {} llama_data_buffer_context(uint8_t * p) : ptr(p) {}
void write(const void* src, size_t size) override { void write(const void * src, size_t size) override {
memcpy(ptr, src, size); memcpy(ptr, src, size);
ptr += size; ptr += size;
size_written += size; size_written += size;
@ -177,9 +177,9 @@ struct llama_data_file_context : llama_data_context {
llama_file* file; llama_file* file;
size_t size_written = 0; size_t size_written = 0;
llama_data_file_context(llama_file* f) : file(f) {} llama_data_file_context(llama_file * f) : file(f) {}
void write(const void* src, size_t size) override { void write(const void * src, size_t size) override {
file->write_raw(src, size); file->write_raw(src, size);
size_written += size; size_written += size;
} }