nncombinator 0.3.0

A library of neural networks that can be written type-safely
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61

extern crate cc;

use std::env;
use std::process::Command;

fn find_library_paths() -> Vec<String> {
	match env::var("CUDA_PATH") {
		Ok(path) => {
			let split_char = if cfg!(target_os = "windows") { ";" } else { ":" };

			path.split(split_char).map(|s| s.to_owned()).collect::<Vec<_>>()
		}
		Err(_) => vec![],
	}
}

fn main() {
	println!("cargo:rerun-if-changed=src/kernel.cu");

	let library_paths = find_library_paths();

	let out_dir = env::var("OUT_DIR").unwrap();

	if cfg!(target_os = "windows") {
		Command::new("nvcc")
			.args(&["-O3",
				"src/kernel.cu",
				// static library (.lib) を出力させる
				"-lib",
				// Cコンパイラのパスを指定
				"-ccbin",
				"cl.exe",
				// Cコンパイラにwarning 4819を無視するよう指示
				"-Xcompiler", "-wd4819",
				"-o",
			])
			.arg(&format!("{}/kernel.lib", out_dir))
			.status()
			.unwrap();

		for p in library_paths {
			println!("cargo:rustc-link-search=native={}/lib/x64", p);
		}
		println!("cargo:rustc-link-search=native={}", out_dir);
		println!("cargo:rustc-link-lib=static=kernel");
	} else {
		cc::Build::new()
			.cuda(true)
			.flag("-cudart=shared")
			.flag("-gencode")
			.flag("arch=compute_86,code=sm_86")
			.flag("-gencode")
			.flag("arch=compute_80,code=sm_80")
			.flag("-gencode")
			.flag("arch=compute_72,code=sm_72")
			.flag("-gencode")
			.flag("arch=compute_61,code=sm_61")
			.file("src/kernel.cu")
			.compile("libkernel.a");
	}
}