forge-filter 0.1.0

GPU filter+compact for Apple Silicon — 10x+ over Polars on numeric WHERE clauses
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
43
44
45
46
47
48
49

use std::env;
use std::path::PathBuf;
use std::process::Command;

fn main() {
    let out_dir = PathBuf::from(env::var("OUT_DIR").unwrap());
    let profile = env::var("PROFILE").unwrap_or_default();

    let metal_file = "shaders/filter.metal";
    let air_file = out_dir.join("filter.air");

    let mut cmd = Command::new("xcrun");
    cmd.args(["-sdk", "macosx", "metal", "-std=metal3.2", "-c"]);
    if profile == "release" {
        cmd.arg("-O2");
    }
    cmd.args([metal_file, "-o", air_file.to_str().unwrap()]);

    let status = cmd.status().expect("Failed to run xcrun metal compiler");
    if !status.success() {
        panic!("Metal shader compilation failed for {}", metal_file);
    }
    println!("cargo:rerun-if-changed={}", metal_file);

    let metallib_path = out_dir.join("filter.metallib");
    let status = Command::new("xcrun")
        .args([
            "-sdk",
            "macosx",
            "metallib",
            air_file.to_str().unwrap(),
            "-o",
            metallib_path.to_str().unwrap(),
        ])
        .status()
        .expect("Failed to run xcrun metallib linker");
    if !status.success() {
        panic!("Metal library linking failed");
    }

    println!(
        "cargo:rustc-env=FILTER_METALLIB_PATH={}",
        metallib_path.display()
    );
    println!(
        "cargo:warning=Built filter.metallib at {}",
        metallib_path.display()
    );
}