zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-20 18:39:45 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

nit in comment

Browse Source
This commit is contained in:
Awni Hannun 2023-12-28 17:01:41 -08:00
parent 58f3c15963
commit 5ff2a6c762
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/backend/metal/allocator.cpp
View File

@ -51,7 +51,7 @@ MTL::Buffer* BufferCache::reuse_from_cache(size_t size) {
// Find the closest buffer in pool
MTL::Buffer* pbuf = nullptr;
// Make sure we use > 50% of the available memory
// Make sure we use most of the available memory
if (auto it = buffer_pool_.lower_bound(size); it != buffer_pool_.end() &&
it->first < std::min(2 * size, size + vm_page_size)) {
// Collect from the cache