Merged
修改产品名 #11348
yanglu-1Add on Sep 29
修改产品名 #11348
Merged
Merged fromyanglu-1/ops-nn_9495masterintocann/ops-nnmaster
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 | * strides | 47 | * 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 | * pads | 51 | * 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 | * dilations | 55 | * 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_padding | 59 | * 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 | * strides | 67 | * 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 | * pads | 71 | * 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 | * dilations | 75 | * 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 | * strides | 84 | * 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 | * pads | 88 | * 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 | * dilations | 92 | * 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_padding | 96 | * 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系列产品:取值为ascend910b | 62 | - Atlas A2系列产品:取值为ascend910b |
| 63 | -- Atlas A3 训练系列产品/Atlas A3 推理系列产品:取值为ascend910_93 | 63 | +- Atlas A3系列产品:取值为ascend910_93 |
| 64 | - 950系列产品:取值为ascend950 | 64 | - 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系列产品:取值为ascend910b | 157 | - Atlas A2系列产品:取值为ascend910b |
| 158 | - - Atlas A3 训练系列产品/Atlas A3 推理系列产品:取值为ascend910_93 | 158 | + - Atlas A3系列产品:取值为ascend910_93 |
| 159 | - 950系列产品:取值为ascend950 | 159 | - 950系列产品:取值为ascend950 |
| 160 | 160 | ||
| 161 | 2. **重新安装**: | 161 | 2. **重新安装**: |
| @@ -55,7 +55,7 @@ | |||
| 55 | # bash build.sh --pkg --experimental --soc=ascend910b --ops=transpose_batch_mat_mul -j16 | 55 | # 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 | ## 本地编译运行 UT | 44 | ## 本地编译运行 UT |
| 45 | 45 | ||
| 46 | -环境准备请参考项目环境部署文档。Atlas A2 系列产品使用 `ascend910b`;其他可选 SoC 为 `ascend910_93` 和 `ascend950`。 | 46 | +环境准备请参考项目环境部署文档。Atlas A2系列产品使用 `ascend910b`;其他可选 SoC 为 `ascend910_93` 和 `ascend950`。 |
| 47 | 47 | ||
| 48 | ```bash | 48 | ```bash |
| 49 | # Host UT:注册、shape/type 推导和 tiling | 49 | # 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 \% coreNum | 96 | 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>| × | |
Mloss/cross_entropy_sum_exp_and_index_logit/docs/torchapi_cross_entropy_sum_exp_and_index_logit.md+1-1
| @@ -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 | ```Cpp | 317 | ```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>:支持 |