cubecl-cpu 0.9.0-pre.2

CPU runtime for CubeCL
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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135

[build-dependencies.tracel-llvm-bundler]
version = "20.1.4-5"

[dependencies.bytemuck]
version = "1.16.1"

[dependencies.cubecl-common]
default-features = false
version = "0.9.0-pre.2"

[dependencies.cubecl-convolution]
features = ["export_tests"]
version = "0.9.0-pre.2"

[dependencies.cubecl-core]
default-features = false
version = "0.9.0-pre.2"

[dependencies.cubecl-matmul]
features = ["export_tests"]
version = "0.9.0-pre.2"

[dependencies.cubecl-opt]
default-features = false
version = "0.9.0-pre.2"

[dependencies.cubecl-reduce]
features = ["export_tests"]
version = "0.9.0-pre.2"

[dependencies.cubecl-runtime]
default-features = false
features = ["channel-mutex"]
version = "0.9.0-pre.2"

[dependencies.cubecl-std]
default-features = false
version = "0.9.0-pre.2"

[dependencies.derive-new]
default-features = false
version = "0.7.0"

[dependencies.half]
default-features = false
features = ["alloc", "num-traits", "serde"]
version = "2.5"

[dependencies.log]
default-features = false
version = "0.4.22"

[dependencies.serde]
default-features = false
features = ["derive", "alloc"]
version = "1.0.204"

[dependencies.sysinfo]
version = "0.36.1"

[dependencies.tracel-llvm]
features = ["mlir-helpers"]
version = "20.1.4-5"

[dev-dependencies.cubecl-core]
features = ["export_tests"]
version = "0.9.0-pre.2"

[dev-dependencies.cubecl-random]
features = ["export_tests"]
version = "0.9.0-pre.2"

[dev-dependencies.cubecl-reduce]
features = ["export_tests"]
version = "0.9.0-pre.2"

[dev-dependencies.cubecl-std]
features = ["export_tests"]
version = "0.9.0-pre.2"

[dev-dependencies.paste]
version = "1"

[dev-dependencies.pretty_assertions]
version = "1.4"

[features]
conv_tests = ["cubecl-convolution/conv_tests"]
default = ["std", "cubecl-runtime/default", "cubecl-common/default", "cubecl-core/default"]
matmul_tests_all = ["matmul_tests_unit", "matmul_tests_plane", "matmul_tests_vecmat", "matmul_tests_tma", "matmul_tests_double", "matmul_tests_simple", "matmul_tests_ordered", "matmul_tests_cyclic", "matmul_tests_strided", "matmul_tests_tilewise", "matmul_tests_hybrid", "matmul_tests_barrier", "matmul_tests_specialized", "matmul_tests_f16", "matmul_tests_f32", "matmul_tests_layouts", "matmul_tests_alt_shapes", "matmul_tests_partition_buffering", "matmul_tests_hypercube"]
matmul_tests_alt_shapes = ["cubecl-matmul/matmul_tests_alt_shapes"]
matmul_tests_barrier = ["cubecl-matmul/matmul_tests_barrier"]
matmul_tests_base = ["matmul_tests_plane", "matmul_tests_vecmat", "matmul_tests_double", "matmul_tests_simple", "matmul_tests_ordered", "matmul_tests_cyclic", "matmul_tests_f16"]
matmul_tests_cyclic = ["cubecl-matmul/matmul_tests_cyclic"]
matmul_tests_double = ["cubecl-matmul/matmul_tests_double"]
matmul_tests_f16 = ["cubecl-matmul/matmul_tests_f16"]
matmul_tests_f32 = ["cubecl-matmul/matmul_tests_f32"]
matmul_tests_hybrid = ["cubecl-matmul/matmul_tests_hybrid"]
matmul_tests_hypercube = ["cubecl-matmul/matmul_tests_hypercube"]
matmul_tests_layouts = ["cubecl-matmul/matmul_tests_layouts"]
matmul_tests_ordered = ["cubecl-matmul/matmul_tests_ordered"]
matmul_tests_partition_buffering = ["cubecl-matmul/matmul_tests_partition_buffering"]
matmul_tests_plane = ["cubecl-matmul/matmul_tests_plane"]
matmul_tests_simple = ["cubecl-matmul/matmul_tests_simple"]
matmul_tests_specialized = ["cubecl-matmul/matmul_tests_specialized"]
matmul_tests_strided = ["cubecl-matmul/matmul_tests_strided"]
matmul_tests_tilewise = ["cubecl-matmul/matmul_tests_tilewise"]
matmul_tests_tma = ["cubecl-matmul/matmul_tests_tma"]
matmul_tests_unit = ["cubecl-matmul/matmul_tests_unit"]
matmul_tests_vecmat = ["cubecl-matmul/matmul_tests_vecmat"]
mlir-dump = []
std = ["cubecl-runtime/std", "cubecl-common/std", "cubecl-core/std"]

[lib]
name = "cubecl_cpu"
path = "src/lib.rs"

[package]
authors = ["marcantoinem <marc-antoine.m@outlook.com>"]
autobenches = false
autobins = false
autoexamples = false
autolib = false
autotests = false
build = "build.rs"
categories = ["science"]
description = "CPU runtime for CubeCL"
edition = "2024"
keywords = ["cpu"]
license = "MIT OR Apache-2.0"
name = "cubecl-cpu"
readme = "README.md"
repository = "https://github.com/tracel-ai/cubecl/tree/main/crates/cubecl-cpu"
resolver = "2"
version = "0.9.0-pre.2"