zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-17 17:28:10 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use int64_t instead of ssize_t (#1673)

Browse Source
This commit is contained in:
Cheng
2024-12-08 13:10:44 +09:00
committed by GitHub
parent 7c10c93a1f
commit 6f316b8bf5
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mlx/utils.h
View File

@@ -70,7 +70,7 @@ bool is_same_shape(const std::vector<array>& arrays);
template <typename T>
int check_shape_dim(const T dim) {
constexpr bool is_signed = std::numeric_limits<T>::is_signed;
using U = std::conditional_t<is_signed, ssize_t, size_t>;
using U = std::conditional_t<is_signed, int64_t, size_t>;
constexpr U min = static_cast<U>(std::numeric_limits<int>::min());
constexpr U max = static_cast<U>(std::numeric_limits<int>::max());