zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Metal validation (#432)

Browse Source 
* tests clear metal validation

* add cpp test with metal validation to circleci

* nit
This commit is contained in:
Awni Hannun
2024-01-11 11:57:24 -08:00
committed by GitHub
parent 975e265f74
commit c9934fe8a4
10 changed files with 142 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -153,6 +153,11 @@ MetalAllocator::MetalAllocator()
gc_limit_(0.95 * device_->recommendedMaxWorkingSetSize()) {}
Buffer MetalAllocator::malloc(size_t size, bool allow_swap /* = false */) {
// Metal doesn't like empty buffers
if (size == 0) {
return Buffer{nullptr};
}
// Align up memory
if (size > vm_page_size) {
size = vm_page_size * ((size + vm_page_size - 1) / vm_page_size);