Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 4 additions & 3 deletions CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -142,9 +142,9 @@ if(${HMLL_ENABLE_SAFETENSORS})
add_compile_definitions(__HMLL_SAFETENSORS_ENABLED__=1)
add_compile_definitions(__HMLL_TENSORS_ENABLED__=1)
fetchcontent_declare(
yyjson
GIT_REPOSITORY https://github.com/ibireme/yyjson.git
GIT_TAG master
yyjson
GIT_REPOSITORY https://github.com/ibireme/yyjson.git
GIT_TAG master
)
fetchcontent_makeavailable(yyjson)

Expand Down Expand Up @@ -232,6 +232,7 @@ if(${HMLL_BUILD_TESTS})
tests/tests_hmll_dtype.cpp
tests/tests_hmll_safetensors.cpp
tests/tests_hmll_integration_safetensors.cpp
tests/tests_hmll_fetchv.cpp
)
endif()

Expand Down
1 change: 1 addition & 0 deletions include/hmll/hmll.h
Original file line number Diff line number Diff line change
Expand Up @@ -132,6 +132,7 @@ HMLL_EXTERN struct hmll_error hmll_get_mmap_view(struct hmll *ctx, int iofile, s
/** Tensors manipulation stubs - enabled if a higher-level tensor format is enabled **/
#ifdef __HMLL_TENSORS_ENABLED__
HMLL_EXTERN uint8_t hmll_nbits(enum hmll_dtype dtype) NO_EXCEPT;
HMLL_EXTERN size_t hmll_nbytes(const struct hmll_tensor_specs *specs) NO_EXCEPT;
HMLL_EXTERN size_t hmll_numel(const struct hmll_tensor_specs *specs) NO_EXCEPT;
HMLL_EXTERN void hmll_free_registry(struct hmll_registry *reg) NO_EXCEPT;
HMLL_EXTERN unsigned char hmll_contains(const struct hmll *ctx, const struct hmll_registry *reg, const char *name) NO_EXCEPT;
Expand Down
4 changes: 0 additions & 4 deletions include/hmll/linux/backend/iouring.h
Original file line number Diff line number Diff line change
Expand Up @@ -11,10 +11,6 @@
#define HMLL_URING_BUFFER_SIZE (512U * 1024)
#endif

#ifndef HMLL_URING_CQE_BATCH_SIZE
#define HMLL_URING_CQE_BATCH_SIZE 16
#endif

#include <liburing.h>
#include "hmll/types.h"

Expand Down
1 change: 1 addition & 0 deletions lib/context.c
Original file line number Diff line number Diff line change
Expand Up @@ -19,5 +19,6 @@ void hmll_destroy(struct hmll *ctx)
if (ctx->fetcher) {
ctx->fetcher->backend_free(ctx->fetcher->backend_impl_);
free(ctx->fetcher);
ctx->fetcher = NULL;
}
}
Loading
Loading