zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-21 11:38:06 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

increase block limit to original

Browse Source
This commit is contained in:
Awni Hannun 2024-01-02 09:59:58 -08:00
parent 97ea5b23cb
commit 3541a58445
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

@ -148,7 +148,7 @@ MetalAllocator::MetalAllocator()
: device_(device(mlx::core::Device::gpu).mtl_device()), : device_(device(mlx::core::Device::gpu).mtl_device()),
buffer_cache_(device_), buffer_cache_(device_),
peak_allocated_size_(0), peak_allocated_size_(0),
block_limit_(device_->recommendedMaxWorkingSetSize()), block_limit_(1.5 * device_->recommendedMaxWorkingSetSize()),
gc_limit_(0.95 * device_->recommendedMaxWorkingSetSize()) {} gc_limit_(0.95 * device_->recommendedMaxWorkingSetSize()) {}
Buffer MetalAllocator::malloc(size_t size, bool allow_swap /* = false */) { Buffer MetalAllocator::malloc(size_t size, bool allow_swap /* = false */) {