-
Notifications
You must be signed in to change notification settings - Fork 10
/
Copy pathCargo.toml
66 lines (53 loc) · 1.2 KB
/
Cargo.toml
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
[package]
name = "candle-ext"
version = "0.1.7"
edition = "2021"
description = "An extension library to Candle that provides PyTorch functions not currently available in Candle"
license = "MIT OR Apache-2.0"
repository = "https://github.com/mokeyish/candle-ext"
[lib]
doctest = false
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[features]
default = ["all"]
all = [
"chunk",
# "cumsum",
"equal",
# "eye",
"logical_not",
"logical_or",
"masked_fill",
"outer",
"scatter",
"scaled_dot_product_attention",
"triangular",
"unbind",
]
# functions feature
chunk = ["to_tuple"]
cumsum = []
equal = []
eye = []
full = []
full_like = ["full"]
logical_not = []
logical_or = []
masked_fill = ["full_like"]
outer = []
scaled_dot_product_attention = ["masked_fill", "logical_not", "tril"]
triangular = []
to_tuple = []
scatter = []
tril = ["triangular"]
trilu = ["triangular"]
unbind = ["to_tuple"]
cuda = ["candle-core/cuda"]
# features for unit tests
test_masked_fill = ["masked_fill", "triangular", "logical_not"]
[dependencies]
candle-core = "0.4"
candle-nn = "0.4"
num-traits = "0.2"
[build-dependencies]
anyhow = "1.0.75"