htdy7703's picture
Add build_stubs.cpp
21059f2 verified
Raw
History Blame Contribute Delete
824 Bytes
#include <cstddef>
#include <cstdint>
#include <cstdio>
#include <cstring>
extern "C" {
int32_t llama_split_path(char * split_path, size_t maxlen, const char * path_prefix, int32_t split_no, int32_t split_count) {
return std::snprintf(split_path, maxlen, "%s-%05d-of-%05d.gguf", path_prefix, split_no, split_count);
}
int32_t llama_split_prefix(char * split_prefix, size_t maxlen, const char * split_path, int32_t split_no, int32_t split_count) {
(void) split_no;
(void) split_count;
const char * suffix = std::strrchr(split_path, '-');
size_t len = suffix ? static_cast<size_t>(suffix - split_path) : std::strlen(split_path);
if (len + 1 > maxlen) {
return 0;
}
std::memcpy(split_prefix, split_path, len);
split_prefix[len] = '\0';
return static_cast<int32_t>(len);
}
}