MCPcopy Create free account
hub / github.com/appdevforall/CodeOnTheGo / init_mappings

Method init_mappings

subprojects/llama.cpp/src/llama-model-loader.cpp:898–929  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

896}
897
898void llama_model_loader::init_mappings(bool prefetch, llama_mlocks * mlock_mmaps) {
899 if (use_mmap) {
900 mappings.reserve(files.size());
901 mmaps_used.reserve(files.size());
902 for (const auto & file : files) {
903 bool is_numa = false;
904
905 auto * dev = ggml_backend_dev_by_type(GGML_BACKEND_DEVICE_TYPE_CPU);
906 if (dev) {
907 auto * reg = ggml_backend_dev_backend_reg(dev);
908 auto * is_numa_fn = (decltype(ggml_is_numa) *) ggml_backend_reg_get_proc_address(reg, "ggml_backend_cpu_is_numa");
909 if (is_numa_fn) {
910 is_numa = is_numa_fn();
911 }
912 }
913
914 std::unique_ptr<llama_mmap> mapping = std::make_unique<llama_mmap>(file.get(), prefetch ? -1 : 0, is_numa);
915 mmaps_used.emplace_back(mapping->size(), 0);
916 if (mlock_mmaps) {
917 std::unique_ptr<llama_mlock> mlock_mmap(new llama_mlock());
918 mlock_mmap->init(mapping->addr());
919 mlock_mmaps->emplace_back(std::move(mlock_mmap));
920 }
921 mappings.emplace_back(std::move(mapping));
922 }
923 }
924
925 // compute the total size of all tensors for progress reporting
926 for (const auto & it : weights_map) {
927 size_data += ggml_nbytes(it.second.tensor);
928 }
929}
930
931void llama_model_loader::get_mapping_range(size_t * first, size_t * last, void ** addr, int idx, ggml_context * ctx) const {
932 GGML_ASSERT(!mappings.empty());

Callers 2

load_tensorsMethod · 0.80

Calls 8

ggml_backend_dev_by_typeFunction · 0.85
ggml_nbytesFunction · 0.85
addrMethod · 0.80
sizeMethod · 0.65
getMethod · 0.65
initMethod · 0.65

Tested by

no test coverage detected