使用此RISC Zero性能分析器如履平地

本仓库提供了一个RISC Zero程序的插件，它可以计算程序不同部分贡献的周期数，检测导致大量周期数的执行步骤，并解释其根本原因。

开发者可以在程序中添加 start_timer!、stop_start_timer! 和 stop_timer! 来跟踪周期的来源。以下是一个示例。

start_timer!("Load data");
......

    start_timer!("Read from the host");
    ......

    stop_start_timer!("Check the length");
    ......

    stop_start_timer!("Hash");
    ......

    stop_timer!();

stop_timer!();

分析器将输出有关周期分解的彩色信息。具体来说，如果分析器看到一个执行步骤，尽管如此，但导致了大量周期，它将突出显示并找出根本原因。

有人可能会问，我们为什么说这个分析器是“如履平地”。这是因为，与基于 eprintln! 的先前解决方案不同，分析器本身非常努力地不干扰程序的原始执行，特别是周期计数。

如何使用？

在RISC Zero程序的主机和虚拟机都需要进行必要的更改。

主机

主机应使用主机crate l2r0-profiler-host 并使用 ExecutorEnv 来运行程序。

let cycle_tracer = Rc::new(RefCell::new(CycleTracer::default()));

let env = ExecutorEnv::builder()
        .write_slice(&task.a)
        .write_slice(&task.b)
        .write_slice(&task.long_form_c)
        .write_slice(&task.k)
        .write_slice(&task.long_form_kn)
        .trace_callback(|e| {
            cycle_tracer.borrow_mut().handle_event(e);
            Ok(())
        })
        .build()
        .unwrap();

let mut exec = ExecutorImpl::from_elf(env, METHOD_ELF).unwrap();
let _ = exec.run().unwrap();

cycle_tracer.borrow().print();

在上面的示例中，我们首先创建周期跟踪器。

let cycle_tracer = Rc::new(RefCell::new(CycleTracer::default()));

然后，我们使用 trace_callback 请求 ExecutorEnv 将执行跟踪发送回分析器。

.trace_callback(|e| {
    cycle_tracer.borrow_mut().handle_event(e);
    Ok(())
})

执行完成后，请求周期跟踪器输出分析结果。

cycle_tracer.borrow().print();

虚拟机

虚拟机也有自己的crate，l2r0-profiler-guest。导入它，并记得开启 print-trace 功能。

当程序启动时，如下所示。

fn main() {
    l2r0_profiler_guest::init_trace_logger();
    start_timer!("Total");
    ......
    stop_timer!();
}

我们首先初始化跟踪记录器。

l2r0_profiler_guest::init_trace_logger();

然后，虚拟机可以使用宏将程序分解成更小的部分进行检查。

它如何工作？

分析器的工作方式类似于硬件观察点。

当客户程序启动时，客户端周期跟踪器使用一些虚拟指令（写入零寄存器）来通知主机端周期跟踪器监视的缓冲区。代码在这里：这里。

#[inline(always)]
pub fn init_trace_logger() {
    unsafe {
        core::arch::asm!(
            r#"
            nop
            li x0, 0xCDCDCDCD
            la x0, TRACE_MSG_CHANNEL
            la x0, TRACE_MSG_LEN_CHANNEL
            la x0, TRACE_SIGNAL_CHANNEL
            nop
        "#
        );
    }
}

然后，主机端周期跟踪器将监视这三个通道。如果程序将这些数据写入这些内存位置，主机端周期跟踪器可以捕捉这些变化并获取通道中的信息。

例如，

• start_timer!(msg) 将消息 msg 复制到 TRACE_MSG_CHANNEL，并将消息长度写入 TRACE_MSG_LEN_CHANNEL，这会触发主机端周期跟踪器标记新的计时器已开始。
• end_timer!() 将零写入 TRACE_SIGNAL_CHANNEL，这会触发主机端周期跟踪器标记之前的计时器已停止。

这两个计时器都设计得非常简洁，因为我们希望它们不会占用太多的周期。这与之前在客户端使用 eprintln!("{}", env::get_cycle_count()); 的方法相比，这本身就会创建很多周期并影响计算。