forked from chalk-lab/Mooncake.jl
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathProject.toml
More file actions
122 lines (117 loc) · 3.8 KB
/
Copy pathProject.toml
File metadata and controls
122 lines (117 loc) · 3.8 KB
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
name = "Mooncake"
uuid = "da2b9cff-9c12-43a0-ae48-6db2b0edb7d6"
authors = ["Will Tebbutt, Hong Ge, and contributors"]
version = "0.5.26"
[deps]
ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"
ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
DispatchDoctor = "8d63f2c5-f18a-4cf2-ba9d-b3f60fc568c8"
ExprTools = "e2ba6199-217a-4e67-a87a-7c52f15ade04"
Graphs = "86223c79-3864-5bf0-83f7-82e725a168b6"
LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
MistyClosures = "dbe65cb8-6be2-42dd-bbc5-4196aaced4f4"
PrecompileTools = "aea7be01-6a6a-4083-8856-8a6e6704d82a"
Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
[weakdeps]
AllocCheck = "9b6a8646-10ed-4001-bbdc-1d2f46dfbb1a"
BFloat16s = "ab4f0b2a-ad5b-11e8-123f-65d77653426b"
ChainRules = "082447d4-558c-5d27-93f4-14fc19e9eca2"
CUDA = "052768ef-5323-5732-b1bb-66c8b64840ba"
cuDNN = "02a925ec-e4fe-4b08-9a7e-0d78e3d38ccd"
Distributions = "31c24e10-a181-5473-b8eb-7969acd0382f"
DynamicExpressions = "a40a106e-89c9-4ca8-8020-a735e8728b6b"
Flux = "587475ba-b771-5e3f-ad9e-33799f191a9c"
FunctionWrappers = "069b7b12-0de2-55c6-9aab-29f3d0a68a2e"
GPUArraysCore = "46192b85-c4d5-4398-a991-12ede77f4527"
JET = "c3a54625-cd67-489e-a8e7-0a5a0ff4e31b"
LogExpFunctions = "2ab3a3ac-af41-5b50-aa03-7779005ae688"
LuxLib = "82251201-b29d-42c6-8e01-566dec8acb11"
MLDataDevices = "7e8f7934-dd98-4c1a-8fe8-92b47a384d40"
NNlib = "872c559c-99b0-510c-b3b7-b6c96a88d5cd"
SLEEFPirates = "476501e8-09a2-5ece-8869-fb82de89a1fa"
SpecialFunctions = "276daf66-3868-5448-9aa4-cd146d93841b"
Static = "aedffcd0-7271-4cad-89d0-dc628f76c6d3"
[extensions]
MooncakeAllocCheckExt = "AllocCheck"
MooncakeBFloat16sExt = "BFloat16s"
MooncakeCUDAExt = "CUDA"
MooncakeChainRulesExt = "ChainRules"
MooncakecuDNNExt = ["CUDA", "cuDNN"]
MooncakeDistributionsExt = "Distributions"
MooncakeDynamicExpressionsExt = "DynamicExpressions"
MooncakeFluxExt = "Flux"
MooncakeFunctionWrappersExt = "FunctionWrappers"
MooncakeJETExt = "JET"
MooncakeLogExpFunctionsExt = "LogExpFunctions"
MooncakeLuxLibExt = ["LuxLib", "MLDataDevices", "Static"]
MooncakeLuxLibSLEEFPiratesExtension = ["LuxLib", "SLEEFPirates"]
MooncakeLuxLibNNlibExt = ["LuxLib", "NNlib"]
MooncakeNNlibExt = ["NNlib", "LinearAlgebra"]
MooncakeSpecialFunctionsExt = "SpecialFunctions"
[compat]
ADTypes = "1.9"
AllocCheck = "0.2"
Aqua = "0.8.9"
BFloat16s = "0.6"
BenchmarkTools = "1"
CUDA = "5"
cuDNN = "1"
ChainRules = "1.71.0"
ChainRulesCore = "1"
DiffTests = "0.1"
DispatchDoctor = "0.4.28"
Distributions = "0.25"
DynamicExpressions = "2"
ExprTools = "0.1"
Flux = "0.16.3"
FunctionWrappers = "1.1.3"
GPUArraysCore = "0.1, 0.2"
Graphs = "1"
JET = "0.9, 0.10, 0.11"
LinearAlgebra = "1"
LogExpFunctions = "0.3"
Logging = "1"
LuxLib = "1.11"
MLDataDevices = "1.10.0"
MistyClosures = "2"
NNlib = "0.9"
PrecompileTools = "1"
Pkg = "1"
Random = "1"
Revise = "3.0 - 3.12"
SLEEFPirates = "0.6.43"
SpecialFunctions = "2"
StableRNGs = "1"
Static = "1.1.1"
Test = "1"
julia = "~1.10.8, 1.11.6, 1.12.1"
[extras]
AllocCheck = "9b6a8646-10ed-4001-bbdc-1d2f46dfbb1a"
Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
BenchmarkTools = "6e4b80f9-dd63-53aa-95a3-0cdb28fa8baf"
ChainRules = "082447d4-558c-5d27-93f4-14fc19e9eca2"
DiffTests = "de460e47-3fe3-5279-bb4a-814414816d5d"
DynamicExpressions = "a40a106e-89c9-4ca8-8020-a735e8728b6b"
JET = "c3a54625-cd67-489e-a8e7-0a5a0ff4e31b"
Logging = "56ddb016-857b-54e1-b83d-db4d58db5568"
Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f"
Revise = "295af30f-e4ad-537b-8983-00126c2a3abe"
StableRNGs = "860ef19b-820b-49d6-a774-d7a799459cd3"
Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
LogExpFunctions = "2ab3a3ac-af41-5b50-aa03-7779005ae688"
[targets]
test = [
"AllocCheck",
"Aqua",
"BenchmarkTools",
"ChainRules",
"DiffTests",
"JET",
"Logging",
"Pkg",
"Revise",
"StableRNGs",
"Test",
"LogExpFunctions",
]