mlxrs 0.1.0

Safe Rust bindings for Apple's MLX array framework, with LM, VLM, audio, and embeddings support
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

use std::env::{self, var};

// Tarpaulin cfg detection only. ALL tokenizer codegen moved out of this
// build script into the `xtask` crate so a normal `mlxrs` build never
// compiles `tokenizers`/`serde_json`/`toml`. The regenerated artifacts are
// committed source:
//   - mlxrs/src/tokenizer/generated.rs
//   - mlxrs/tests/fixtures/tokenizer.json
//   - mlxrs/tests/fixtures/tokenizer_config.json
// `cargo xtask-codegen --check` (alias for
// `cargo run -p xtask --features codegen -- codegen --check`) is the drift
// guard against the `mlxrs/data/tokenizer/` source-of-truth. The `codegen`
// subcommand is gated behind xtask's `codegen` feature, so bare
// `cargo xtask codegen` intentionally fails ("requires the codegen
// feature"); use the alias. This script has NO `[build-dependencies]`.
fn main() {
  // Don't rerun this on changes other than build.rs, as we only depend on
  // the rustc version.
  println!("cargo:rerun-if-changed=build.rs");

  // Check for `--features=tarpaulin`.
  let tarpaulin = var("CARGO_FEATURE_TARPAULIN").is_ok();

  if tarpaulin {
    use_feature("tarpaulin");
  } else {
    // Always rerun if these env vars change.
    println!("cargo:rerun-if-env-changed=CARGO_TARPAULIN");
    println!("cargo:rerun-if-env-changed=CARGO_CFG_TARPAULIN");

    // Detect tarpaulin by environment variable
    if env::var("CARGO_TARPAULIN").is_ok() || env::var("CARGO_CFG_TARPAULIN").is_ok() {
      use_feature("tarpaulin");
    }
  }

  // Rerun this script if any of our features or configuration flags change,
  // or if the toolchain we used for feature detection changes.
  println!("cargo:rerun-if-env-changed=CARGO_FEATURE_TARPAULIN");
}

fn use_feature(feature: &str) {
  println!("cargo:rustc-cfg={}", feature);
}