File tree
14 files changed
+115
-72
lines changed- .github/workflows
- dipu
- scripts
- autogen_diopi_wrapper
- ci/ascend
- op_capture
- third_party
- torch_dipu
- csrc_dipu
- aten/ops
- vendor/droplet
- dipu
14 files changed
+115
-72
lines changedLines changed: 15 additions & 0 deletions
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
353 | 353 |
| |
354 | 354 |
| |
355 | 355 |
| |
| 356 | + | |
| 357 | + | |
| 358 | + | |
| 359 | + | |
| 360 | + | |
| 361 | + | |
| 362 | + | |
| 363 | + | |
| 364 | + | |
| 365 | + | |
| 366 | + | |
| 367 | + | |
| 368 | + | |
| 369 | + | |
| 370 | + | |
356 | 371 |
| |
357 | 372 |
| |
358 | 373 |
| |
|
Lines changed: 2 additions & 0 deletions
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
5 | 5 |
| |
6 | 6 |
| |
7 | 7 |
| |
| 8 | + | |
| 9 | + | |
8 | 10 |
| |
9 | 11 |
| |
10 | 12 |
| |
|
Lines changed: 6 additions & 15 deletions
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
183 | 183 |
| |
184 | 184 |
| |
185 | 185 |
| |
186 |
| - | |
187 |
| - | |
188 |
| - | |
189 |
| - | |
190 |
| - | |
191 |
| - | |
192 |
| - | |
193 |
| - | |
194 |
| - | |
195 |
| - | |
196 |
| - | |
197 |
| - | |
198 |
| - | |
199 |
| - | |
200 |
| - | |
| 186 | + | |
| 187 | + | |
| 188 | + | |
| 189 | + | |
| 190 | + | |
| 191 | + | |
201 | 192 |
| |
202 | 193 |
| |
203 | 194 |
| |
|
Lines changed: 1 addition & 0 deletions
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
218 | 218 |
| |
219 | 219 |
| |
220 | 220 |
| |
| 221 | + | |
221 | 222 |
| |
222 | 223 |
| |
223 | 224 |
| |
|
Lines changed: 25 additions & 16 deletions
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
114 | 114 |
| |
115 | 115 |
| |
116 | 116 |
| |
| 117 | + | |
117 | 118 |
| |
118 | 119 |
| |
119 | 120 |
| |
120 |
| - | |
| 121 | + | |
121 | 122 |
| |
122 | 123 |
| |
123 | 124 |
| |
124 | 125 |
| |
125 |
| - | |
| 126 | + | |
126 | 127 |
| |
127 | 128 |
| |
128 | 129 |
| |
129 | 130 |
| |
130 | 131 |
| |
131 | 132 |
| |
132 | 133 |
| |
133 |
| - | |
| 134 | + | |
134 | 135 |
| |
135 | 136 |
| |
136 | 137 |
| |
137 | 138 |
| |
138 | 139 |
| |
139 | 140 |
| |
140 |
| - | |
141 |
| - | |
142 |
| - | |
143 |
| - | |
144 |
| - | |
145 |
| - | |
146 |
| - | |
| 141 | + | |
| 142 | + | |
| 143 | + | |
| 144 | + | |
| 145 | + | |
| 146 | + | |
| 147 | + | |
| 148 | + | |
| 149 | + | |
| 150 | + | |
| 151 | + | |
147 | 152 |
| |
148 | 153 |
| |
149 | 154 |
| |
| |||
161 | 166 |
| |
162 | 167 |
| |
163 | 168 |
| |
164 |
| - | |
| 169 | + | |
165 | 170 |
| |
166 |
| - | |
167 | 171 |
| |
168 | 172 |
| |
169 | 173 |
| |
| |||
487 | 491 |
| |
488 | 492 |
| |
489 | 493 |
| |
| 494 | + | |
| 495 | + | |
| 496 | + | |
490 | 497 |
| |
491 | 498 |
| |
492 | 499 |
| |
| |||
550 | 557 |
| |
551 | 558 |
| |
552 | 559 |
| |
553 |
| - | |
554 | 560 |
| |
555 | 561 |
| |
556 | 562 |
| |
| |||
972 | 978 |
| |
973 | 979 |
| |
974 | 980 |
| |
975 |
| - | |
976 |
| - | |
977 |
| - | |
| 981 | + | |
| 982 | + | |
| 983 | + | |
| 984 | + | |
| 985 | + | |
| 986 | + | |
978 | 987 |
| |
979 | 988 |
| |
980 | 989 |
| |
|
Lines changed: 21 additions & 0 deletions
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
| 1 | + | |
| 2 | + | |
| 3 | + | |
| 4 | + | |
| 5 | + | |
| 6 | + | |
| 7 | + | |
| 8 | + | |
| 9 | + | |
| 10 | + | |
| 11 | + | |
| 12 | + | |
| 13 | + | |
| 14 | + | |
| 15 | + | |
| 16 | + | |
| 17 | + | |
| 18 | + | |
| 19 | + | |
| 20 | + | |
| 21 | + |
Lines changed: 3 additions & 0 deletions
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
1648 | 1648 |
| |
1649 | 1649 |
| |
1650 | 1650 |
| |
| 1651 | + | |
| 1652 | + | |
| 1653 | + | |
1651 | 1654 |
| |
1652 | 1655 |
| |
1653 | 1656 |
| |
|
Lines changed: 30 additions & 24 deletions
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
10 | 10 |
| |
11 | 11 |
| |
12 | 12 |
| |
13 |
| - | |
14 |
| - | |
15 |
| - | |
16 |
| - | |
| 13 | + | |
| 14 | + | |
| 15 | + | |
| 16 | + | |
| 17 | + | |
| 18 | + | |
17 | 19 |
| |
18 | 20 |
| |
19 | 21 |
| |
20 | 22 |
| |
21 |
| - | |
22 |
| - | |
23 |
| - | |
24 |
| - | |
25 |
| - | |
| 23 | + | |
| 24 | + | |
| 25 | + | |
| 26 | + | |
| 27 | + | |
| 28 | + | |
26 | 29 |
| |
27 | 30 |
| |
28 | 31 |
| |
29 | 32 |
| |
30 | 33 |
| |
31 | 34 |
| |
32 |
| - | |
| 35 | + | |
33 | 36 |
| |
34 | 37 |
| |
35 | 38 |
| |
36 | 39 |
| |
37 |
| - | |
| 40 | + | |
38 | 41 |
| |
39 |
| - | |
| 42 | + | |
40 | 43 |
| |
41 | 44 |
| |
42 | 45 |
| |
43 |
| - | |
44 |
| - | |
45 |
| - | |
46 |
| - | |
47 |
| - | |
48 |
| - | |
49 |
| - | |
50 |
| - | |
51 |
| - | |
52 |
| - | |
53 |
| - | |
54 |
| - | |
| 46 | + | |
| 47 | + | |
| 48 | + | |
| 49 | + | |
| 50 | + | |
| 51 | + | |
| 52 | + | |
| 53 | + | |
| 54 | + | |
| 55 | + | |
| 56 | + | |
| 57 | + | |
| 58 | + | |
| 59 | + | |
| 60 | + | |
55 | 61 |
| |
56 | 62 |
|
Lines changed: 1 addition & 0 deletions
Original file line number | Diff line number | Diff line change | |
---|---|---|---|
| |||
48 | 48 |
| |
49 | 49 |
| |
50 | 50 |
| |
| 51 | + | |
51 | 52 |
| |
52 | 53 |
| |
53 | 54 |
| |
|
- diopi_test/diopi_stub/csrc/litert.cpp+12
- diopi_test/diopi_stub/include/litert.hpp+10
- diopi_test/python/configs/diopi_configs.py+142-11
- diopi_test/python/conformance/diopi_functions.py+56-10
- diopi_test/python/conformance/gen_output.py+26-4
- impl/ascend/device_configs.py-5
- impl/ascend_npu/ascend_config.yaml+5-5
- impl/ascend_npu/diopi_impl/col2im.cpp+2-2
- impl/ascend_npu/diopi_impl/functions_ext/flash_attention.cpp+38-19
- impl/ascend_npu/diopi_impl/functions_ext/flash_attentionv2.cpp+20-10
- impl/ascend_npu/diopi_impl/functions_ext/scaled_masked_softmax.cpp+42
- impl/ascend_npu/diopi_impl/matmul.cpp+2-2
- impl/ascend_npu/diopi_impl/remainder.cpp+10-12
- impl/ascend_npu/diopi_impl/threshold.cpp+33
- impl/camb/CMakeLists.txt+6-2
- impl/camb/cnnl_helper.hpp+1
- impl/camb/device_configs.py+6-2
- impl/camb/diopi_helper.hpp+1
- impl/camb/functions_ext/rotaryEmbedding.cpp+158
- proto/include/diopi/diopirt.h+2
- proto/include/diopi/functions_ext.h+70-30
0 commit comments