【发布时间】:2014-10-09 14:04:57
【问题描述】:
我想了解一些关于 rust 任务的知识,所以我做了一个蒙特卡罗计算 PI。现在我的困惑是为什么单线程 C 版本快 4 倍 比 4 路线程 Rust 版本。很明显我做错了什么,或者我的心理表现模型离题了。
这是 C 版本:
#include <stdlib.h>
#include <sys/types.h>
#include <unistd.h>
#include <stdio.h>
#define PI 3.1415926535897932
double monte_carlo_pi(int nparts)
{
int i, in=0;
double x, y;
srand(getpid());
for (i=0; i<nparts; i++) {
x = (double)rand()/(double)RAND_MAX;
y = (double)rand()/(double)RAND_MAX;
if (x*x + y*y < 1.0) {
in++;
}
}
return in/(double)nparts * 4.0;
}
int main(int argc, char **argv)
{
int nparts;
double mc_pi;
nparts = atoi(argv[1]);
mc_pi = monte_carlo_pi(nparts);
printf("computed: %f error: %f\n", mc_pi, mc_pi - PI);
}
Rust 版本不是逐行移植:
use std::rand;
use std::rand::distributions::{IndependentSample,Range};
fn monte_carlo_pi(nparts: uint ) -> uint {
let between = Range::new(0f64,1f64);
let mut rng = rand::task_rng();
let mut in_circle = 0u;
for _ in range(0u, nparts) {
let a = between.ind_sample(&mut rng);
let b = between.ind_sample(&mut rng);
if a*a + b*b <= 1.0 {
in_circle += 1;
}
}
in_circle
}
fn main() {
let (tx, rx) = channel();
let ntasks = 4u;
let nparts = 100000000u; /* I haven't learned how to parse cmnd line args yet!*/
for _ in range(0u, ntasks) {
let child_tx = tx.clone();
spawn(proc() {
child_tx.send(monte_carlo_pi(nparts/ntasks));
});
}
let result = rx.recv() + rx.recv() + rx.recv() + rx.recv();
println!("pi is {}", (result as f64)/(nparts as f64)*4.0);
}
C 版本的构建和计时:
$ clang -O2 mc-pi.c -o mc-pi-c; time ./mc-pi-c 100000000
computed: 3.141700 error: 0.000108
./mc-pi-c 100000000 1.68s user 0.00s system 99% cpu 1.683 total
构建 Rust 版本并计时:
$ rustc -v
rustc 0.12.0-nightly (740905042 2014-09-29 23:52:21 +0000)
$ rustc --opt-level 2 --debuginfo 0 mc-pi.rs -o mc-pi-rust; time ./mc-pi-rust
pi is 3.141327
./mc-pi-rust 2.40s user 24.56s system 352% cpu 7.654 tota
【问题讨论】:
-
不要在打开调试符号的情况下编译。
-
the single-threaded C version is 4 times slower than the 4-way threaded Rust version。您发布的数字似乎与此相反 -
@RobLatham 这里的瓶颈可能是随机数生成器。尝试使用
rand::XorShiftRng::new_unseeded()而不是rand::task_rng()以获得更快的随机生成器。 -
您可以创建一个随机播种的
XorShiftRng,例如let mut rng: XorShiftRng = rand::random();(速度的提升来自于算法的改变,而不是缺少种子)。 -
现在更像了。在 4 核系统上,四路线程 rust 比单线程 C 版本快 4.5 倍。如果 Dogbert 想把它写下来,我会接受它,或者我会在几天后自行回答。
标签: c performance rust