stensor 0.1.1

Cross-platform GPU tensor library with Slang and Rust.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

use minislang::{SlangCompiler, shader_slang::CompileTarget};
use std::path::PathBuf;
use std::str::FromStr;

pub fn main() {
    let slang = SlangCompiler::new(vec![PathBuf::from_str("./shaders").unwrap()]);

    let targets = [
        CompileTarget::Wgsl,
        #[cfg(feature = "cuda")]
        CompileTarget::CudaSource,
    ];

    for target in targets {
        slang.compile_all(target, "../shaders", "./src/autogen", &[]);
    }
}