Linux内核级最大内存带宽检测方法咨询

以下是几种可行的内核层（或内核模块）实现方案：

1. 借助硬件性能监控单元(PMU)实现被动检测 #

现代CPU（如Intel、AMD）的内存控制器内置性能计数器，可直接统计内存读写的字节数或事务数，内核模块可通过访问模型特定寄存器(MSR)或内核PMU接口获取这些数据，进而计算总带宽。

• 核心思路：读取内存控制器相关事件计数器（如Intel的UNC_M_CAS_COUNT.READ/UNC_M_CAS_COUNT.WRITE，AMD的北桥内存读写事件），记录采样周期内的计数差值，结合内存事务对应的字节大小（如DDR4每次CAS对应64字节），计算总吞吐量：带宽 = (读字节数 + 写字节数) / 采样时间。
• 实现要点：内核模块中需先通过cpuid指令检测CPU是否支持目标MSR，再用rdmsr/wrmsr操作寄存器；需注意权限控制，部分MSR需内核态权限才能访问。

2. 基于内核内存统计接口的近似估算 #

利用内核内置的内存分配/回收统计数据，通过周期性采样差值计算内存吞吐量：

• 核心思路：读取内核vmstat结构体中的nr_page_alloc、nr_page_free等统计字段，或page_alloc tracepoint的计数，每隔固定时间取差值，乘以系统页大小（如4KB、2MB）得到这段时间内的内存数据量，再除以时间得到带宽。
• 优缺点：实现简单，无需硬件依赖，但精度较低，仅能反映内存页级别的操作吞吐量，无法区分缓存命中与内存总线操作。

3. 内核模块主动基准测试最大带宽 #

通过在内核模块中创建后台线程，执行无缓存的连续内存读写操作，直接测量系统内存总线的最大带宽：

• 核心实现：
  1. 用kmalloc或__get_free_pages分配大尺寸物理连续缓冲区（需开启CONFIG_CONTIG_ALLOC）；
  2. 使用非临时存储指令（如x86的MOVNTI）绕过CPU缓存，确保操作直接作用于内存总线；
  3. 绑定线程到单个CPU（通过set_cpus_allowed_ptr）避免调度干扰，用ktime_get_ns()精确计时；
  4. 循环执行读写操作后，计算总数据量与时间的比值得到最大带宽。
• 示例代码片段：

  #include <linux/ktime.h>
  #include <linux/sched.h>
  
  #define BUF_SIZE (1024 * 1024 * 64) // 64MB缓冲区
  static int __init mem_bw_test_init(void) {
      int *buf = (int *)__get_free_pages(GFP_KERNEL, get_order(BUF_SIZE));
      if (!buf) return -ENOMEM;
  
      struct task_struct *task = kthread_run([](void *data) {
          int *buf = data;
          u64 start = ktime_get_ns();
          for (size_t i = 0; i < BUF_SIZE / sizeof(int); i++) {
              // 非临时写操作，绕过缓存
              asm volatile("movnti %0, %1" :: "r"(i), "m"(buf[i]) : "memory");
          }
          mb(); // 确保所有写操作完成
          u64 end = ktime_get_ns();
          u64 bytes = BUF_SIZE;
          u64 bw_mbps = (bytes * 8 * 1000000ULL) / (end - start); // 转换为Mbps
          pr_info("Max memory write bandwidth: %llu Mbps\n", bw_mbps);
          return 0;
      }, buf, "mem-bw-test");
  
      if (IS_ERR(task)) {
          free_pages((unsigned long)buf, get_order(BUF_SIZE));
          return PTR_ERR(task);
      }
      return 0;
  }
  module_init(mem_bw_test_init);
  

• 优缺点：能直接测量系统最大内存带宽，但会占用一定内存和CPU资源，仅适合离线测试场景。

4. 利用内核tracepoint统计全系统内存流量 #

通过注册内核mm_page_alloc、mm_page_free或page_fault等tracepoint的回调函数，被动统计全系统（用户态+内核态）的内存操作流量：

• 核心思路：在回调函数中累加每次内存操作对应的字节数（如页大小），定时计算采样周期内的总数据量，进而得到带宽。
• 实现要点：需在模块中使用tracepoint_probe_register注册探针，注意过滤不必要的事件（如内核内部小内存分配）以减少开销。

内容的提问来源于stack exchange，提问作者Mohammad Siavashi