| 896 | } |
| 897 | |
| 898 | void 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 | |
| 931 | void 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()); |
no test coverage detected