Merged
修改产品名 #11348
yanglu-1Add on Sep 29
修改产品名 #11348
Merged
yanglu-1Add on Sep 29
Total: 198 files changed+338-338
@@ -7,7 +7,7 @@
7- <term>Ascend 950PR&950DT系列产品</term>:支持7- <term>Ascend 950PR&950DT系列产品</term>:支持
8<!-- end id1 -->8<!-- end id1 -->
9<!-- npu="A3" id2 -->9<!-- npu="A3" id2 -->
10-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:不支持10+- <term>Atlas A3系列产品</term>:不支持
11<!-- end id2 -->11<!-- end id2 -->
12<!-- npu="910b" id3 -->12<!-- npu="910b" id3 -->
13- <term>Atlas A2系列产品</term>:不支持13- <term>Atlas A2系列产品</term>:不支持
@@ -12,7 +12,7 @@
12<td style="text-align:center">√</td>12<td style="text-align:center">√</td>
13</tr>13</tr>
14<tr>14<tr>
15-<td><term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term></td>15+<td><term>Atlas A3系列产品</term></td>
16<td style="text-align:center">×</td>16<td style="text-align:center">×</td>
17</tr>17</tr>
18<tr>18<tr>
@@ -47,15 +47,15 @@
47* strides47* strides
48 - N和C的维度必须为1。48 - N和C的维度必须为1。
49 - Ascend 950PR&950DT系列产品:D、H和W维度的取值必须范围在 [1,2147483646] 之间。49 - Ascend 950PR&950DT系列产品:D、H和W维度的取值必须范围在 [1,2147483646] 之间。
50- - Atlas A2系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品:H和W的维度的取值范围必须在 [1,63] 之间,D维度的取值范围必须在 [1,255] 之间。50+ - Atlas A2系列产品、Atlas A3系列产品:H和W的维度的取值范围必须在 [1,63] 之间,D维度的取值范围必须在 [1,255] 之间。
51* pads51* pads
52 - 填充顺序为:[front, back, top, bottom, left, right]。52 - 填充顺序为:[front, back, top, bottom, left, right]。
53 - Ascend 950PR&950DT系列产品:H、W和D维度的取值范围必须在 [1,2147483646] 之间。53 - Ascend 950PR&950DT系列产品:H、W和D维度的取值范围必须在 [1,2147483646] 之间。
54- - Atlas A2系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品:H、W和D维度的取值范围必须在 [0,255] 之间。54+ - Atlas A2系列产品、Atlas A3系列产品:H、W和D维度的取值范围必须在 [0,255] 之间。
55* dilations55* dilations
56 - N与C的维度必须为1。56 - N与C的维度必须为1。
57 - Ascend 950PR&950DT系列产品:H、W和D维度的取值范围必须在 [1,2147483646] 之间。57 - Ascend 950PR&950DT系列产品:H、W和D维度的取值范围必须在 [1,2147483646] 之间。
58- - Atlas A2系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品:H、W和D维度的取值范围必须在 [1,255] 之间。58+ - Atlas A2系列产品、Atlas A3系列产品:H、W和D维度的取值范围必须在 [1,255] 之间。
59* output_padding59* output_padding
60 - 仅<term>Ascend 950PR&950DT系列产品</term>支持非零值,其他产品仅支持全零。60 - 仅<term>Ascend 950PR&950DT系列产品</term>支持非零值,其他产品仅支持全零。
61 - D、H和W维度的取值范围必须在 [0,2147483646] 之间。61 - D、H和W维度的取值范围必须在 [0,2147483646] 之间。
@@ -67,15 +67,15 @@
67* strides67* strides
68 - N和C的维度必须为1。68 - N和C的维度必须为1。
69 - Ascend 950PR&950DT系列产品:D、H和W维度的取值范围必须在 [1,2147483647] 之间。69 - Ascend 950PR&950DT系列产品:D、H和W维度的取值范围必须在 [1,2147483647] 之间。
70- - Atlas A2系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品:H和W的维度的取值范围必须在 [1,63] 之间,D维度的取值范围必须在 [1,255] 之间。70+ - Atlas A2系列产品、Atlas A3系列产品:H和W的维度的取值范围必须在 [1,63] 之间,D维度的取值范围必须在 [1,255] 之间。
71* pads71* pads
72 - 填充顺序为:[front, back, top, bottom, left, right]。72 - 填充顺序为:[front, back, top, bottom, left, right]。
73 - Ascend 950PR&950DT系列产品:D、H和W维度的取值范围必须在 [0,2147483647] 之间。73 - Ascend 950PR&950DT系列产品:D、H和W维度的取值范围必须在 [0,2147483647] 之间。
74- - Atlas A2系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品:D、H和W维度的取值范围必须在 [0,255] 之间。74+ - Atlas A2系列产品、Atlas A3系列产品:D、H和W维度的取值范围必须在 [0,255] 之间。
75* dilations75* dilations
76 - N与C的维度必须为1。76 - N与C的维度必须为1。
77 - Ascend 950PR&950DT系列产品:D、H和W维度的取值范围必须在 [1,2147483647] 之间。77 - Ascend 950PR&950DT系列产品:D、H和W维度的取值范围必须在 [1,2147483647] 之间。
78- - Atlas A2系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品:H、W和D维度的取值范围必须在 [1,255] 之间。78+ - Atlas A2系列产品、Atlas A3系列产品:H、W和D维度的取值范围必须在 [1,255] 之间。
79* 由于硬件资源限制,算子在部分参数取值组合场景下会执行失败,请根据日志信息提示分析并排查问题。若无法解决,请单击 [Link](https://www.hiascend.com/support)获取技术支持。79* 由于硬件资源限制,算子在部分参数取值组合场景下会执行失败,请根据日志信息提示分析并排查问题。若无法解决,请单击 [Link](https://www.hiascend.com/support)获取技术支持。
80 80 
81## 调用说明81## 调用说明
@@ -84,15 +84,15 @@
84* strides84* strides
85 - N和C的维度必须为1。85 - N和C的维度必须为1。
86 - Ascend 950PR&950DT系列产品:D、H和W维度的取值范围必须在 [1,2147483647] 之间。86 - Ascend 950PR&950DT系列产品:D、H和W维度的取值范围必须在 [1,2147483647] 之间。
87- - Atlas A2系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品:H和W维度的取值范围必须在 [1,63] 之间,D维度的取值范围必须在 [1,255] 之间。87+ - Atlas A2系列产品、Atlas A3系列产品:H和W维度的取值范围必须在 [1,63] 之间,D维度的取值范围必须在 [1,255] 之间。
88* pads88* pads
89 - 填充顺序为:[front, back, top, bottom, left, right]。89 - 填充顺序为:[front, back, top, bottom, left, right]。
90 - Ascend 950PR&950DT系列产品:D、H和W维度的取值范围必须在 [0,2147483647] 之间。90 - Ascend 950PR&950DT系列产品:D、H和W维度的取值范围必须在 [0,2147483647] 之间。
91- - Atlas A2系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品:D、H和W维度的取值范围必须在 [0,255] 之间。91+ - Atlas A2系列产品、Atlas A3系列产品:D、H和W维度的取值范围必须在 [0,255] 之间。
92* dilations92* dilations
93 - N与C的维度必须为1。93 - N与C的维度必须为1。
94 - Ascend 950PR&950DT系列产品:D、H和W维度的取值范围必须在 [1,2147483647] 之间。94 - Ascend 950PR&950DT系列产品:D、H和W维度的取值范围必须在 [1,2147483647] 之间。
95- - Atlas A2系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品:D、H和W维度的取值范围必须在 [1,255] 之间。95+ - Atlas A2系列产品、Atlas A3系列产品:D、H和W维度的取值范围必须在 [1,255] 之间。
96* output_padding96* output_padding
97 - N和C维度必须为0,仅允许在深度、高度、宽度方向上添加。97 - N和C维度必须为0,仅允许在深度、高度、宽度方向上添加。
98 - D、H和W维度的取值范围必须在 [0,2147483647] 之间。98 - D、H和W维度的取值范围必须在 [0,2147483647] 之间。
@@ -12,7 +12,7 @@
12<td style="text-align:center">√</td>12<td style="text-align:center">√</td>
13</tr>13</tr>
14<tr>14<tr>
15-<td><term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term></td>15+<td><term>Atlas A3系列产品</term></td>
16<td style="text-align:center">√</td>16<td style="text-align:center">√</td>
17</tr>17</tr>
18<tr>18<tr>
@@ -166,7 +166,7 @@
166</tr>166</tr>
167</table>167</table>
168 168 
169-- Atlas A2系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品:169+- Atlas A2系列产品、Atlas A3系列产品:
170 - 不支持`HIFLOAT8`数据类型。170 - 不支持`HIFLOAT8`数据类型。
171 - `scale`和`offset`参数仅支持`FLOAT`类型。171 - `scale`和`offset`参数仅支持`FLOAT`类型。
172 - 输入为`INT8`数据类型时,`bias`为必选输入,`groups`仅支持1。172 - 输入为`INT8`数据类型时,`bias`为必选输入,`groups`仅支持1。
@@ -176,7 +176,7 @@
176 176 
177## 约束说明177## 约束说明
178 178 
179-- Atlas A2系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品:179+- Atlas A2系列产品、Atlas A3系列产品:
180 - `filter`的`H`、`W`维度范围:[1,511]。180 - `filter`的`H`、`W`维度范围:[1,511]。
181 - 不支持空`tensor`。181 - 不支持空`tensor`。
182 - 当`groups`为1, `dilation`全为1,`padding`全为0,`filter`没有为1的维度,`x`的`D`*`H`*`W`小于65536,`bias`为`FLOAT`时,会进入`Pointwise`分支,可以使用`NCDHW`格式。182 - 当`groups`为1, `dilation`全为1,`padding`全为0,`filter`没有为1的维度,`x`的`D`*`H`*`W`小于65536,`bias`为`FLOAT`时,会进入`Pointwise`分支,可以使用`NCDHW`格式。
@@ -76,7 +76,7 @@
76* <term>Atlas推理系列产品</term>、<term>Atlas训练系列产品</term>:76* <term>Atlas推理系列产品</term>、<term>Atlas训练系列产品</term>:
77 - 不支持BFLOAT16、HIFLOAT8、FLOAT8_E4M3FN。77 - 不支持BFLOAT16、HIFLOAT8、FLOAT8_E4M3FN。
78 - gradOutput、weight参数下,不支持空tensor。78 - gradOutput、weight参数下,不支持空tensor。
79-* <term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:79+* <term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term>:
80 - 不支持HIFLOAT8、FLOAT8_E4M3FN。80 - 不支持HIFLOAT8、FLOAT8_E4M3FN。
81 81 
82## 约束说明82## 约束说明
@@ -6,7 +6,7 @@
6- <term>Ascend 950PR&950DT系列产品</term>:支持6- <term>Ascend 950PR&950DT系列产品</term>:支持
7<!-- end id1 -->7<!-- end id1 -->
8<!-- npu="A3" id2 -->8<!-- npu="A3" id2 -->
9-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持9+- <term>Atlas A3系列产品</term>:支持
10<!-- end id2 -->10<!-- end id2 -->
11<!-- npu="910b" id3 -->11<!-- npu="910b" id3 -->
12- <term>Atlas A2系列产品</term>:支持12- <term>Atlas A2系列产品</term>:支持
@@ -375,7 +375,7 @@ aclnnStatus aclnnConvTbcBackward(
375- 确定性计算375- 确定性计算
376 376 
377 <!-- npu="A3,910b,910,310b" id7 -->377 <!-- npu="A3,910b,910,310b" id7 -->
378- - <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>、<term>Atlas A2系列产品</term>、<term>Atlas 200I/500 A2推理产品</term>、<term>Atlas训练系列产品</term>:aclnnConvTbcBackward默认非确定性实现,支持通过aclrtCtxSetSysParamOpt开启确定性。378+ - <term>Atlas A3系列产品</term>、<term>Atlas A2系列产品</term>、<term>Atlas 200I/500 A2推理产品</term>、<term>Atlas训练系列产品</term>:aclnnConvTbcBackward默认非确定性实现,支持通过aclrtCtxSetSysParamOpt开启确定性。
379 <!-- end id7 -->379 <!-- end id7 -->
380 <!-- npu="950" id8 -->380 <!-- npu="950" id8 -->
381 - <term>Ascend 950PR&950DT系列产品</term>:aclnnConvTbcBackward默认确定性实现。381 - <term>Ascend 950PR&950DT系列产品</term>:aclnnConvTbcBackward默认确定性实现。
@@ -394,7 +394,7 @@ aclnnStatus aclnnConvTbcBackward(
394 <th><term>Ascend 950PR&950DT系列产品</term></th>394 <th><term>Ascend 950PR&950DT系列产品</term></th>
395 <!-- end id9 -->395 <!-- end id9 -->
396 <!-- npu="A3,910b" id10 -->396 <!-- npu="A3,910b" id10 -->
397- <th><term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term></th>397+ <th><term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term></th>
398 <!-- end id10 -->398 <!-- end id10 -->
399 <!-- npu="910" id11 -->399 <!-- npu="910" id11 -->
400 <th><term>Atlas训练系列产品</term></th>400 <th><term>Atlas训练系列产品</term></th>
@@ -6,7 +6,7 @@
6- <term>Ascend 950PR&950DT系列产品</term>:支持6- <term>Ascend 950PR&950DT系列产品</term>:支持
7<!-- end id1 -->7<!-- end id1 -->
8<!-- npu="A3" id2 -->8<!-- npu="A3" id2 -->
9-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持9+- <term>Atlas A3系列产品</term>:支持
10<!-- end id2 -->10<!-- end id2 -->
11<!-- npu="910b" id3 -->11<!-- npu="910b" id3 -->
12- <term>Atlas A2系列产品</term>:支持12- <term>Atlas A2系列产品</term>:支持
@@ -453,7 +453,7 @@ aclnnStatus aclnnConvolutionBackward(
453- 确定性计算453- 确定性计算
454 454 
455 <!-- npu="A3,910b,910,310b" id7 -->455 <!-- npu="A3,910b,910,310b" id7 -->
456- - <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>、<term>Atlas A2系列产品</term>、<term>Atlas 200I/500 A2推理产品</term>、<term>Atlas训练系列产品</term>:aclnnConvolutionBackward默认非确定性实现,支持通过aclrtCtxSetSysParamOpt开启确定性。456+ - <term>Atlas A3系列产品</term>、<term>Atlas A2系列产品</term>、<term>Atlas 200I/500 A2推理产品</term>、<term>Atlas训练系列产品</term>:aclnnConvolutionBackward默认非确定性实现,支持通过aclrtCtxSetSysParamOpt开启确定性。
457 <!-- end id7 -->457 <!-- end id7 -->
458 <!-- npu="950" id8 -->458 <!-- npu="950" id8 -->
459 - <term>Ascend 950PR&950DT系列产品</term>:aclnnConvolutionBackward默认确定性实现。459 - <term>Ascend 950PR&950DT系列产品</term>:aclnnConvolutionBackward默认确定性实现。
@@ -472,7 +472,7 @@ aclnnStatus aclnnConvolutionBackward(
472 <th><term>Ascend 950PR&950DT系列产品</term></th>472 <th><term>Ascend 950PR&950DT系列产品</term></th>
473 <!-- end id9 -->473 <!-- end id9 -->
474 <!-- npu="A3,910b" id10 -->474 <!-- npu="A3,910b" id10 -->
475- <th><term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term></th>475+ <th><term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term></th>
476 <!-- end id10 -->476 <!-- end id10 -->
477 <!-- npu="910,310p" id11 -->477 <!-- npu="910,310p" id11 -->
478 <th><term>Atlas推理系列产品</term>、<term>Atlas训练系列产品</term></th>478 <th><term>Atlas推理系列产品</term>、<term>Atlas训练系列产品</term></th>
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:不支持11+- <term>Atlas A3系列产品</term>:不支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:不支持14- <term>Atlas A2系列产品</term>:不支持
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -284,7 +284,7 @@ aclnnStatus aclnnConvDepthwise2d(
284 <tr>284 <tr>
285 <th>约束类型</th>285 <th>约束类型</th>
286 <!-- npu="A3,910b" id7 -->286 <!-- npu="A3,910b" id7 -->
287- <th><term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term></th>287+ <th><term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term></th>
288 <!-- end id7 -->288 <!-- end id7 -->
289 <!-- npu="950" id8 -->289 <!-- npu="950" id8 -->
290 <th><term>Ascend 950PR&950DT系列产品</term></th>290 <th><term>Ascend 950PR&950DT系列产品</term></th>
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -276,7 +276,7 @@ aclnnStatus aclnnConvTbc(
276 <tr>276 <tr>
277 <th>约束类型</th>277 <th>约束类型</th>
278 <!-- npu="A3,910b" id7 -->278 <!-- npu="A3,910b" id7 -->
279- <th><term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term></th>279+ <th><term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term></th>
280 <!-- end id7 -->280 <!-- end id7 -->
281 <!-- npu="950" id8 -->281 <!-- npu="950" id8 -->
282 <th><term>Ascend 950PR&950DT系列产品</term></th>282 <th><term>Ascend 950PR&950DT系列产品</term></th>
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -340,7 +340,7 @@ aclnnStatus aclnnConvolution(
340 <tr>340 <tr>
341 <th>约束类型</th>341 <th>约束类型</th>
342 <!-- npu="A3,910b" id7 -->342 <!-- npu="A3,910b" id7 -->
343- <th><term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term></th>343+ <th><term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term></th>
344 <!-- end id7 -->344 <!-- end id7 -->
345 <!-- npu="310b" id8 -->345 <!-- npu="310b" id8 -->
346 <th><term>Atlas 200I/500 A2推理产品</term></th>346 <th><term>Atlas 200I/500 A2推理产品</term></th>
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -348,7 +348,7 @@ aclnnStatus aclnnQuantConvolution(
348 <tr>348 <tr>
349 <th>约束类型</th>349 <th>约束类型</th>
350 <!-- npu="A3,910b" id7 -->350 <!-- npu="A3,910b" id7 -->
351- <th><term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term></th>351+ <th><term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term></th>
352 <!-- end id7 -->352 <!-- end id7 -->
353 </tr>353 </tr>
354 </thead>354 </thead>
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -350,7 +350,7 @@ aclnnStatus aclnnQuantConvolutionWeightNz(
350 <tr>350 <tr>
351 <th>约束类型</th>351 <th>约束类型</th>
352 <!-- npu="A3,910b" id7 -->352 <!-- npu="A3,910b" id7 -->
353- <th><term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term></th>353+ <th><term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term></th>
354 <!-- end id7 -->354 <!-- end id7 -->
355 </tr>355 </tr>
356 </thead>356 </thead>
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -392,7 +392,7 @@ aclnnStatus aclnnDeformableConv2d(
392 <tr>392 <tr>
393 <th>参数名</th>393 <th>参数名</th>
394 <!-- npu="A3,910b" id7 -->394 <!-- npu="A3,910b" id7 -->
395- <th><term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term></th>395+ <th><term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term></th>
396 <!-- end id7 -->396 <!-- end id7 -->
397 <!-- npu="950" id8 -->397 <!-- npu="950" id8 -->
398 <th><term>Ascend 950PR&950DT系列产品</term></th>398 <th><term>Ascend 950PR&950DT系列产品</term></th>
@@ -12,7 +12,7 @@
12<td style="text-align:center">√</td>12<td style="text-align:center">√</td>
13</tr>13</tr>
14<tr>14<tr>
15-<td><term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term></td>15+<td><term>Atlas A3系列产品</term></td>
16<td style="text-align:center">×</td>16<td style="text-align:center">×</td>
17</tr>17</tr>
18<tr>18<tr>
@@ -60,7 +60,7 @@ bash build.sh --pkg --soc=${soc_version} --ops=add_example -j16
60产品名对应的\$\{soc\_version\}取值如下,请按实际场景传参。60产品名对应的\$\{soc\_version\}取值如下,请按实际场景传参。
61 61 
62- Atlas A2系列产品:取值为ascend910b62- Atlas A2系列产品:取值为ascend910b
63-- Atlas A3 训练系列产品/Atlas A3 推理系列产品:取值为ascend910_9363+- Atlas A3系列产品:取值为ascend910_93
64- 950系列产品:取值为ascend95064- 950系列产品:取值为ascend950
65 65 
66若提示如下信息,说明编译成功。66若提示如下信息,说明编译成功。
@@ -155,7 +155,7 @@ __aicore__ inline void AddExample<T>::Compute(int64_t currentNum)
155 产品名对应的\$\{soc\_version\}取值如下,请按实际场景传参。155 产品名对应的\$\{soc\_version\}取值如下,请按实际场景传参。
156 156 
157 - Atlas A2系列产品:取值为ascend910b157 - Atlas A2系列产品:取值为ascend910b
158- - Atlas A3 训练系列产品/Atlas A3 推理系列产品:取值为ascend910_93158+ - Atlas A3系列产品:取值为ascend910_93
159 - 950系列产品:取值为ascend950159 - 950系列产品:取值为ascend950
160 160 
1612. **重新安装**:1612. **重新安装**:
@@ -55,7 +55,7 @@
55 # bash build.sh --pkg --experimental --soc=ascend910b --ops=transpose_batch_mat_mul -j1655 # bash build.sh --pkg --experimental --soc=ascend910b --ops=transpose_batch_mat_mul -j16
56 ```56 ```
57 57 
58- - --soc:\$\{soc\_version\}表示NPU型号。Atlas A2系列产品使用"ascend910b"(默认),Atlas A3 训练系列产品/Atlas A3 推理系列产品使用"ascend910_93",Ascend 950PR&950DT系列产品产品使用"ascend950"。58+ - --soc:\$\{soc\_version\}表示NPU型号。Atlas A2系列产品使用"ascend910b"(默认),Atlas A3系列产品使用"ascend910_93",Ascend 950PR&950DT系列产品产品使用"ascend950"。
59 - --vendor_name(可选):\$\{vendor\_name\}表示构建的自定义算子包名,默认名为custom。59 - --vendor_name(可选):\$\{vendor\_name\}表示构建的自定义算子包名,默认名为custom。
60 - --ops(可选):\$\{op\_list\}表示待编译算子,不指定时默认编译所有算子。格式形如"transpose_batch_mat_mul,fatrelu_mul,...",多算子之间用英文逗号","分隔。60 - --ops(可选):\$\{op\_list\}表示待编译算子,不指定时默认编译所有算子。格式形如"transpose_batch_mat_mul,fatrelu_mul,...",多算子之间用英文逗号","分隔。
61 - --experimental(可选):表示编译用户保存在experimental贡献目录下的算子。61 - --experimental(可选):表示编译用户保存在experimental贡献目录下的算子。
@@ -102,7 +102,7 @@
102 # bash build.sh --pkg --experimental --soc=${soc_version} [-j${n}]102 # bash build.sh --pkg --experimental --soc=${soc_version} [-j${n}]
103 ```103 ```
104 104 
105- - --soc:\$\{soc\_version\}表示NPU型号。Atlas A2系列产品使用"ascend910b"(默认),Atlas A3 训练系列产品/Atlas A3 推理系列产品使用"ascend910_93",Ascend 950PR&950DT系列产品产品使用"ascend950"。105+ - --soc:\$\{soc\_version\}表示NPU型号。Atlas A2系列产品使用"ascend910b"(默认),Atlas A3系列产品使用"ascend910_93",Ascend 950PR&950DT系列产品产品使用"ascend950"。
106 - --experimental(可选):表示编译用户保存在experimental目录下的算子。106 - --experimental(可选):表示编译用户保存在experimental目录下的算子。
107 - -j(可选):指定编译线程数,加快编译速度。107 - -j(可选):指定编译线程数,加快编译速度。
108 108 
@@ -5,7 +5,7 @@
5| 产品 | 是否支持 |5| 产品 | 是否支持 |
6| ---- | :----:|6| ---- | :----:|
7|Ascend 950PR&950DT系列产品|√|7|Ascend 950PR&950DT系列产品|√|
8-|Atlas A3 训练系列产品/Atlas A3 推理系列产品|√|8+|Atlas A3系列产品|√|
9|Atlas A2系列产品|√|9|Atlas A2系列产品|√|
10|Atlas 200I/500 A2推理产品|×|10|Atlas 200I/500 A2推理产品|×|
11|Atlas推理系列产品|×|11|Atlas推理系列产品|×|
@@ -5,7 +5,7 @@
5| 产品 | 是否支持 |5| 产品 | 是否支持 |
6| ---- | :----:|6| ---- | :----:|
7|Ascend 950PR&950DT系列产品|×|7|Ascend 950PR&950DT系列产品|×|
8-|Atlas A3 训练系列产品/Atlas A3 推理系列产品|√|8+|Atlas A3系列产品|√|
9|Atlas A2系列产品|√|9|Atlas A2系列产品|√|
10|Atlas 200I/500 A2推理产品|×|10|Atlas 200I/500 A2推理产品|×|
11|Atlas推理系列产品|×|11|Atlas推理系列产品|×|
@@ -5,7 +5,7 @@
5|产品|是否支持|5|产品|是否支持|
6|:---|:---:|6|:---|:---:|
7|<term>Ascend 950PR&950DT系列产品</term>|×|7|<term>Ascend 950PR&950DT系列产品</term>|×|
8-|<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>|√|8+|<term>Atlas A3系列产品</term>|√|
9|<term>Atlas A2系列产品</term>|√|9|<term>Atlas A2系列产品</term>|√|
10|<term>Atlas 200I/500 A2推理产品</term>|×|10|<term>Atlas 200I/500 A2推理产品</term>|×|
11|<term>Atlas推理系列产品</term>|√|11|<term>Atlas推理系列产品</term>|√|
@@ -5,7 +5,7 @@
5|产品|是否支持|5|产品|是否支持|
6|:---|:---:|6|:---|:---:|
7|<term>Ascend 950PR&950DT系列产品</term>|√|7|<term>Ascend 950PR&950DT系列产品</term>|√|
8-|<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>|√|8+|<term>Atlas A3系列产品</term>|√|
9|<term>Atlas A2系列产品</term>|√|9|<term>Atlas A2系列产品</term>|√|
10|<term>Atlas 200I/500 A2推理产品</term>|×|10|<term>Atlas 200I/500 A2推理产品</term>|×|
11|<term>Atlas推理系列产品</term>|√|11|<term>Atlas推理系列产品</term>|√|
@@ -5,7 +5,7 @@
5| 产品 | 是否支持 |5| 产品 | 是否支持 |
6| :--- | :---: |6| :--- | :---: |
7| Atlas A2训练系列产品 | √ |7| Atlas A2训练系列产品 | √ |
8-| Atlas A3 系列产品 | √ |8+| Atlas A3系列产品 | √ |
9 9 
10## 功能说明10## 功能说明
11 11 
@@ -5,7 +5,7 @@
5| 产品 | 是否支持 |5| 产品 | 是否支持 |
6| :--- | :---: |6| :--- | :---: |
7| Atlas A2训练系列产品 | √ |7| Atlas A2训练系列产品 | √ |
8-| Atlas A3 系列产品 | √ |8+| Atlas A3系列产品 | √ |
9 9 
10## 功能描述10## 功能描述
11 11 
@@ -5,7 +5,7 @@
5|产品|是否支持|5|产品|是否支持|
6|:---|:---:|6|:---|:---:|
7|<term>Ascend 950PR&950DT系列产品</term>|√|7|<term>Ascend 950PR&950DT系列产品</term>|√|
8-|<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>|√|8+|<term>Atlas A3系列产品</term>|√|
9|<term>Atlas A2系列产品</term>|√|9|<term>Atlas A2系列产品</term>|√|
10|<term>Atlas 200I/500 A2推理产品</term>|×|10|<term>Atlas 200I/500 A2推理产品</term>|×|
11|<term>Atlas推理系列产品</term>|√|11|<term>Atlas推理系列产品</term>|√|
@@ -131,7 +131,7 @@ aclnnStatus aclnnMishBackward(
131 </table>131 </table>
132 132 
133 - <term>Atlas训练系列产品</term>:数据类型支持 FLOAT16、FLOAT32。133 - <term>Atlas训练系列产品</term>:数据类型支持 FLOAT16、FLOAT32。
134- - <term>Ascend 950PR&950DT系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>、<term>Atlas A2系列产品</term>:额外支持 BFLOAT16。134+ - <term>Ascend 950PR&950DT系列产品</term>、<term>Atlas A3系列产品</term>、<term>Atlas A2系列产品</term>:额外支持 BFLOAT16。
135 135 
136- **返回值:**136- **返回值:**
137 137 
@@ -5,7 +5,7 @@
5|产品|是否支持|5|产品|是否支持|
6|:---|:---:|6|:---|:---:|
7|<term>Ascend 950PR&950DT系列产品</term>|√|7|<term>Ascend 950PR&950DT系列产品</term>|√|
8-|<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>|√|8+|<term>Atlas A3系列产品</term>|√|
9|<term>Atlas A2系列产品</term>|√|9|<term>Atlas A2系列产品</term>|√|
10|<term>Atlas 200I/500 A2推理产品</term>|×|10|<term>Atlas 200I/500 A2推理产品</term>|×|
11|<term>Atlas推理系列产品</term>|√|11|<term>Atlas推理系列产品</term>|√|
@@ -5,7 +5,7 @@
5|产品|是否支持|5|产品|是否支持|
6|:---|:---:|6|:---|:---:|
7|<term>Ascend 950PR&950DT系列产品</term>|√|7|<term>Ascend 950PR&950DT系列产品</term>|√|
8-|<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>|√|8+|<term>Atlas A3系列产品</term>|√|
9|<term>Atlas A2系列产品</term>|√|9|<term>Atlas A2系列产品</term>|√|
10|<term>Atlas 200I/500 A2推理产品</term>|×|10|<term>Atlas 200I/500 A2推理产品</term>|×|
11|<term>Atlas推理系列产品</term>|√|11|<term>Atlas推理系列产品</term>|√|
@@ -5,7 +5,7 @@
5|产品|是否支持|5|产品|是否支持|
6|:---|:---:|6|:---|:---:|
7|<term>Ascend 950PR&950DT系列产品</term>|√|7|<term>Ascend 950PR&950DT系列产品</term>|√|
8-|<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>|√|8+|<term>Atlas A3系列产品</term>|√|
9|<term>Atlas A2系列产品</term>|√|9|<term>Atlas A2系列产品</term>|√|
10|<term>Atlas 200I/500 A2推理产品</term>|×|10|<term>Atlas 200I/500 A2推理产品</term>|×|
11|<term>Atlas推理系列产品</term>|√|11|<term>Atlas推理系列产品</term>|√|
@@ -5,7 +5,7 @@
5| 产品 | 是否支持 |5| 产品 | 是否支持 |
6|:--|:--:|6|:--|:--:|
7| Atlas A2训练系列产品 | √ |7| Atlas A2训练系列产品 | √ |
8-| Atlas A3 系列产品 | √ |8+| Atlas A3系列产品 | √ |
9| 其他产品 | × |9| 其他产品 | × |
10 10 
11## 功能说明11## 功能说明
@@ -5,7 +5,7 @@
5| 产品 | 是否支持 |5| 产品 | 是否支持 |
6|:--|:--:|6|:--|:--:|
7| Atlas A2训练系列产品 | √ |7| Atlas A2训练系列产品 | √ |
8-| Atlas A3 系列产品 | √ |8+| Atlas A3系列产品 | √ |
9| 其他产品 | × |9| 其他产品 | × |
10 10 
11## 功能说明11## 功能说明
@@ -59,7 +59,7 @@ $$
59 </tbody>59 </tbody>
60 </table>60 </table>
61 61 
62- - <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>、<term>Atlas A2系列产品</term>:数据类型支持FLOAT、FLOAT16、BFLOAT16、INT16、INT8、UINT8。62+ - <term>Atlas A3系列产品</term>、<term>Atlas A2系列产品</term>:数据类型支持FLOAT、FLOAT16、BFLOAT16、INT16、INT8、UINT8。
63 63 
64## 约束说明64## 约束说明
65 65 
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -141,7 +141,7 @@ aclnnStatus aclnnInplaceSigmoid(
141 </table>141 </table>
142 142 
143 <!-- npu="A3,910b" id7 -->143 <!-- npu="A3,910b" id7 -->
144- - <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>、<term>Atlas A2系列产品</term>:数据类型支持FLOAT、FLOAT16、BFLOAT16、INT16、INT8、UINT8。144+ - <term>Atlas A3系列产品</term>、<term>Atlas A2系列产品</term>:数据类型支持FLOAT、FLOAT16、BFLOAT16、INT16、INT8、UINT8。
145 <!-- end id7 -->145 <!-- end id7 -->
146 146 
147- **返回值:**147- **返回值:**
@@ -284,7 +284,7 @@ aclnnStatus aclnnInplaceSigmoid(
284 - <term>Atlas推理系列产品</term>、<term>Atlas训练系列产品</term>:数据类型支持FLOAT、FLOAT16、DOUBLE、COMPLEX64、COMPLEX128。284 - <term>Atlas推理系列产品</term>、<term>Atlas训练系列产品</term>:数据类型支持FLOAT、FLOAT16、DOUBLE、COMPLEX64、COMPLEX128。
285 <!-- end id8 -->285 <!-- end id8 -->
286 <!-- npu="A3,910b" id9 -->286 <!-- npu="A3,910b" id9 -->
287- - <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>、<term>Atlas A2系列产品</term>:数据类型支持FLOAT、FLOAT16、DOUBLE、COMPLEX64、COMPLEX128、BFLOAT16、INT16、INT8、UINT8。287+ - <term>Atlas A3系列产品</term>、<term>Atlas A2系列产品</term>:数据类型支持FLOAT、FLOAT16、DOUBLE、COMPLEX64、COMPLEX128、BFLOAT16、INT16、INT8、UINT8。
288 <!-- end id9 -->288 <!-- end id9 -->
289 289 
290- **返回值:**290- **返回值:**
@@ -5,7 +5,7 @@
5|产品|是否支持|5|产品|是否支持|
6|:---|:---:|6|:---|:---:|
7|<term>Ascend 950PR&950DT系列产品</term>|√|7|<term>Ascend 950PR&950DT系列产品</term>|√|
8-|<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>|√|8+|<term>Atlas A3系列产品</term>|√|
9|<term>Atlas A2系列产品</term>|√|9|<term>Atlas A2系列产品</term>|√|
10|<term>Atlas 200I/500 A2推理产品</term>|×|10|<term>Atlas 200I/500 A2推理产品</term>|×|
11|<term>Atlas推理系列产品</term>|√|11|<term>Atlas推理系列产品</term>|√|
@@ -5,7 +5,7 @@
5| 产品 | 是否支持 |5| 产品 | 是否支持 |
6| :--- | :------: |6| :--- | :------: |
7| Ascend 950PR&950DT系列产品 | 是 |7| Ascend 950PR&950DT系列产品 | 是 |
8-| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | 是 |8+| Atlas A3系列产品 | 是 |
9| Atlas A2系列产品 | 是 |9| Atlas A2系列产品 | 是 |
10 10 
11## 功能说明11## 功能说明
@@ -4,7 +4,7 @@
4 4 
5|产品|是否支持|5|产品|是否支持|
6|:---|:---:|6|:---|:---:|
7-|Atlas A3 训练系列产品/Atlas A3 推理系列产品|√|7+|Atlas A3系列产品|√|
8|Atlas A2系列产品|√|8|Atlas A2系列产品|√|
9 9 
10## 功能说明10## 功能说明
@@ -5,7 +5,7 @@
5|产品|是否支持|5|产品|是否支持|
6|:---|:---:|6|:---|:---:|
7|<term>Ascend 950PR&950DT系列产品</term>|√|7|<term>Ascend 950PR&950DT系列产品</term>|√|
8-|<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>|√|8+|<term>Atlas A3系列产品</term>|√|
9|<term>Atlas A2系列产品</term>|√|9|<term>Atlas A2系列产品</term>|√|
10|<term>Atlas 200I/500 A2推理产品</term>|×|10|<term>Atlas 200I/500 A2推理产品</term>|×|
11|<term>Atlas推理系列产品</term>|√|11|<term>Atlas推理系列产品</term>|√|
@@ -5,7 +5,7 @@
5| 产品 | 是否支持 |5| 产品 | 是否支持 |
6| ---- | :----:|6| ---- | :----:|
7| Ascend 950PR/Ascend 950DT | √ |7| Ascend 950PR/Ascend 950DT | √ |
8-| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | × |8+| Atlas A3系列产品 | × |
9| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | × |9| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | × |
10| Atlas 200I/500 A2 推理产品 | × |10| Atlas 200I/500 A2 推理产品 | × |
11| Atlas 推理系列产品 | × |11| Atlas 推理系列产品 | × |
@@ -4,8 +4,8 @@
4 4 
5| 产品 | 是否支持 |5| 产品 | 是否支持 |
6| ---- | ------- |6| ---- | ------- |
7-| Ascend 950PR(本 v2 新增) | √ |7+| Ascend 950PR | √ |
8-| Atlas A2 训练系列产品/Atlas 800I A2 推理产品 | ×(请使用 experimental/index/masked_scatter) |8+| Atlas A2 训练系列产品/Atlas 800I A2推理产品 | ×(请使用 experimental/index/masked_scatter) |
9 9 
10## 功能说明10## 功能说明
11 11 
@@ -4,7 +4,7 @@
4 4 
5| 产品 | 是否支持 |5| 产品 | 是否支持 |
6| ---- | :----:|6| ---- | :----:|
7-|Atlas A3 训练系列产品/Atlas A3 推理系列产品|√|7+|Atlas A3系列产品|√|
8|Atlas A2系列产品|√|8|Atlas A2系列产品|√|
9|Ascend 950PR&950DT系列产品|×|9|Ascend 950PR&950DT系列产品|×|
10|Atlas 200I/500 A2推理产品|×|10|Atlas 200I/500 A2推理产品|×|
@@ -2,7 +2,7 @@
2 2 
3## 支持的产品型号3## 支持的产品型号
4 4 
5-- Atlas A2训练系列产品 / Atlas A3 系列产品。5+- Atlas A2训练系列产品 / Atlas A3系列产品。
6 6 
7## 接口原型7## 接口原型
8 8 
@@ -6,7 +6,7 @@
6| :----------------------------------------------------------- | :------: |6| :----------------------------------------------------------- | :------: |
7| <term>Ascend 950PR&950DT系列产品</term> | × |7| <term>Ascend 950PR&950DT系列产品</term> | × |
8| <term>Atlas A3系列产品</term>| √ |8| <term>Atlas A3系列产品</term>| √ |
9-| <term>Atlas A2系列产品/A200I A2 Box 异构组件</term> | √ |9+| <term>Atlas A2系列产品/A200I A2 Box异构组件</term> | √ |
10| <term>Atlas 200I/500 A2推理产品</term> | × |10| <term>Atlas 200I/500 A2推理产品</term> | × |
11| <term>Atlas推理系列产品</term> | × |11| <term>Atlas推理系列产品</term> | × |
12| <term>Atlas训练系列产品</term> | × |12| <term>Atlas训练系列产品</term> | × |
@@ -50,7 +50,7 @@
50- `x` 和 `y` 必须为 2 维张量,`indices`、`sort_idx` 必须为 1 维张量。50- `x` 和 `y` 必须为 2 维张量,`indices`、`sort_idx` 必须为 1 维张量。
51- 隐藏维度 H 需满足:`align(H) * (element_size * 3 + 4 * 3) + 32 <= 180KB`(受 UB 容量限制,其中 `align(H)` 为 H 按 32 字节对齐后的元素个数,`element_size` 为 2 字节)。51- 隐藏维度 H 需满足:`align(H) * (element_size * 3 + 4 * 3) + 32 <= 180KB`(受 UB 容量限制,其中 `align(H)` 为 H 按 32 字节对齐后的元素个数,`element_size` 为 2 字节)。
52- `sort_idx` 必须是 `indices` 的 argsort 结果,即 `sort_idx = argsort(indices)`。52- `sort_idx` 必须是 `indices` 的 argsort 结果,即 `sort_idx = argsort(indices)`。
53-- 当前支持 Atlas A2 和 Atlas A3 系列产品。53+- 当前支持 Atlas A2 和 Atlas A3系列产品。
54 54 
55## 调用说明55## 调用说明
56 56 
@@ -6,7 +6,7 @@
6- <term>Ascend 950PR&950DT系列产品</term>:不支持6- <term>Ascend 950PR&950DT系列产品</term>:不支持
7<!-- end id1 -->7<!-- end id1 -->
8<!-- npu="A3" id2 -->8<!-- npu="A3" id2 -->
9-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持9+- <term>Atlas A3系列产品</term>:支持
10<!-- end id2 -->10<!-- end id2 -->
11<!-- npu="910b" id3 -->11<!-- npu="910b" id3 -->
12- <term>Atlas A2系列产品</term>:支持12- <term>Atlas A2系列产品</term>:支持
@@ -44,7 +44,7 @@
44- `x`、`s`、`indices`、`x_grad`、`s_grad` 的 `dim[0]` 必须相同。44- `x`、`s`、`indices`、`x_grad`、`s_grad` 的 `dim[0]` 必须相同。
45- `y_grad` 和 `x` 必须为 2 维张量,`s` 和 `indices` 必须为 1 维张量。45- `y_grad` 和 `x` 必须为 2 维张量,`s` 和 `indices` 必须为 1 维张量。
46- 隐藏维度 H 需满足:`align(H) * (element_size * 4 + 4 * 4) + 32 <= 180KB`(受 UB 容量限制,其中 `align(H)` 为 H 按 32 字节对齐后的元素个数,`element_size` 为 2 字节)。46- 隐藏维度 H 需满足:`align(H) * (element_size * 4 + 4 * 4) + 32 <= 180KB`(受 UB 容量限制,其中 `align(H)` 为 H 按 32 字节对齐后的元素个数,`element_size` 为 2 字节)。
47-- 当前支持 Atlas A2 和 Atlas A3 系列产品。47+- 当前支持 Atlas A2 和 Atlas A3系列产品。
48 48 
49## 调用说明49## 调用说明
50 50 
@@ -6,7 +6,7 @@
6- <term>Ascend 950PR&950DT系列产品</term>:不支持6- <term>Ascend 950PR&950DT系列产品</term>:不支持
7<!-- end id1 -->7<!-- end id1 -->
8<!-- npu="A3" id2 -->8<!-- npu="A3" id2 -->
9-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持9+- <term>Atlas A3系列产品</term>:支持
10<!-- end id2 -->10<!-- end id2 -->
11<!-- npu="910b" id3 -->11<!-- npu="910b" id3 -->
12- <term>Atlas A2系列产品</term>:支持12- <term>Atlas A2系列产品</term>:支持
@@ -37,7 +37,7 @@
37- `var`、`indices` 和 `updates` 的维度数应一致。37- `var`、`indices` 和 `updates` 的维度数应一致。
38- `indices` 与 `updates` 的形状应一致,非 `axis` 轴的大小不得超过 `var` 对应轴的大小。38- `indices` 与 `updates` 的形状应一致,非 `axis` 轴的大小不得超过 `var` 对应轴的大小。
39- `indices` 中的值必须位于 `[-var.shape[axis], var.shape[axis])` 范围内。39- `indices` 中的值必须位于 `[-var.shape[axis], var.shape[axis])` 范围内。
40-- 当前支持 Atlas A2 和 Atlas A3 系列产品。40+- 当前支持 Atlas A2 和 Atlas A3系列产品。
41 41 
42## 调用说明42## 调用说明
43 43 
@@ -6,7 +6,7 @@
6- <term>Ascend 950PR&950DT系列产品</term>:支持6- <term>Ascend 950PR&950DT系列产品</term>:支持
7<!-- end id1 -->7<!-- end id1 -->
8<!-- npu="A3" id2 -->8<!-- npu="A3" id2 -->
9-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持9+- <term>Atlas A3系列产品</term>:支持
10<!-- end id2 -->10<!-- end id2 -->
11<!-- npu="910b" id3 -->11<!-- npu="910b" id3 -->
12- <term>Atlas A2系列产品</term>:支持12- <term>Atlas A2系列产品</term>:支持
@@ -6,7 +6,7 @@
6- <term>Ascend 950PR&950DT系列产品</term>:不支持6- <term>Ascend 950PR&950DT系列产品</term>:不支持
7<!-- end id1 -->7<!-- end id1 -->
8<!-- npu="A3" id2 -->8<!-- npu="A3" id2 -->
9-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持9+- <term>Atlas A3系列产品</term>:支持
10<!-- end id2 -->10<!-- end id2 -->
11<!-- npu="910b" id3 -->11<!-- npu="910b" id3 -->
12- <term>Atlas A2系列产品</term>:支持12- <term>Atlas A2系列产品</term>:支持
@@ -5,7 +5,7 @@
5| 产品 | 是否支持 |5| 产品 | 是否支持 |
6| ---- | :----:|6| ---- | :----:|
7|Ascend 950PR&950DT系列产品|×|7|Ascend 950PR&950DT系列产品|×|
8-|Atlas A3 训练系列产品/Atlas A3 推理系列产品|×|8+|Atlas A3系列产品|×|
9|Atlas A2系列产品|√|9|Atlas A2系列产品|√|
10|Atlas 200I/500 A2推理产品|×|10|Atlas 200I/500 A2推理产品|×|
11|Atlas推理系列产品|×|11|Atlas推理系列产品|×|
@@ -43,7 +43,7 @@ $$
43 43 
44## 本地编译运行 UT44## 本地编译运行 UT
45 45 
46-环境准备请参考项目环境部署文档。Atlas A2 系列产品使用 `ascend910b`;其他可选 SoC 为 `ascend910_93` 和 `ascend950`。46+环境准备请参考项目环境部署文档。Atlas A2系列产品使用 `ascend910b`;其他可选 SoC 为 `ascend910_93` 和 `ascend950`。
47 47 
48```bash48```bash
49# Host UT:注册、shape/type 推导和 tiling49# Host UT:注册、shape/type 推导和 tiling
@@ -5,7 +5,7 @@
5| 产品 | 是否支持 |5| 产品 | 是否支持 |
6| ---- | :----:|6| ---- | :----:|
7|Ascend 950PR&950DT系列产品|×|7|Ascend 950PR&950DT系列产品|×|
8-|Atlas A3 训练系列产品/Atlas A3 推理系列产品|×|8+|Atlas A3系列产品|×|
9|Atlas A2系列产品|√|9|Atlas A2系列产品|√|
10|Atlas 200I/500 A2推理产品|×|10|Atlas 200I/500 A2推理产品|×|
11|Atlas推理系列产品|×|11|Atlas推理系列产品|×|
@@ -4,7 +4,7 @@
4 4 
5### 1.1 需求来源5### 1.1 需求来源
6 6 
7-通过 CANN 训练营 2026 暑期季社区任务完成 `MseLoss` 算子 Ascend C 实现,并向昇腾开源算子仓贡献。任务要求参考昇腾 CANN 内置 `aclnnMseLoss` / TBE 算子语义,在 Atlas A2训练系列产品和 Atlas A3 系列产品上实现功能一致、精度达标、性能不低于 TBE 版本的 Ascend C 自定义算子。7+通过 CANN 训练营 2026 暑期季社区任务完成 `MseLoss` 算子 Ascend C 实现,并向昇腾开源算子仓贡献。任务要求参考昇腾 CANN 内置 `aclnnMseLoss` / TBE 算子语义,在 Atlas A2训练系列产品和 Atlas A3系列产品上实现功能一致、精度达标、性能不低于 TBE 版本的 Ascend C 自定义算子。
8 8 
9### 1.2 背景介绍9### 1.2 背景介绍
10 10 
@@ -437,7 +437,7 @@ flowchart TD
437| 支持的芯片版本 | 涉及勾选 |437| 支持的芯片版本 | 涉及勾选 |
438| --- | --- |438| --- | --- |
439| Atlas A2训练系列产品 | √ |439| Atlas A2训练系列产品 | √ |
440-| Atlas A3 系列产品 | √ |440+| Atlas A3系列产品 | √ |
441 441 
442### 3.4 算子约束限制442### 3.4 算子约束限制
443 443 
@@ -5,7 +5,7 @@
5| 产品 | 是否支持 |5| 产品 | 是否支持 |
6|:---|:---:|6|:---|:---:|
7| Atlas A2系列产品 | ✓ |7| Atlas A2系列产品 | ✓ |
8-| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | ✓ |8+| Atlas A3系列产品 | ✓ |
9 9 
10## 功能说明10## 功能说明
11 11 
@@ -13,7 +13,7 @@ kernel 计算逻辑改编自经过验证的 pto-ISA 参考实现(`pto-kernels/
13| 产品 | 是否支持 |13| 产品 | 是否支持 |
14| ---- | :----: |14| ---- | :----: |
15| Atlas A2系列产品 | √ |15| Atlas A2系列产品 | √ |
16-| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |16+| Atlas A3系列产品 | √ |
17 17 
18> 说明:A2(`ascend910b`)与 A3(`ascend910_93`)共享 `dav-2201` 架构与 pto `npu/a2a3` 后端,同一二进制适用于两者;当前在 A2 上完成验证。kernel 内 UB 布局按 A2/A3 的 184KB Unified Buffer 规划,`blockDim` 采用 A2/A3 的向量核数(40)作为启动网格大小,`batch` 在 kernel 内按实际核数切分,故该常量仅影响启动规模、不影响正确性。18> 说明:A2(`ascend910b`)与 A3(`ascend910_93`)共享 `dav-2201` 架构与 pto `npu/a2a3` 后端,同一二进制适用于两者;当前在 A2 上完成验证。kernel 内 UB 布局按 A2/A3 的 184KB Unified Buffer 规划,`blockDim` 采用 A2/A3 的向量核数(40)作为启动网格大小,`batch` 在 kernel 内按实际核数切分,故该常量仅影响启动规模、不影响正确性。
19 19 
@@ -11,7 +11,7 @@
11| 产品 | 是否支持 |11| 产品 | 是否支持 |
12| ---- | :----: |12| ---- | :----: |
13| Atlas A2系列产品 | √ |13| Atlas A2系列产品 | √ |
14-| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |14+| Atlas A3系列产品 | √ |
15 15 
16> 说明:A2(`ascend910b`)与 A3(`ascend910_93`)共享 `dav-2201` 架构与 pto `npu/a2a3` 后端,同一二进制适用于两者;当前在 A2 上完成验证。16> 说明:A2(`ascend910b`)与 A3(`ascend910_93`)共享 `dav-2201` 架构与 pto `npu/a2a3` 后端,同一二进制适用于两者;当前在 A2 上完成验证。
17 17 
@@ -11,7 +11,7 @@
11| 产品 | 是否支持 |11| 产品 | 是否支持 |
12| ---- | :----: |12| ---- | :----: |
13| Atlas A2系列产品 | √ |13| Atlas A2系列产品 | √ |
14-| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | √ |14+| Atlas A3系列产品 | √ |
15 15 
16> 说明:A2(`ascend910b`)与 A3(`ascend910_93`)共享 `dav-2201` 架构与 pto `npu/a2a3` 后端,同一二进制适用于两者;当前在 A2 上完成验证。16> 说明:A2(`ascend910b`)与 A3(`ascend910_93`)共享 `dav-2201` 架构与 pto `npu/a2a3` 后端,同一二进制适用于两者;当前在 A2 上完成验证。
17 17 
@@ -8,7 +8,7 @@
8 8 
9| 产品 | 是否支持 |9| 产品 | 是否支持 |
10| ---- | :----:|10| ---- | :----:|
11-|Atlas A3 训练系列产品/Atlas A3 推理系列产品|√|11+|Atlas A3系列产品|√|
12|Atlas A2系列产品|√|12|Atlas A2系列产品|√|
13 13 
14## 目录结构介绍14## 目录结构介绍
@@ -5,7 +5,7 @@
5| 产品 | 是否支持 |5| 产品 | 是否支持 |
6| ---- | :----:|6| ---- | :----:|
7|Ascend 950PR&950DT系列产品|×|7|Ascend 950PR&950DT系列产品|×|
8-|Atlas A3 训练系列产品/Atlas A3 推理系列产品|×|8+|Atlas A3系列产品|×|
9|Atlas A2系列产品|√|9|Atlas A2系列产品|√|
10|Atlas 200I/500 A2推理产品|×|10|Atlas 200I/500 A2推理产品|×|
11|Atlas推理系列产品|×|11|Atlas推理系列产品|×|
@@ -5,7 +5,7 @@
5| 产品 | 是否支持 |5| 产品 | 是否支持 |
6| ---- | :----:|6| ---- | :----:|
7|Ascend 950PR&950DT系列产品|×|7|Ascend 950PR&950DT系列产品|×|
8-|Atlas A3 训练系列产品/Atlas A3 推理系列产品|×|8+|Atlas A3系列产品|×|
9|Atlas A2系列产品|√|9|Atlas A2系列产品|√|
10|Atlas 200I/500 A2推理产品|×|10|Atlas 200I/500 A2推理产品|×|
11|Atlas推理系列产品|×|11|Atlas推理系列产品|×|
@@ -6,7 +6,7 @@
6- <term>Ascend 950PR&950DT系列产品</term>:支持6- <term>Ascend 950PR&950DT系列产品</term>:支持
7<!-- end id1 -->7<!-- end id1 -->
8<!-- npu="A3" id2 -->8<!-- npu="A3" id2 -->
9-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持9+- <term>Atlas A3系列产品</term>:支持
10<!-- end id2 -->10<!-- end id2 -->
11<!-- npu="910b" id3 -->11<!-- npu="910b" id3 -->
12- <term>Atlas A2系列产品</term>:支持12- <term>Atlas A2系列产品</term>:支持
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:不支持11+- <term>Atlas A3系列产品</term>:不支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:不支持14- <term>Atlas A2系列产品</term>:不支持
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:不支持11+- <term>Atlas A3系列产品</term>:不支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:不支持14- <term>Atlas A2系列产品</term>:不支持
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -5,7 +5,7 @@
5| 产品 | 是否支持 |5| 产品 | 是否支持 |
6| ---- | :----:|6| ---- | :----:|
7|Ascend 950PR&950DT系列产品|√|7|Ascend 950PR&950DT系列产品|√|
8-|Atlas A3 训练系列产品/Atlas A3 推理系列产品|√|8+|Atlas A3系列产品|√|
9|Atlas A2系列产品|√|9|Atlas A2系列产品|√|
10|Atlas 200I/500 A2推理产品|×|10|Atlas 200I/500 A2推理产品|×|
11|Atlas推理系列产品|√|11|Atlas推理系列产品|√|
@@ -6,7 +6,7 @@
6- <term>Ascend 950PR&950DT系列产品</term>:支持6- <term>Ascend 950PR&950DT系列产品</term>:支持
7<!-- end id1 -->7<!-- end id1 -->
8<!-- npu="A3" id2 -->8<!-- npu="A3" id2 -->
9-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持9+- <term>Atlas A3系列产品</term>:支持
10<!-- end id2 -->10<!-- end id2 -->
11<!-- npu="910b" id3 -->11<!-- npu="910b" id3 -->
12- <term>Atlas A2系列产品</term>:支持12- <term>Atlas A2系列产品</term>:支持
@@ -28,7 +28,7 @@
28 假设输入weight的shape为(numWeight, embeddingDim),indices的shape为(bagIndices),offsets的shape为(bagOffsets)。28 假设输入weight的shape为(numWeight, embeddingDim),indices的shape为(bagIndices),offsets的shape为(bagOffsets)。
29 29 
30 <!-- npu="A3,910b,910,310p" id7 -->30 <!-- npu="A3,910b,910,310p" id7 -->
31- - <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>、<term>Atlas A2系列产品</term>、<term>Atlas推理系列产品</term>、<term>Atlas训练系列产品</term>:31+ - <term>Atlas A3系列产品</term>、<term>Atlas A2系列产品</term>、<term>Atlas推理系列产品</term>、<term>Atlas训练系列产品</term>:
32 32 
33 - 当mode为sum模式:33 - 当mode为sum模式:
34 34 
@@ -300,7 +300,7 @@ aclnnStatus aclnnEmbeddingBag(
300 </tbody></table>300 </tbody></table>
301 301 
302 <!-- npu="A3,910b,910,310p" id9 -->302 <!-- npu="A3,910b,910,310p" id9 -->
303- - <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>、<term>Atlas A2系列产品</term>、<term>Atlas推理系列产品</term>、<term>Atlas训练系列产品</term>: indices仅支持0-1维,perSampleWeights仅支持1维且在除sum模式外的其他模式必须为nullptr 。303+ - <term>Atlas A3系列产品</term>、<term>Atlas A2系列产品</term>、<term>Atlas推理系列产品</term>、<term>Atlas训练系列产品</term>: indices仅支持0-1维,perSampleWeights仅支持1维且在除sum模式外的其他模式必须为nullptr 。
304 - indices和offsets的数据类型至少一个为INT32或INT64;当数据类型为INT16、INT8、UINT8时,接口内部会转换为INT32进行计算。304 - indices和offsets的数据类型至少一个为INT32或INT64;当数据类型为INT16、INT8、UINT8时,接口内部会转换为INT32进行计算。
305 305 
306 <!-- end id9 -->306 <!-- end id9 -->
@@ -95,7 +95,7 @@
95 $$95 $$
96 countsSize = numWeights / coreNum + numWeights \% coreNum96 countsSize = numWeights / coreNum + numWeights \% coreNum
97 $$97 $$
98-- <term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:98+- <term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term>:
99 - 在参数shape超过以下限制时,输出无法保证高精度,若开启了确定性计算,也无法保证高性能99 - 在参数shape超过以下限制时,输出无法保证高精度,若开启了确定性计算,也无法保证高性能
100 - grad合轴成二维shape后,第一个维度超过INT32_MAX(2147483647)100 - grad合轴成二维shape后,第一个维度超过INT32_MAX(2147483647)
101 - numWeights超过INT32_MAX(2147483647)101 - numWeights超过INT32_MAX(2147483647)
@@ -6,7 +6,7 @@
6- <term>Ascend 950PR&950DT系列产品</term>:支持6- <term>Ascend 950PR&950DT系列产品</term>:支持
7<!-- end id1 -->7<!-- end id1 -->
8<!-- npu="A3" id2 -->8<!-- npu="A3" id2 -->
9-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持9+- <term>Atlas A3系列产品</term>:支持
10<!-- end id2 -->10<!-- end id2 -->
11<!-- npu="910b" id3 -->11<!-- npu="910b" id3 -->
12- <term>Atlas A2系列产品</term>:支持12- <term>Atlas A2系列产品</term>:支持
@@ -42,7 +42,7 @@
42 - <term>Atlas训练系列产品</term>:数据类型支持FLOAT16、FLOAT。42 - <term>Atlas训练系列产品</term>:数据类型支持FLOAT16、FLOAT。
43 <!-- end id7 -->43 <!-- end id7 -->
44 <!-- npu="950,A3,910b" id8 -->44 <!-- npu="950,A3,910b" id8 -->
45- - <term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持BFLOAT16、FLOAT16、FLOAT。45+ - <term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持BFLOAT16、FLOAT16、FLOAT。
46 <!-- end id8 -->46 <!-- end id8 -->
47 47 
48 - indices(aclTensor*,计算输入):grad输入对应的索引值,Device侧的aclTensor,取值范围为[0, numWeights),支持维度1-8维,支持[非连续的Tensor](../../../docs/zh/context/non_contiguous_tensor.md),[数据格式](../../../docs/zh/context/data_format.md)支持ND。数据类型支持FLOAT、FLOAT16、DOUBLE、INT32、INT64、INT16、INT8、UINT8、BOOL。48 - indices(aclTensor*,计算输入):grad输入对应的索引值,Device侧的aclTensor,取值范围为[0, numWeights),支持维度1-8维,支持[非连续的Tensor](../../../docs/zh/context/non_contiguous_tensor.md),[数据格式](../../../docs/zh/context/data_format.md)支持ND。数据类型支持FLOAT、FLOAT16、DOUBLE、INT32、INT64、INT16、INT8、UINT8、BOOL。
@@ -55,7 +55,7 @@
55 - <term>Atlas训练系列产品</term>:数据类型支持FLOAT16、FLOAT。55 - <term>Atlas训练系列产品</term>:数据类型支持FLOAT16、FLOAT。
56 <!-- end id9 -->56 <!-- end id9 -->
57 <!-- npu="950,A3,910b" id10 -->57 <!-- npu="950,A3,910b" id10 -->
58- - <term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持BFLOAT16、FLOAT16、FLOAT。58+ - <term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持BFLOAT16、FLOAT16、FLOAT。
59 <!-- end id10 -->59 <!-- end id10 -->
60 60 
61 - workspaceSize(uint64_t *,出参): 返回需要在Device侧申请的workspace大小。61 - workspaceSize(uint64_t *,出参): 返回需要在Device侧申请的workspace大小。
@@ -142,7 +142,7 @@
142 142 
143<!-- end id11 -->143<!-- end id11 -->
144<!-- npu="A3,910b" id12 -->144<!-- npu="A3,910b" id12 -->
145-- <term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:145+- <term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term>:
146 - 在参数shape超过以下限制时,输出无法保证高精度,若开启了确定性计算,也无法保证高性能146 - 在参数shape超过以下限制时,输出无法保证高精度,若开启了确定性计算,也无法保证高性能
147 - grad合轴成二维shape后,第一个维度超过INT32_MAX(2147483647)147 - grad合轴成二维shape后,第一个维度超过INT32_MAX(2147483647)
148 - numWeights超过INT32_MAX(2147483647)148 - numWeights超过INT32_MAX(2147483647)
@@ -77,11 +77,11 @@
77 </tr>77 </tr>
78 </tbody></table>78 </tbody></table>
79 79 
80-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>、<term>Atlas A2系列产品</term>:grad不支持BFLOAT16、FLOAT16,indices不支持INT64。80+- <term>Atlas A3系列产品</term>、<term>Atlas A2系列产品</term>:grad不支持BFLOAT16、FLOAT16,indices不支持INT64。
81 81 
82## 约束说明82## 约束说明
83 83 
84-- Atlas A3 训练系列产品/Atlas A3 推理系列产品、Atlas A2系列产品:84+- Atlas A3系列产品、Atlas A2系列产品:
85 - 在参数shape超过以下限制时,输出无法保证高精度,若开启了确定性计算,也无法保证高性能。85 - 在参数shape超过以下限制时,输出无法保证高精度,若开启了确定性计算,也无法保证高性能。
86 - grad合轴成二维shape后,第一个维度超过INT32_MAX(2147483647)。86 - grad合轴成二维shape后,第一个维度超过INT32_MAX(2147483647)。
87 - num_weights超过INT32_MAX(2147483647)。87 - num_weights超过INT32_MAX(2147483647)。
@@ -6,7 +6,7 @@
6- <term>Ascend 950PR&950DT系列产品</term>:支持6- <term>Ascend 950PR&950DT系列产品</term>:支持
7<!-- end id1 -->7<!-- end id1 -->
8<!-- npu="A3" id2 -->8<!-- npu="A3" id2 -->
9-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持9+- <term>Atlas A3系列产品</term>:支持
10<!-- end id2 -->10<!-- end id2 -->
11<!-- npu="910b" id3 -->11<!-- npu="910b" id3 -->
12- <term>Atlas A2系列产品</term>:支持12- <term>Atlas A2系列产品</term>:支持
@@ -272,7 +272,7 @@ aclnnStatus aclnnEmbeddingDenseBackward(
272 272 
273<!-- end id8 -->273<!-- end id8 -->
274<!-- npu="A3,910b" id9 -->274<!-- npu="A3,910b" id9 -->
275-- <term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:275+- <term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term>:
276 - 在参数shape超过以下限制时,输出无法保证高精度,若开启了确定性计算,也无法保证高性能276 - 在参数shape超过以下限制时,输出无法保证高精度,若开启了确定性计算,也无法保证高性能
277 - grad合轴成二维shape后,第一个维度超过INT32_MAX(2147483647)277 - grad合轴成二维shape后,第一个维度超过INT32_MAX(2147483647)
278 - numWeights超过INT32_MAX(2147483647)278 - numWeights超过INT32_MAX(2147483647)
@@ -6,7 +6,7 @@
6- <term>Ascend 950PR&950DT系列产品</term>:支持6- <term>Ascend 950PR&950DT系列产品</term>:支持
7<!-- end id1 -->7<!-- end id1 -->
8<!-- npu="A3" id2 -->8<!-- npu="A3" id2 -->
9-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持9+- <term>Atlas A3系列产品</term>:支持
10<!-- end id2 -->10<!-- end id2 -->
11<!-- npu="910b" id3 -->11<!-- npu="910b" id3 -->
12- <term>Atlas A2系列产品</term>:支持12- <term>Atlas A2系列产品</term>:支持
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -93,7 +93,7 @@
93 </tr>93 </tr>
94 </tbody></table>94 </tbody></table>
95 95 
96-- <term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:数据类型不支持DOUBLE、INT16、UINT16、UINT32、UINT64。96+- <term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term>:数据类型不支持DOUBLE、INT16、UINT16、UINT32、UINT64。
97- <term>Atlas训练系列产品</term>、<term>Atlas推理系列产品</term>:数据类型不支持BFLOAT16、DOUBLE、INT16、UINT16、UINT32、UINT64。97- <term>Atlas训练系列产品</term>、<term>Atlas推理系列产品</term>:数据类型不支持BFLOAT16、DOUBLE、INT16、UINT16、UINT32、UINT64。
98 98 
99## 约束说明99## 约束说明
@@ -6,7 +6,7 @@
6- <term>Ascend 950PR&950DT系列产品</term>:支持6- <term>Ascend 950PR&950DT系列产品</term>:支持
7<!-- end id1 -->7<!-- end id1 -->
8<!-- npu="A3" id2 -->8<!-- npu="A3" id2 -->
9-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持9+- <term>Atlas A3系列产品</term>:支持
10<!-- end id2 -->10<!-- end id2 -->
11<!-- npu="910b" id3 -->11<!-- npu="910b" id3 -->
12- <term>Atlas A2系列产品</term>:支持12- <term>Atlas A2系列产品</term>:支持
@@ -166,7 +166,7 @@ aclnnStatus aclnnGatherNd(
166 </tbody></table>166 </tbody></table>
167 167 
168 <!-- npu="A3,910b" id7 -->168 <!-- npu="A3,910b" id7 -->
169- - <term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:数据类型不支持DOUBLE、INT16、UINT16、UINT32、UINT64。169+ - <term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term>:数据类型不支持DOUBLE、INT16、UINT16、UINT32、UINT64。
170 <!-- end id7 -->170 <!-- end id7 -->
171 171 
172- **返回值**172- **返回值**
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -176,7 +176,7 @@ aclnnStatus aclnnGatherV3(
176 </tbody></table>176 </tbody></table>
177 177 
178 <!-- npu="A3,910b" id7 -->178 <!-- npu="A3,910b" id7 -->
179- - <term>Atlas A2系列产品</term>、 <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>: dim当前仅支持0,batchDims当前仅支持0。179+ - <term>Atlas A2系列产品</term>、 <term>Atlas A3系列产品</term>: dim当前仅支持0,batchDims当前仅支持0。
180 180 
181 <!-- end id7 -->181 <!-- end id7 -->
182 182 
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -6,7 +6,7 @@
6- <term>Ascend 950PR&950DT系列产品</term>:支持6- <term>Ascend 950PR&950DT系列产品</term>:支持
7<!-- end id1 -->7<!-- end id1 -->
8<!-- npu="A3" id2 -->8<!-- npu="A3" id2 -->
9-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持9+- <term>Atlas A3系列产品</term>:支持
10<!-- end id2 -->10<!-- end id2 -->
11<!-- npu="910b" id3 -->11<!-- npu="910b" id3 -->
12- <term>Atlas A2系列产品</term>:支持12- <term>Atlas A2系列产品</term>:支持
@@ -190,7 +190,7 @@
190 </tr>190 </tr>
191 </tbody></table>191 </tbody></table>
192 <!-- npu="A3,910b" id7 -->192 <!-- npu="A3,910b" id7 -->
193- - <term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:数据类型支持FLOAT16、FLOAT、INT32、INT64、BOOL、BFLOAT16。193+ - <term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term>:数据类型支持FLOAT16、FLOAT、INT32、INT64、BOOL、BFLOAT16。
194 <!-- end id7 -->194 <!-- end id7 -->
195 <!-- npu="950" id8 -->195 <!-- npu="950" id8 -->
196 - <term>Ascend 950PR&950DT系列产品</term>:数据类型支持FLOAT16、FLOAT、INT32、INT64、BOOL、BFLOAT16、INT8、UINT8、INT16、DOUBLE。196 - <term>Ascend 950PR&950DT系列产品</term>:数据类型支持FLOAT16、FLOAT、INT32、INT64、BOOL、BFLOAT16、INT8、UINT8、INT16、DOUBLE。
@@ -367,7 +367,7 @@
367 </tr>367 </tr>
368 </tbody></table>368 </tbody></table>
369 <!-- npu="A3,910b" id9 -->369 <!-- npu="A3,910b" id9 -->
370- - <term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:数据类型支持FLOAT16、FLOAT、INT32、INT64、BOOL、BFLOAT16。370+ - <term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term>:数据类型支持FLOAT16、FLOAT、INT32、INT64、BOOL、BFLOAT16。
371 <!-- end id9 -->371 <!-- end id9 -->
372 <!-- npu="950" id10 -->372 <!-- npu="950" id10 -->
373 - <term>Ascend 950PR&950DT系列产品</term>:数据类型支持FLOAT16、FLOAT、INT32、INT64、BOOL、BFLOAT16、INT8、UINT8、INT16、DOUBLE。373 - <term>Ascend 950PR&950DT系列产品</term>:数据类型支持FLOAT16、FLOAT、INT32、INT64、BOOL、BFLOAT16、INT8、UINT8、INT16、DOUBLE。
@@ -6,7 +6,7 @@
6- <term>Ascend 950PR&950DT系列产品</term>:支持6- <term>Ascend 950PR&950DT系列产品</term>:支持
7<!-- end id1 -->7<!-- end id1 -->
8<!-- npu="A3" id2 -->8<!-- npu="A3" id2 -->
9-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持9+- <term>Atlas A3系列产品</term>:支持
10<!-- end id2 -->10<!-- end id2 -->
11<!-- npu="910b" id3 -->11<!-- npu="910b" id3 -->
12- <term>Atlas A2系列产品</term>:支持12- <term>Atlas A2系列产品</term>:支持
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -105,7 +105,7 @@
105| 产品 | 参数或场景 | `x`、`v`、`y`数据类型 | 静态shape能力 | 动态shape能力 | 索引、空Tensor及规模限制 |105| 产品 | 参数或场景 | `x`、`v`、`y`数据类型 | 静态shape能力 | 动态shape能力 | 索引、空Tensor及规模限制 |
106| :--- | :----------- | :-------------------- | :------------ | :------------ | :----------------------- |106| :--- | :----------- | :-------------------- | :------------ | :------------ | :----------------------- |
107| <term>Ascend 950PR&950DT系列产品</term> | `indices`为运行期输入或编译期常量 | FLOAT16、FLOAT、BFLOAT16、INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64、COMPLEX32、COMPLEX64 | 输入ND->输出ND;`x`、`v`、`y`支持1~8维,`indices`仅支持1维。 | 输入ND->输出ND;支持动态shape和动态Rank。 | 当$N>0$时,按$row_j=((indices_j \bmod N)+N)\bmod N$确定目标行。支持$K=0$;当$N=0$时要求$K=0$,当任一尾维为0时输出为空。$N$和$K$不超过INT32最大值;尾维大小、张量元素数、合计处理规模及字节大小必须在INT64可表示范围内。 |107| <term>Ascend 950PR&950DT系列产品</term> | `indices`为运行期输入或编译期常量 | FLOAT16、FLOAT、BFLOAT16、INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64、COMPLEX32、COMPLEX64 | 输入ND->输出ND;`x`、`v`、`y`支持1~8维,`indices`仅支持1维。 | 输入ND->输出ND;支持动态shape和动态Rank。 | 当$N>0$时,按$row_j=((indices_j \bmod N)+N)\bmod N$确定目标行。支持$K=0$;当$N=0$时要求$K=0$,当任一尾维为0时输出为空。$N$和$K$不超过INT32最大值;尾维大小、张量元素数、合计处理规模及字节大小必须在INT64可表示范围内。 |
108-| <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term><br><term>Atlas A2系列产品</term><br><term>Atlas 200I/500 A2推理产品</term><br><term>Atlas推理系列产品</term> | `indices`为运行期输入或编译期常量 | FLOAT16、FLOAT、INT32 | 输入ND->输出ND;`x`、`v`、`y`支持1~8维,`indices`仅支持1维。 | 输入ND->输出ND;支持动态shape和动态Rank。 | 要求$0\leq indices_j<N$;不支持空Tensor。 |108+| <term>Atlas A3系列产品</term><br><term>Atlas A2系列产品</term><br><term>Atlas 200I/500 A2推理产品</term><br><term>Atlas推理系列产品</term> | `indices`为运行期输入或编译期常量 | FLOAT16、FLOAT、INT32 | 输入ND->输出ND;`x`、`v`、`y`支持1~8维,`indices`仅支持1维。 | 输入ND->输出ND;支持动态shape和动态Rank。 | 要求$0\leq indices_j<N$;不支持空Tensor。 |
109| <term>Atlas训练系列产品</term> | `x`、`v`、`y`为FLOAT或INT32,`indices`为运行期输入或编译期常量 | FLOAT、INT32 | 输入ND->输出ND;`x`、`v`、`y`支持1~8维,`indices`仅支持1维。 | 输入ND->输出ND;支持动态shape和动态Rank。 | 要求$0\leq indices_j<N$;不支持空Tensor。 |109| <term>Atlas训练系列产品</term> | `x`、`v`、`y`为FLOAT或INT32,`indices`为运行期输入或编译期常量 | FLOAT、INT32 | 输入ND->输出ND;`x`、`v`、`y`支持1~8维,`indices`仅支持1维。 | 输入ND->输出ND;支持动态shape和动态Rank。 | 要求$0\leq indices_j<N$;不支持空Tensor。 |
110| <term>Atlas训练系列产品</term> | `x`、`v`、`y`为FLOAT16,且`indices`在编译期为常量 | FLOAT16 | 仅支持静态shape,输入ND->输出ND;`x`、`v`、`y`支持1~8维,`indices`仅支持1维。 | 不支持。 | 要求$0\leq indices_j<N$;不支持空Tensor。 |110| <term>Atlas训练系列产品</term> | `x`、`v`、`y`为FLOAT16,且`indices`在编译期为常量 | FLOAT16 | 仅支持静态shape,输入ND->输出ND;`x`、`v`、`y`支持1~8维,`indices`仅支持1维。 | 不支持。 | 要求$0\leq indices_j<N$;不支持空Tensor。 |
111 111 
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -7,7 +7,7 @@
7|产品 | 是否支持 |7|产品 | 是否支持 |
8|:----------------------------|:-----------:|8|:----------------------------|:-----------:|
9|<term>Ascend 950PR&950DT系列产品</term>| √ |9|<term>Ascend 950PR&950DT系列产品</term>| √ |
10-|<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>| × |10+|<term>Atlas A3系列产品</term>| × |
11|<term>Atlas A2系列产品</term>| × |11|<term>Atlas A2系列产品</term>| × |
12|<term>Atlas 200I/500 A2推理产品</term>| × |12|<term>Atlas 200I/500 A2推理产品</term>| × |
13|<term>Atlas推理系列产品</term>| × |13|<term>Atlas推理系列产品</term>| × |
@@ -6,7 +6,7 @@
6- <term>Ascend 950PR&950DT系列产品</term>:支持6- <term>Ascend 950PR&950DT系列产品</term>:支持
7<!-- end id1 -->7<!-- end id1 -->
8<!-- npu="A3" id2 -->8<!-- npu="A3" id2 -->
9-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:不支持9+- <term>Atlas A3系列产品</term>:不支持
10<!-- end id2 -->10<!-- end id2 -->
11<!-- npu="910b" id3 -->11<!-- npu="910b" id3 -->
12- <term>Atlas A2系列产品</term>:不支持12- <term>Atlas A2系列产品</term>:不支持
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -209,7 +209,7 @@ aclnnStatus aclnnCtcLoss(
209 - logAlphaOut:209 - logAlphaOut:
210 210 
211 <!-- npu="A3,910b" id7 -->211 <!-- npu="A3,910b" id7 -->
212- - <term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:shape为($N, T, (2*max(targetlengths)+8)/8*8$)。212+ - <term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term>:shape为($N, T, (2*max(targetlengths)+8)/8*8$)。
213 <!-- end id7 -->213 <!-- end id7 -->
214 <!-- npu="950" id8 -->214 <!-- npu="950" id8 -->
215 - <term>Ascend 950PR&950DT系列产品</term>:shape为($N, T, (2*max(targetlengths)+1)$)。215 - <term>Ascend 950PR&950DT系列产品</term>:shape为($N, T, (2*max(targetlengths)+1)$)。
@@ -312,7 +312,7 @@ aclnnStatus aclnnCtcLoss(
312示例代码如下,仅供参考,具体编译和执行过程请参考[编译与运行样例](../../../docs/zh/context/compile_and_run_sample.md)。312示例代码如下,仅供参考,具体编译和执行过程请参考[编译与运行样例](../../../docs/zh/context/compile_and_run_sample.md)。
313 313 
314<!-- npu="A3,910b" id9 -->314<!-- npu="A3,910b" id9 -->
315-- <term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:315+- <term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term>:
316 316 
317 ```Cpp317 ```Cpp
318 #include <iostream>318 #include <iostream>
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -18,15 +18,15 @@
18## 参数说明18## 参数说明
19 19 
20- gradOutput(aclTensor*,计算输入):Device侧的aclTensor,[数据格式](../../docs/zh/context/data_format.md)支持ND。shape需要与self满足[broadcast关系](../../docs/zh/context/broadcast_relationship.md)。支持[非连续的Tensor](../../docs/zh/context/non_contiguous_tensor.md)。20- gradOutput(aclTensor*,计算输入):Device侧的aclTensor,[数据格式](../../docs/zh/context/data_format.md)支持ND。shape需要与self满足[broadcast关系](../../docs/zh/context/broadcast_relationship.md)。支持[非连续的Tensor](../../docs/zh/context/non_contiguous_tensor.md)。
21- - <term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持FLOAT、FLOAT16、BFLOAT16。21+ - <term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持FLOAT、FLOAT16、BFLOAT16。
22- self(aclTensor*,计算输入):Device侧的aclTensor。[数据格式](../../docs/zh/context/data_format.md)支持ND。支持[非连续的Tensor](../../docs/zh/context/non_contiguous_tensor.md)。22- self(aclTensor*,计算输入):Device侧的aclTensor。[数据格式](../../docs/zh/context/data_format.md)支持ND。支持[非连续的Tensor](../../docs/zh/context/non_contiguous_tensor.md)。
23- - <term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持FLOAT、FLOAT16、BFLOAT16。23+ - <term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持FLOAT、FLOAT16、BFLOAT16。
24- target(aclTensor*,计算输入):Device侧的aclTensor,[数据格式](../../docs/zh/context/data_format.md)支持ND。shape需要与self满足[broadcast关系](../../docs/zh/context/broadcast_relationship.md)。支持[非连续的Tensor](../../docs/zh/context/non_contiguous_tensor.md)。24- target(aclTensor*,计算输入):Device侧的aclTensor,[数据格式](../../docs/zh/context/data_format.md)支持ND。shape需要与self满足[broadcast关系](../../docs/zh/context/broadcast_relationship.md)。支持[非连续的Tensor](../../docs/zh/context/non_contiguous_tensor.md)。
25- - <term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持FLOAT、FLOAT16、BFLOAT16。25+ - <term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持FLOAT、FLOAT16、BFLOAT16。
26- reduction(int64_t,计算输入):Host侧的int64_t,[数据格式](../../docs/zh/context/data_format.md)支持ND。指定要应用到输出的缩减。支持0(‘none’)| 1(‘mean’)| 2(‘sum’)|3(‘batchmean’)。‘none’表示不应用减少,‘mean’表示输出的总和将除以输出中的元素数,‘sum’表示输出将被求和,‘batchmean’表示输出的总和将除以batch的个数。26- reduction(int64_t,计算输入):Host侧的int64_t,[数据格式](../../docs/zh/context/data_format.md)支持ND。指定要应用到输出的缩减。支持0(‘none’)| 1(‘mean’)| 2(‘sum’)|3(‘batchmean’)。‘none’表示不应用减少,‘mean’表示输出的总和将除以输出中的元素数,‘sum’表示输出将被求和,‘batchmean’表示输出的总和将除以batch的个数。
27- logTarget(bool,计算输入):Host侧的BOOL类型,是否对target进行log空间转换。27- logTarget(bool,计算输入):Host侧的BOOL类型,是否对target进行log空间转换。
28- out(aclTensor*,计算输出):Device侧的aclTensor。支持[非连续的Tensor](../../docs/zh/context/non_contiguous_tensor.md)。28- out(aclTensor*,计算输出):Device侧的aclTensor。支持[非连续的Tensor](../../docs/zh/context/non_contiguous_tensor.md)。
29- - <term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持FLOAT、FLOAT16、BFLOAT16。29+ - <term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持FLOAT、FLOAT16、BFLOAT16。
30 30 
31## 约束说明31## 约束说明
32 32 
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -18,14 +18,14 @@
18## 参数说明18## 参数说明
19 19 
20- gradOutput (aclTensor*,计算输入):梯度反向输入,公式中的输入`gradOutput`,Device侧的aclTensor。数据类型与self、target的数据类型满足数据类型推导规则(参见[互推导关系](../../docs/zh/context/deduction_relationship.md))。shape需要与self、target满足[broadcast关系](../../docs/zh/context/broadcast_relationship.md),且小于9维。支持[非连续的Tensor](../../docs/zh/context/non_contiguous_tensor.md),[数据格式](../../docs/zh/context/data_format.md)支持ND。20- gradOutput (aclTensor*,计算输入):梯度反向输入,公式中的输入`gradOutput`,Device侧的aclTensor。数据类型与self、target的数据类型满足数据类型推导规则(参见[互推导关系](../../docs/zh/context/deduction_relationship.md))。shape需要与self、target满足[broadcast关系](../../docs/zh/context/broadcast_relationship.md),且小于9维。支持[非连续的Tensor](../../docs/zh/context/non_contiguous_tensor.md),[数据格式](../../docs/zh/context/data_format.md)支持ND。
21- - <term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持BFLOAT16、FLOAT16、FLOAT32。21+ - <term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持BFLOAT16、FLOAT16、FLOAT32。
22- self (aclTensor*,计算输入):公式中的输入`self`,Device侧的aclTensor。数据类型与gradOutput、target满足数据类型推导规则(参见[互推导关系](../../docs/zh/context/deduction_relationship.md))。shape需要与gradOutput、target满足[broadcast关系](../../docs/zh/context/broadcast_relationship.md),且小于9维。支持[非连续的Tensor](../../docs/zh/context/non_contiguous_tensor.md),[数据格式](../../docs/zh/context/data_format.md)支持ND。22- self (aclTensor*,计算输入):公式中的输入`self`,Device侧的aclTensor。数据类型与gradOutput、target满足数据类型推导规则(参见[互推导关系](../../docs/zh/context/deduction_relationship.md))。shape需要与gradOutput、target满足[broadcast关系](../../docs/zh/context/broadcast_relationship.md),且小于9维。支持[非连续的Tensor](../../docs/zh/context/non_contiguous_tensor.md),[数据格式](../../docs/zh/context/data_format.md)支持ND。
23- - <term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持BFLOAT16、FLOAT16、FLOAT32。23+ - <term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持BFLOAT16、FLOAT16、FLOAT32。
24- target (aclTensor*,计算输入):公式中的输入`target`,Device侧的aclTensor。数据类型要与self、gradOutput满足数据类型推导规则(参见[互推导关系](../../docs/zh/context/deduction_relationship.md))。shape需要与gradOutput、self满足[broadcast关系](../../docs/zh/context/broadcast_relationship.md),且小于9维。支持[非连续的Tensor](../../docs/zh/context/non_contiguous_tensor.md),[数据格式](../../docs/zh/context/data_format.md)支持ND。24- target (aclTensor*,计算输入):公式中的输入`target`,Device侧的aclTensor。数据类型要与self、gradOutput满足数据类型推导规则(参见[互推导关系](../../docs/zh/context/deduction_relationship.md))。shape需要与gradOutput、self满足[broadcast关系](../../docs/zh/context/broadcast_relationship.md),且小于9维。支持[非连续的Tensor](../../docs/zh/context/non_contiguous_tensor.md),[数据格式](../../docs/zh/context/data_format.md)支持ND。
25- - <term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持BFLOAT16、FLOAT16、FLOAT32。25+ - <term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持BFLOAT16、FLOAT16、FLOAT32。
26- reduction (int64_t,计算输入):用于指定损失函数的计算方式,Host侧的整型。取值支持0('none') | 1('mean') | 2('sum')。'none'表示不应用减少,'mean'表示输出的总和将除以输出中的元素数,'sum'表示输出将被求和。26- reduction (int64_t,计算输入):用于指定损失函数的计算方式,Host侧的整型。取值支持0('none') | 1('mean') | 2('sum')。'none'表示不应用减少,'mean'表示输出的总和将除以输出中的元素数,'sum'表示输出将被求和。
27- gradInput (aclTensor*,计算输出):公式中的输出`gradInput`,Device侧的aclTensor。数据类型要与self满足数据类型推导规则(参见[互推导关系](../../docs/zh/context/deduction_relationship.md))。shape需要与target、self、gradOutput满足broadcast关系[broadcast关系](../../docs/zh/context/broadcast_relationship.md),且小于9维。支持[非连续的Tensor](../../docs/zh/context/non_contiguous_tensor.md),[数据格式](../../docs/zh/context/data_format.md)支持ND。27- gradInput (aclTensor*,计算输出):公式中的输出`gradInput`,Device侧的aclTensor。数据类型要与self满足数据类型推导规则(参见[互推导关系](../../docs/zh/context/deduction_relationship.md))。shape需要与target、self、gradOutput满足broadcast关系[broadcast关系](../../docs/zh/context/broadcast_relationship.md),且小于9维。支持[非连续的Tensor](../../docs/zh/context/non_contiguous_tensor.md),[数据格式](../../docs/zh/context/data_format.md)支持ND。
28- - <term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持BFLOAT16、FLOAT16、FLOAT32。28+ - <term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持BFLOAT16、FLOAT16、FLOAT32。
29 29 
30## 约束说明30## 约束说明
31 31 
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持
@@ -26,9 +26,9 @@
26## 参数说明26## 参数说明
27 27 
28- x (aclTensor*,计算输入):公式中的输入`x`,Device侧的aclTensor。支持[非连续的Tensor](../../docs/zh/context/non_contiguous_tensor.md),[数据格式](../../docs/zh/context/data_format.md)支持ND。28- x (aclTensor*,计算输入):公式中的输入`x`,Device侧的aclTensor。支持[非连续的Tensor](../../docs/zh/context/non_contiguous_tensor.md),[数据格式](../../docs/zh/context/data_format.md)支持ND。
29- - <term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持BFLOAT16、FLOAT16、FLOAT。29+ - <term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持BFLOAT16、FLOAT16、FLOAT。
30- y (aclTensor*,计算输出):公式中的输出`y`,Device侧的aclTensor。数据类型与x一致。shape为0维(标量)。支持[非连续的Tensor](../../docs/zh/context/non_contiguous_tensor.md),[数据格式](../../docs/zh/context/data_format.md)支持ND。30- y (aclTensor*,计算输出):公式中的输出`y`,Device侧的aclTensor。数据类型与x一致。shape为0维(标量)。支持[非连续的Tensor](../../docs/zh/context/non_contiguous_tensor.md),[数据格式](../../docs/zh/context/data_format.md)支持ND。
31- - <term>Atlas A2系列产品</term>、<term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持BFLOAT16、FLOAT16、FLOAT。31+ - <term>Atlas A2系列产品</term>、<term>Atlas A3系列产品</term>、<term>Ascend 950PR&950DT系列产品</term>:数据类型支持BFLOAT16、FLOAT16、FLOAT。
32 32 
33## 约束说明33## 约束说明
34 34 
@@ -8,7 +8,7 @@
8- <term>Ascend 950PR&950DT系列产品</term>:支持8- <term>Ascend 950PR&950DT系列产品</term>:支持
9<!-- end id1 -->9<!-- end id1 -->
10<!-- npu="A3" id2 -->10<!-- npu="A3" id2 -->
11-- <term>Atlas A3 训练系列产品/Atlas A3 推理系列产品</term>:支持11+- <term>Atlas A3系列产品</term>:支持
12<!-- end id2 -->12<!-- end id2 -->
13<!-- npu="910b" id3 -->13<!-- npu="910b" id3 -->
14- <term>Atlas A2系列产品</term>:支持14- <term>Atlas A2系列产品</term>:支持