-
Notifications
You must be signed in to change notification settings - Fork 98
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Neptune now also runs on CUDA. Closes #104.
- Loading branch information
Showing
19 changed files
with
716 additions
and
333 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,83 @@ | ||
/// The build script is needed to compile the CUDA kernel. | ||
|
||
#[cfg(feature = "cuda")] | ||
fn main() { | ||
use std::path::PathBuf; | ||
use std::process::Command; | ||
use std::{env, fs}; | ||
|
||
use blstrs::Scalar as Fr; | ||
use ec_gpu_gen::Limb32; | ||
use sha2::{Digest, Sha256}; | ||
|
||
#[path = "src/proteus/sources.rs"] | ||
mod sources; | ||
|
||
let kernel_source = sources::generate_program::<Fr, Limb32>(); | ||
let out_dir = env::var("OUT_DIR").expect("OUT_DIR was not set."); | ||
|
||
// Make it possible to override the default options. Though the source and output file is | ||
// always set automatically. | ||
let mut nvcc = match env::var("NEPTUNE_CUDA_NVCC_ARGS") { | ||
Ok(args) => execute::command(format!("nvcc {}", args)), | ||
Err(_) => { | ||
let mut command = Command::new("nvcc"); | ||
command | ||
.arg("--optimize=6") | ||
.arg("--fatbin") | ||
.arg("--gpu-architecture=sm_86") | ||
.arg("--generate-code=arch=compute_86,code=sm_86") | ||
.arg("--generate-code=arch=compute_80,code=sm_80") | ||
.arg("--generate-code=arch=compute_75,code=sm_75"); | ||
command | ||
} | ||
}; | ||
|
||
// Hash the source and and the compile flags. Use that as the filename, so that the kernel is | ||
// only rebuilt if any of them change. | ||
let mut hasher = Sha256::new(); | ||
hasher.update(kernel_source.as_bytes()); | ||
hasher.update(&format!("{:?}", &nvcc)); | ||
let kernel_digest = hex::encode(hasher.finalize()); | ||
|
||
let source_path: PathBuf = [&out_dir, &format!("{}.cu", &kernel_digest)] | ||
.iter() | ||
.collect(); | ||
let fatbin_path: PathBuf = [&out_dir, &format!("{}.fatbin", &kernel_digest)] | ||
.iter() | ||
.collect(); | ||
|
||
fs::write(&source_path, &kernel_source).unwrap_or_else(|_| { | ||
panic!( | ||
"Cannot write kernel source at {}.", | ||
source_path.to_str().unwrap() | ||
) | ||
}); | ||
|
||
// Only compile if the output doesn't exist yet. | ||
if !fatbin_path.as_path().exists() { | ||
let status = nvcc | ||
.arg("--output-file") | ||
.arg(&fatbin_path) | ||
.arg(&source_path) | ||
.status() | ||
.expect("Cannot run nvcc."); | ||
|
||
if !status.success() { | ||
panic!( | ||
"nvcc failed. See the kernel source at {}", | ||
source_path.to_str().unwrap() | ||
); | ||
} | ||
} | ||
|
||
// The idea to put the path to the farbin into a compile-time env variable is from | ||
// https://github.com/LutzCle/fast-interconnects-demo/blob/b80ea8e04825167f486ab8ac1b5d67cf7dd51d2c/rust-demo/build.rs | ||
println!( | ||
"cargo:rustc-env=CUDA_FATBIN={}", | ||
fatbin_path.to_str().unwrap() | ||
); | ||
} | ||
|
||
#[cfg(not(feature = "cuda"))] | ||
fn main() {} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.