Geodesic-Attention-Engine-GAE-/Cargo.toml at main · RegularJoe-CEO/Geodesic-Attention-Engine-GAE- · GitHub

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
[package]
name = "ate"
version = "1.0.0"
edition = "2021"
authors = ["Eric Waller <e@ewaller.com>"]
description = "Geodesic Attention Engine - Physics-based AI compute optimization"
license = "Proprietary"

[lib]
path = "rust_src/src/lib.rs"

[[bin]]
name = "ate"
path = "rust_src/src/main.rs"

[features]
default = ["std", "rayon"]
std = []
rayon = ["dep:rayon"]
wgpu = ["dep:wgpu", "dep:bytemuck", "dep:pollster"]

[dependencies]
thiserror = "1.0"
rayon = { version = "1.10", optional = true }
wgpu = { version = "23.0", optional = true }
bytemuck = { version = "1.14", features = ["derive"], optional = true }
pollster = { version = "0.3", optional = true }

[dev-dependencies]
criterion = { version = "0.5", features = ["html_reports"] }
rand = "0.8"
approx = "0.5"

[[bench]]
name = "attention_benchmark"
harness = false

[profile.release]
lto = true
codegen-units = 1