Skip to content

Commit 6a29e89

Browse files
authored
[Feature](mlu-ops): upgrade bangc_kernls_path.json . (Cambricon#1205)
1 parent 8e84ae3 commit 6a29e89

File tree

1 file changed

+33
-18
lines changed

1 file changed

+33
-18
lines changed
Original file line numberDiff line numberDiff line change
@@ -1,21 +1,36 @@
11
{
2-
"common": [
3-
"bangc_helper_dtype.h",
4-
"bangc_kernels.h",
5-
"kernels/debug.h",
6-
"kernels/device_check.h",
7-
"kernels/kernel.h"
8-
],
2+
"version": "v1.5.0",
3+
4+
"defaultDeps": [
5+
{
6+
"cntoolkit": "v4.0.0",
7+
"driver": "v6.0.3"
8+
}
9+
],
910

10-
"operators": [
11-
{
12-
"name": "mluAdamW",
13-
"header": "kernels/adam_w/adam_w_lite.h",
14-
"sources": [
15-
"kernels/adam_w/adam_w_lite.h",
16-
"kernels/adam_w/adam_w_union1.mluh",
17-
"kernels/adam_w/adam_w_lite_union1.mlu"
18-
]
19-
}
20-
]
11+
"optionalDeps": [
12+
{
13+
"cnnl": "v1.28.0"
14+
}
15+
],
16+
17+
"common": [
18+
"bangc_helper_dtype.h",
19+
"bangc_kernels.h",
20+
"kernels/debug.h",
21+
"kernels/device_check.h",
22+
"kernels/kernel.h"
23+
],
24+
25+
"operators": [
26+
{
27+
"name": "mluAdamW",
28+
"header": "kernels/adam_w/adam_w_lite.h",
29+
"sources": [
30+
"kernels/adam_w/adam_w_lite.h",
31+
"kernels/adam_w/adam_w_union1.mluh",
32+
"kernels/adam_w/adam_w_lite_union1.mlu"
33+
]
34+
}
35+
]
2136
}

0 commit comments

Comments
 (0)