torch-build 0.1.0

Utilities to link libtorch FFI interface.
Documentation 
1
2
3
4
5
6
7

use anyhow::Result;

fn main() -> Result<()> {
    let arches = torch_build::cuda::cuda_arches()?;
    dbg!(arches);
    Ok(())
}