From 268c439d6541bbe09f1833b31becf2f1dffa7cb4 Mon Sep 17 00:00:00 2001
From: Jiacheng Huang <huangjiacheng0709@outlook.com>
Date: Tue, 14 Jul 2026 17:02:47 +0800
Subject: [PATCH] fix: include fixed-width integer type

---
 src/device.h | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/device.h b/src/device.h
index 67e1b72..64eb7c9 100644
--- a/src/device.h
+++ b/src/device.h
@@ -1,6 +1,7 @@
 #ifndef INFINI_CCL_DEVICE_H_
 #define INFINI_CCL_DEVICE_H_
 
+#include <cstdint>
 #include <string>
 
 #include "constexpr_map.h"
@@ -155,7 +156,7 @@ struct DevicePriority<Device::Type::kCambricon> {
   static constexpr int value = 5;
 };
 
-enum class MemorySpace : uint8_t { kHost = 0, kDevice = 1, kUnknown };
+enum class MemorySpace : std::uint8_t { kHost = 0, kDevice = 1, kUnknown };
 
 template <Device::Type kDev>
 MemorySpace GetMemorySpace(const void *ptr);