tvm-build 0.2.4

A CLI and library for programmatically building TVM.
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

mod target;

pub use target::Target;

pub fn local_target() -> Target {
    let platform = futures::executor::block_on(heim::host::platform()).unwrap();
    match platform.system() {
        "Darwin" => {
            let cmake_defines = match platform.architecture() {
                heim::host::Arch::Unknown | heim::host::Arch::AARCH64 => {
                    vec![("CMAKE_OSX_ARCHITECTURES".into(), "arm64".into())]
                }
                _ => vec![],
            };

            Target {
                host: "Darwin".into(),
                target_str: "arm64-apple-darwin".into(),
                cmake_defines,
            }
        }
        "Linux" => {
            let target = match platform.architecture() {
                heim::host::Arch::AARCH64 => "aarch64-unknown-linux-gnu",
                heim::host::Arch::X86_64 => "x86_64-unknown-linux-gnu",
                _ => panic!("not supported"),
            };

            Target {
                host: "Linux".into(),
                target_str: target.into(),
                cmake_defines: vec![],
            }
        }
        _ => {
            panic!(
                "Platform `{}` unsupported, please check the issue tracker.",
                platform.system()
            );
        }
    }
}