cubecl-server 0.11.0-pre.4

Toolkit for implementing a CubeCL runtime: memory pools, streams, drivers and the compilation pipeline.
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86

[package]
authors = ["louisfd <louisfd94@gmail.com>", "Nathaniel Simard"]
categories = ["science"]
description = "Toolkit for implementing a CubeCL runtime: memory pools, streams, drivers and the compilation pipeline."
edition.workspace = true
keywords = ["deep-learning", "machine-learning", "data"]
license.workspace = true
name = "cubecl-server"
readme.workspace = true
repository = "https://github.com/tracel-ai/cubecl/tree/main/crates/cubecl-server"
version.workspace = true


[lints]
workspace = true


[features]
autotune-checks = ["cubecl-runtime/autotune-checks"]
default = ["std", "storage-bytes", "persistence", "cubecl-runtime/default"]
exclusive-memory-only = ["cubecl-runtime/exclusive-memory-only"]
persistence = ["std", "cubecl-runtime/persistence"]
profile-tracy = ["cubecl-runtime/profile-tracy", "dep:tracy-client"]
std = ["cubecl-runtime/std", "cubecl-common/std", "cubecl-environment/std", "thiserror/std"]
storage-bytes = []
tokio = ["cubecl-runtime/tokio"]

tracing = ["dep:tracing", "cubecl-runtime/tracing", "cubecl-common/tracing", "cubecl-ir/tracing"]

[dependencies]
# Runtime Deps
log = { workspace = true }
tracing = { workspace = true, features = ["attributes"], optional = true }

bytemuck = { workspace = true }
cfg-if = { workspace = true }
cubecl-common = { path = "../cubecl-common", version = "=0.11.0-pre.4", default-features = false, features = [
    "serde",
    "hash",
] }
cubecl-environment = { path = "../cubecl-environment", version = "=0.11.0-pre.4", default-features = false }
cubecl-ir = { path = "../cubecl-ir", version = "=0.11.0-pre.4", default-features = false, features = [
    "serde",
] }
cubecl-runtime = { path = "../cubecl-runtime", version = "=0.11.0-pre.4", default-features = false }
cubecl-zspace = { path = "../cubecl-zspace", version = "=0.11.0-pre.4", default-features = false }

derive-new = { workspace = true }
derive_more = { workspace = true, features = ["eq"] }
hashbrown = { workspace = true }
itertools = { workspace = true }
pliron = { workspace = true }
serde = { workspace = true, features = ["derive"] }
smallvec = { workspace = true }

foldhash = { workspace = true }
buildid = { workspace = true }
thiserror = { workspace = true }

futures-util = { workspace = true, default-features = false }

# Persistent cache deps - has to match the cfg(std_io) cfg.
[target.'cfg(any(target_os = "windows", target_os = "linux", target_os = "macos", target_os = "android"))'.dependencies]
cubecl-environment = { path = "../cubecl-environment", version = "=0.11.0-pre.4", default-features = false }
serde_json = { workspace = true, features = ["std"] }

# Tracy if enabled.
tracy-client = { workspace = true, optional = true }

[dev-dependencies]
rand = { workspace = true, features = ["thread_rng"] }
serde_json = { workspace = true, features = ["std"] }
serial_test = { workspace = true }
tempfile = { workspace = true }
test-log = { workspace = true, features = ["trace"] }

[build-dependencies]
cfg_aliases = { workspace = true }

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

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