torch-build 0.1.0

Utilities to link libtorch FFI interface.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

use anyhow::Result;
use std::{env, fs, path::PathBuf, str};

fn main() -> Result<()> {
    let out_dir = PathBuf::from(env::var_os("OUT_DIR").unwrap());

    let use_cxx11_abi = {
        let bytes = cc::Build::new()
            .cpp(true)
            .warnings(false)
            .file("csrc/test_cxx11_abi.cpp")
            .expand();
        str::from_utf8(&bytes).unwrap().contains("YES")
    };

    if use_cxx11_abi {
        let path = out_dir.join("use_cxx11_abi");
        fs::File::create(path)?;
    }

    Ok(())
}