diff --git a/src/devices/cpu/device_.h b/src/devices/cpu/device_.h
index 72e0d1a..22b1f36 100644
--- a/src/devices/cpu/device_.h
+++ b/src/devices/cpu/device_.h
@@ -8,6 +8,11 @@ namespace infini::ccl {
 template <>
 struct DeviceEnabled<Device::Type::kCpu> : std::true_type {};
 
+template <>
+MemorySpace GetMemorySpace<Device::Type::kCpu>(const void* /*ptr*/) {
+  return MemorySpace::kHost;
+}
+
 }  // namespace infini::ccl
 
 #endif  // INFINI_CCL_DEVICES_CPU_DEVICE_H_