GCC 会为不同 CPU 架构预定义宏，如 __x86_64__ 代表 Intel 64 位 CPU， __aarch64__代表 ARM64。 网上已经有文档对 GCC 为 CPU 的预定义宏进行了总结。
这些预定义的宏有什么用呢？我们在代码中可以判断出当前的 CPU 架构，那么可以针对 不同 CPU 的特性，进行优化实现。例如RocksDB 对于获取当前时间，在 x86 平台上，会用到 Time Stamp Counter (TSC) 寄存器，使用 RDTSC 指令提取 TSC 中值。对于 ARM 64 也有类似的实现：
// Get the value of tokutime for right now. We want this to be fast, so we// expose the implementation as RDTSC.static inline tokutime_t toku_time_now(void) {#if defined(__x86_64__) || defined(__i386__) uint32_t lo, hi; __asm__ __volatile__("rdtsc" : "=a"(lo), "=d"(hi)); return (uint64_t)hi << 32 | lo;#elif defined(__aarch64__) uint64_t result; __asm __volatile__("mrs %[rt], cntvct_el0" : [ rt ] "=r"(result)); return result;#elif defined(__powerpc__) return __ppc_get_timebase();#elif defined(__s390x__) uint64_t result; asm volatile("stckf %0" : "=Q"(result) : : "cc"); return result;#else#error No timer implementation for this platform#endif}
复制代码
而在将 RocksDB 移植到龙芯的过程中，需要修改上面的代码，判断出当前是龙芯 loongarch64 架构。
网上没有搜到 GCC 对龙芯 CPU 的预定宏的文档说明，只能从源码中找答案：
voidloongarch_cpu_cpp_builtins (cpp_reader *pfile){ ... builtin_define ("__loongarch__"); ...}
复制代码
可以看到，__loongarch__代表龙芯 CPU。 在暂时不知道龙芯是否支持RDTSC的情况下，只能给出通用的实现，以后再查龙芯的 CPU 手册进行优化。
#if defined(__x86_64__) || defined(__i386__) ...#elif defined(__aarch64__) ...#elif defined(__powerpc__) ...#elif defined(__loongarch__) struct timeval tv; gettimeofday(&tv,NULL); return tv.tv_sec*(uint64_t)1000000+tv.tv_usec;#else #error No implementation for this platform#endif
复制代码
评论