oxbitnet 0.5.0

Run BitNet b1.58 ternary LLMs with wgpu
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

use std::fs;
use std::path::Path;

fn main() {
    let shader_src = Path::new("../../packages/core/src/shaders");
    // Fallback: try relative to workspace root
    let shader_src = if shader_src.exists() {
        shader_src.to_path_buf()
    } else {
        // When building from packages/rust/crates/oxbitnet
        Path::new("../../../core/src/shaders").to_path_buf()
    };

    let out_dir = Path::new("src/shaders");
    fs::create_dir_all(out_dir).expect("Failed to create shaders output dir");

    if shader_src.exists() {
        for entry in glob::glob(shader_src.join("*.wgsl").to_str().unwrap())
            .expect("Failed to glob shaders")
        {
            let path = entry.expect("Failed to read shader path");
            let filename = path.file_name().unwrap();
            let dest = out_dir.join(filename);
            fs::copy(&path, &dest).expect("Failed to copy shader");
        }
    }

    println!("cargo:rerun-if-changed=build.rs");
    if shader_src.exists() {
        println!(
            "cargo:rerun-if-changed={}",
            shader_src.to_str().unwrap()
        );
    }
}