本文研究了基于 Rust 具有的所有权语义的一些优化。

程序的机器级表示

采用下面这条指令可以让 Rust 编译器生成汇编代码文本：

rustc filename.rs --crate-type=lib --emit=asm -C opt-level=2

其中 --crate-type=lib 是为了以库的方式编译，这样我们就不必定义 main 函数。 opt-level 设置编译级别。由于 Rust 具有较多的零开销抽象层，至少要开级别 2 的优化，编译器才会将诸如 into_iter 这样的函数调用优化掉，我们才能看到比较清晰的汇编代码。

指针别名

在 Rust 中，两个可变的引用（&mut T）不能指向同一块内存空间。这使得 Rust 编译器可以告诉 LLVM 后端某个指针不存在别名（noalias）。这可以开启一些 C 语言中不可行的优化。例如下面的代码：

pub fn add(a: &mut i64, b: &mut i64) {
    *a += *b;
    *a += *b;
}

将会产生下面的编译输出：

; a in %rdi, b in %rsi
ZN1a3add17h058f239ac4f807c2E:
    movq    (%rsi), %rax
    addq    %rax, %rax
    addq    %rax, (%rdi)
    retq

这相当于

pub fn add2(a: &mut i64, b: &mut i64) {
    *a += 2 * *b;
}

注意到如果 a 与 b 能够指向相同的变量 c \(=x\)，那么 add 中每一行都使 c 翻倍。add(a, b) 最终使 c 变为 \(4x\)， add2(a, b) 使 c 变为 \(2x\)，那么这个优化会改变函数的行为。但是 Rust 可以保证可变引用是独占的，即 a != b，所以 Rust 可以做这种优化。¹

不必要的内存引用

对于 CPU 而言，访问内存显然比访问寄存器更慢。考虑下面的 C 语言循环：

typedef struct {
    long len;
    double *data;
} vec_rec, *vec_ptr;

void combine3(vec_ptr v, double *dest) {
    long length = v->len;
    double *data = v->data;

    *dest = 0;
    for (long i = 0; i < length; i++) {
        *dest += data[i];
    }
}

在循环中，dest 所指向的内存被频繁访问，造成程序低效。然而由于前述指针别名的问题，编译器无法无法保证 dest 不指向 v 中的某个元素，无法对此做优化。在 Rust 中，同样功能的代码

pub fn combine1(vec: &Vec<Data>, dest: &mut Data) {
    *dest = 0;
    for a in vec {
        *dest += a;
    }
}

循环部分编译为如下汇编（优化级别 2）：

; dest in %rsi
.LBB1_8:
	addsd	(%rax), %xmm0
	addsd	8(%rax), %xmm0
	addsd	16(%rax), %xmm0
	addsd	24(%rax), %xmm0
	addsd	32(%rax), %xmm0
	addsd	40(%rax), %xmm0
	addsd	48(%rax), %xmm0
	addsd	56(%rax), %xmm0
	addq	$64, %rax
	cmpq	%rcx, %rax
	jne	.LBB1_8
.LBB1_9:
	movsd	%xmm0, (%rsi)   ; write %xmm0 to *dest
	retq

可以看到所求值被累加进了 %xmm0 寄存器，最后才被写入 %rsi 指向的内存。这样就获得了更佳的性能。²

我同时注意到，将 *dest += a 换为 *dest = *dest + a 后，编译器就不能做这个优化了。可见复合赋值语句不仅能让我们少打几个字符，还能帮助编译器优化。

Footnote