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

Guard nullptr dereference (#1972)

Browse Source 
* guard nullptr dereference

* comment
This commit is contained in:
Awni Hannun
2025-03-19 16:24:10 -07:00
committed by GitHub
parent 3779150750
commit f90206ad74
2 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -20,6 +20,9 @@ Allocator& allocator() {
} }
void* Buffer::raw_ptr() { void* Buffer::raw_ptr() {
if (!ptr_) {
return nullptr;
}
return static_cast<MTL::Buffer*>(ptr_)->contents(); return static_cast<MTL::Buffer*>(ptr_)->contents();
} }

3
mlx/backend/no_metal/allocator.cpp
View File

@@ -10,6 +10,9 @@ Allocator& allocator() {
} }
void* Buffer::raw_ptr() { void* Buffer::raw_ptr() {
if (!ptr_) {
return nullptr;
}
return static_cast<size_t*>(ptr_) + 1; return static_cast<size_t*>(ptr_) + 1;
} }