From 6f316b8bf581ef3f280ba791994d79ae7dddb22e Mon Sep 17 00:00:00 2001
From: Cheng <zcbenz@gmail.com>
Date: Sun, 8 Dec 2024 13:10:44 +0900
Subject: [PATCH] Use int64_t instead of ssize_t (#1673)

---
 mlx/utils.h           | 2 +-
 tests/utils_tests.cpp | 8 ++++----
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/mlx/utils.h b/mlx/utils.h
index eb194b71ec..5e0ef22222 100644
--- a/mlx/utils.h
+++ b/mlx/utils.h
@@ -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());
 
diff --git a/tests/utils_tests.cpp b/tests/utils_tests.cpp
index f975537139..3f2e3f814f 100644
--- a/tests/utils_tests.cpp
+++ b/tests/utils_tests.cpp
@@ -67,15 +67,15 @@ TEST_CASE("test check shape dimension") {
   CHECK_EQ(check_shape_dim(-4), -4);
   CHECK_EQ(check_shape_dim(0), 0);
   CHECK_EQ(check_shape_dim(12), 12);
-  CHECK_EQ(check_shape_dim(static_cast<ssize_t>(dim_min)), dim_min);
-  CHECK_EQ(check_shape_dim(static_cast<ssize_t>(dim_max)), dim_max);
+  CHECK_EQ(check_shape_dim(static_cast<int64_t>(dim_min)), dim_min);
+  CHECK_EQ(check_shape_dim(static_cast<int64_t>(dim_max)), dim_max);
   CHECK_EQ(check_shape_dim(static_cast<size_t>(0)), 0);
   CHECK_EQ(check_shape_dim(static_cast<size_t>(dim_max)), dim_max);
   CHECK_THROWS_AS(
-      check_shape_dim(static_cast<ssize_t>(dim_min) - 1),
+      check_shape_dim(static_cast<int64_t>(dim_min) - 1),
       std::invalid_argument);
   CHECK_THROWS_AS(
-      check_shape_dim(static_cast<ssize_t>(dim_max) + 1),
+      check_shape_dim(static_cast<int64_t>(dim_max) + 1),
       std::invalid_argument);
   CHECK_THROWS_AS(
       check_shape_dim(static_cast<size_t>(dim_max) + 1), std::invalid_argument);