cubecl-cpu 0.8.1

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

[package]
authors = ["marcantoinem <marc-antoine.m@outlook.com>"]
categories = ["science"]
description = "CPU runtime for CubeCL"
edition.workspace = true
keywords = ["cpu"]
license.workspace = true
name = "cubecl-cpu"
readme.workspace = true
repository = "https://github.com/tracel-ai/cubecl/tree/main/crates/cubecl-cpu"
version.workspace = true

[features]
mlir-dump = []
default = [
    "std",
    "cubecl-runtime/default",
    "cubecl-common/default",
    "cubecl-core/default",
]
std = ["cubecl-runtime/std", "cubecl-common/std", "cubecl-core/std"]
matmul_tests_unit = ["cubecl-matmul/matmul_tests_unit"]
matmul_tests_plane = ["cubecl-matmul/matmul_tests_plane"]
matmul_tests_vecmat = ["cubecl-matmul/matmul_tests_vecmat"]
matmul_tests_tma = ["cubecl-matmul/matmul_tests_tma"]
matmul_tests_double = ["cubecl-matmul/matmul_tests_double"]
matmul_tests_simple = ["cubecl-matmul/matmul_tests_simple"]
matmul_tests_ordered = ["cubecl-matmul/matmul_tests_ordered"]
matmul_tests_cyclic = ["cubecl-matmul/matmul_tests_cyclic"]
matmul_tests_strided = ["cubecl-matmul/matmul_tests_strided"]
matmul_tests_tilewise = ["cubecl-matmul/matmul_tests_tilewise"]
matmul_tests_hybrid = ["cubecl-matmul/matmul_tests_hybrid"]
matmul_tests_barrier = ["cubecl-matmul/matmul_tests_barrier"]
matmul_tests_specialized = ["cubecl-matmul/matmul_tests_specialized"]
matmul_tests_f16 = ["cubecl-matmul/matmul_tests_f16"]
matmul_tests_f32 = ["cubecl-matmul/matmul_tests_f32"]
matmul_tests_layouts = ["cubecl-matmul/matmul_tests_layouts"]
matmul_tests_alt_shapes = ["cubecl-matmul/matmul_tests_alt_shapes"]
matmul_tests_partition_buffering = [
    "cubecl-matmul/matmul_tests_partition_buffering",
]
matmul_tests_hypercube = ["cubecl-matmul/matmul_tests_hypercube"]
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_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",
]
conv_tests = ["cubecl-convolution/conv_tests"]

[dependencies]
cubecl-common = { path = "../cubecl-common", version = "0.8.1", default-features = false }
cubecl-std = { path = "../cubecl-std", version = "0.8.1", default-features = false }
cubecl-core = { path = "../cubecl-core", version = "0.8.1", default-features = false }
cubecl-runtime = { path = "../cubecl-runtime", version = "0.8.1", default-features = false, features = [
    "channel-mutex",
] }
cubecl-opt = { path = "../cubecl-opt", version = "0.8.1", default-features = false }
cubecl-matmul = { path = "../cubecl-matmul", version = "0.8.1", features = [
    "export_tests",
] }
cubecl-convolution = { path = "../cubecl-convolution", version = "0.8.1", features = [
    "export_tests",
] }
cubecl-reduce = { path = "../cubecl-reduce", version = "0.8.1", features = [
    "export_tests",
] }

bytemuck = { workspace = true }
derive-new = { workspace = true }
half = { workspace = true }
log = { workspace = true }
serde = { workspace = true }
sysinfo = { workspace = true }
tracel-llvm = { workspace = true }

[dev-dependencies]
cubecl-core = { path = "../cubecl-core", version = "0.8.1", features = [
    "export_tests",
] }
cubecl-reduce = { path = "../cubecl-reduce", version = "0.8.1", features = [
    "export_tests",
] }
cubecl-random = { path = "../cubecl-random", version = "0.8.1", features = [
    "export_tests",
] }
cubecl-std = { path = "../cubecl-std", version = "0.8.1", features = [
    "export_tests",
] }
paste = { workspace = true }
pretty_assertions = { workspace = true }

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