Skip to content

Commit

Permalink
[Important] Rewrite build.rs for russell_sparse
Browse files Browse the repository at this point in the history
  • Loading branch information
@cpmech
cpmech committed Apr 29, 2024
1 parent 5033e92 commit afdd5e3
Showing 1 changed file with 102 additions and 60 deletions.
162 changes: 102 additions & 60 deletions russell_sparse/build.rs
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
// validate intel source command --------------------------------------------------------------------------

#[cfg(feature = "intel_mkl")]
use std::env;

Expand All @@ -12,88 +14,128 @@ fn validate_intel_setvars_completed() {
}
}

#[cfg(not(feature = "intel_mkl"))] // OpenBLAS + OpenMPI
#[cfg(not(feature = "intel_mkl"))]
fn validate_intel_setvars_completed() {}

#[cfg(feature = "intel_mkl")]
fn get_mpi_include_dirs() -> Vec<String> {
vec!["/opt/intel/oneapi/mpi/latest/include/".to_string()]
// information --------------------------------------------------------------------------------------------

// (default) Returns the directories and libraries
// returns `(inc_dirs, lib_dirs, libs)`
#[cfg(not(feature = "local_libs"))]
#[cfg(not(feature = "intel_mkl"))]
fn get_information() -> (Vec<&'static str>, Vec<&'static str>, Vec<&'static str>) {
(
// inc_dirs
vec![
"/usr/lib/x86_64-linux-gnu/openmpi/include", // Ubuntu
"/usr/include/openmpi-x86_64", // Rocky
"/usr/include", // Ubuntu
"/usr/include/MUMPS", // Rocky
"/usr/include/suitesparse", // Ubuntu + Rocky
],
// lib_dirs
vec![
"/usr/lib/x86_64-linux-gnu/openmpi/lib", // Ubuntu
"/usr/lib64/openmpi/lib", // Rocky
],
// libs
vec![
"mpi", //
"dmumps", //
"zmumps", //
"umfpack", //
"klu", //
],
)
}

#[cfg(not(feature = "intel_mkl"))] // OpenBLAS + OpenMPI
fn get_mpi_include_dirs() -> Vec<String> {
vec!["/usr/lib/x86_64-linux-gnu/openmpi/include".to_string()]
// (local_libs) Returns the directories and libraries
// returns `(inc_dirs, lib_dirs, libs)`
#[cfg(feature = "local_libs")]
#[cfg(not(feature = "intel_mkl"))]
fn get_information() -> (Vec<&'static str>, Vec<&'static str>, Vec<&'static str>) {
(
// inc_dirs
vec![
"/usr/lib/x86_64-linux-gnu/openmpi/include", // Ubuntu
"/usr/local/include/mumps", // Ubuntu
"/usr/local/include/umfpack", // Ubuntu
],
// lib_dirs
vec![
"/usr/lib/x86_64-linux-gnu/openmpi/lib", // Ubuntu
"/usr/local/lib/mumps", // Ubuntu
"/usr/local/lib/umfpack", // Ubuntu
],
// libs
vec![
"mpi", //
"dmumps_cpmech", //
"zmumps_cpmech", //
"umfpack", //
"klu", //
],
)
}

// (intel_mkl) Returns the directories and libraries
// returns `(inc_dirs, lib_dirs, libs)`
#[cfg(feature = "intel_mkl")]
fn get_mpi_link_dirs() -> Vec<String> {
vec!["/opt/intel/oneapi/mpi/latest/lib/".to_string()]
fn get_information() -> (Vec<&'static str>, Vec<&'static str>, Vec<&'static str>) {
(
// inc_dirs
vec![
"/opt/intel/oneapi/mpi/latest/include", // Ubuntu
"/usr/local/include/mumps", // Ubuntu
"/usr/local/include/umfpack", // Ubuntu
],
// lib_dirs
vec![
"/opt/intel/oneapi/mpi/latest/lib", // Ubuntu
"/usr/local/lib/mumps", // Ubuntu
"/usr/local/lib/umfpack", // Ubuntu
],
// libs
vec![
"mpi", //
"dmumps_cpmech", //
"zmumps_cpmech", //
"umfpack", //
"klu", //
],
)
}

#[cfg(not(feature = "intel_mkl"))] // OpenBLAS + OpenMPI
fn get_mpi_link_dirs() -> Vec<String> {
vec!["/usr/lib/x86_64-linux-gnu/openmpi".to_string()]
}
// main ---------------------------------------------------------------------------------------------------

#[cfg(feature = "local_libs")]
fn compile_libs() {
// local MUMPS
cc::Build::new()
.file("c_code/interface_complex_mumps.c")
.file("c_code/interface_mumps.c")
.includes(get_mpi_include_dirs())
.include("/usr/local/include/mumps")
.compile("c_code_interface_mumps");
for d in get_mpi_link_dirs() {
println!("cargo:rustc-link-search=native={d}");
}
println!("cargo:rustc-link-search=native=/usr/local/lib/mumps");
println!("cargo:rustc-link-lib=dylib=dmumps_cpmech");
println!("cargo:rustc-link-lib=dylib=zmumps_cpmech");
println!("cargo:rustc-link-lib=dylib=mpi");
println!("cargo:rustc-cfg=local_mumps");
// local UMFPACK and KLU
cc::Build::new()
.file("c_code/interface_complex_umfpack.c")
.file("c_code/interface_complex_klu.c")
.file("c_code/interface_umfpack.c")
.file("c_code/interface_klu.c")
.include("/usr/local/include/umfpack")
.compile("c_code_interface_umfpack");
println!("cargo:rustc-link-search=native=/usr/local/lib/umfpack");
println!("cargo:rustc-link-lib=dylib=umfpack");
println!("cargo:rustc-link-lib=dylib=klu");
println!("cargo:rustc-cfg=local_umfpack");
}
fn main() {
// validate intel setvars
validate_intel_setvars_completed();

// information
let (inc_dirs, lib_dirs, libs) = get_information();

#[cfg(not(feature = "local_libs"))] // Libraries from the distribution
fn compile_libs() {
// MUMPS
cc::Build::new()
.file("c_code/interface_complex_mumps.c")
.file("c_code/interface_mumps.c")
.includes(get_mpi_include_dirs())
.include("/usr/include/mumps")
.includes(&inc_dirs)
.compile("c_code_interface_mumps");
for d in get_mpi_link_dirs() {
println!("cargo:rustc-link-search=native={d}");
}
println!("cargo:rustc-link-lib=dylib=dmumps");
println!("cargo:rustc-link-lib=dylib=zmumps");
println!("cargo:rustc-link-lib=dylib=mpi");

// UMFPACK and KLU
cc::Build::new()
.file("c_code/interface_complex_umfpack.c")
.file("c_code/interface_complex_klu.c")
.file("c_code/interface_umfpack.c")
.file("c_code/interface_klu.c")
.include("/usr/include/suitesparse")
.includes(&inc_dirs)
.compile("c_code_interface_umfpack");
println!("cargo:rustc-link-lib=dylib=umfpack");
println!("cargo:rustc-link-lib=dylib=klu");
}

fn main() {
validate_intel_setvars_completed();
compile_libs();
// libraries
for d in &lib_dirs {
println!("cargo:rustc-link-search=native={}", *d);
}
for l in &libs {
println!("cargo:rustc-link-lib=dylib={}", *l);
}
}

0 comments on commit afdd5e3

Please sign in to comment.