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

fix small vector indexing checks

Browse Source
This commit is contained in:
Ronan Collobert
2025-10-31 11:46:36 -07:00
parent 979abf462b
commit 6343622c67
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mlx/small_vector.h
View File

@@ -302,7 +302,7 @@ class SmallVector {
} }
T& at(int index) { T& at(int index) {
if (index >= size()) { if (index < 0 || index >= size()) {
throw std::out_of_range("SmallVector out of range."); throw std::out_of_range("SmallVector out of range.");
} }
return begin_[index]; return begin_[index];
@@ -312,7 +312,7 @@ class SmallVector {
} }
T& operator[](int index) { T& operator[](int index) {
assert(size() > index); assert(index >= 0 && size() > index);
return begin_[index]; return begin_[index];
} }
const T& operator[](int index) const { const T& operator[](int index) const {