singe-cutensor-sys 0.1.0-alpha.7

Low-level FFI bindings for the NVIDIA cuTENSOR tensor algebra library.
Documentation 
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

use singe_cuda_find::{
    emit_dependency_metadata, emit_link_libraries, emit_rerun_if_env_changed,
    emit_static_link_libraries, find_cuda_runtime, find_cutensor, find_nccl,
};

fn main() {
    println!("cargo:rerun-if-changed=build.rs");
    emit_rerun_if_env_changed();

    link_libraries();
}

fn link_libraries() {
    if emit_dependency_metadata(find_cuda_runtime()).is_some() {
        emit_static_link_libraries(&["cudart_static"]);
    }

    if emit_dependency_metadata(find_cutensor()).is_some() {
        emit_link_libraries(&["cutensor", "cutensorMg", "cutensorMp"]);
    }

    if emit_dependency_metadata(find_nccl()).is_some() {
        emit_link_libraries(&["nccl"]);
    }
}