已合并
修复算子bug&&修改aclnnIndexSelect中的文档 #1562
sakuraqqz创建于 2月5日
修复算子bug&&修改aclnnIndexSelect中的文档 #1562
已合并
共 5 个文件变更+7-5
| @@ -141,7 +141,7 @@ aclnnStatus aclnnIndexSelect( | |||
| 141 | </tr> | 141 | </tr> |
| 142 | </tbody></table> | 142 | </tbody></table> |
| 143 | 143 | ||
| 144 | - - <term>Atlas 推理系列产品</term>、<term>Atlas 训练系列产品</term>：数据类型不支持BFLOAT。 | 144 | + - <term>Atlas 推理系列产品</term>、<term>Atlas 训练系列产品</term>：数据类型不支持BFLOAT16。 |
| 145 | 145 | ||
| 146 | - **返回值** | 146 | - **返回值** |
| 147 | 147 | ||
| @@ -226,6 +226,8 @@ aclnnStatus aclnnIndexSelect( | |||
| 226 | 226 | ||
| 227 | ## 约束说明 | 227 | ## 约束说明 |
| 228 | 228 | ||
| 229 | +- 当self的shape为[]时，index的shape只能为[1]。 | ||
| 230 | + | ||
| 229 | - 确定性计算： | 231 | - 确定性计算： |
| 230 | - aclnnIndexSelect默认确定性实现。 | 232 | - aclnnIndexSelect默认确定性实现。 |
| 231 | 233 | ||
| @@ -201,7 +201,7 @@ static aclnnStatus CheckParams(const aclTensor *self, int64_t dim, const aclTens | |||
| 201 | xsize *= self->GetViewShape().GetDim(i); | 201 | xsize *= self->GetViewShape().GetDim(i); |
| 202 | } | 202 | } |
| 203 | 203 | ||
| 204 | - auto indexSize = ge::GetSizeByDataType(index->GetDataType()); | 204 | + auto indexSize = 1; |
| 205 | auto indexDimNum = index->GetViewShape().GetDimNum(); | 205 | auto indexDimNum = index->GetViewShape().GetDimNum(); |
| 206 | for (int i = 0; i < indexDimNum ; i++) { | 206 | for (int i = 0; i < indexDimNum ; i++) { |
| 207 | indexSize *= index->GetViewShape().GetDim(i); | 207 | indexSize *= index->GetViewShape().GetDim(i); |
| @@ -9,5 +9,5 @@ | |||
| 9 | ; 4)如果default配置和平台配置都有：对应平台的使用平台的配置，非对应的平台的以default值配置。 | 9 | ; 4)如果default配置和平台配置都有：对应平台的使用平台的配置，非对应的平台的以default值配置。 |
| 10 | ; 5)对于自定义simplified key的情况，需要在binary_simplified_key_mode.ini 文件中显式配置为None，不传入 --simplified_key_mode 选项，由opc工具和FE框架自行判断使用何种模式 | 10 | ; 5)对于自定义simplified key的情况，需要在binary_simplified_key_mode.ini 文件中显式配置为None，不传入 --simplified_key_mode 选项，由opc工具和FE框架自行判断使用何种模式 |
| 11 | ; 6)是否是AscendC算子，由 ops/build-in/tbe/op_info_cfg/parser/ascendc_config.json 中配置的算子名字和对于的平台决定 | 11 | ; 6)是否是AscendC算子，由 ops/build-in/tbe/op_info_cfg/parser/ascendc_config.json 中配置的算子名字和对于的平台决定 |
| 12 | -[FusedCrossEntropyLossWithMaxSum] | 12 | +[IndexFill] |
| 13 | default=0 | 13 | default=0 |
| @@ -81,7 +81,7 @@ | |||
| 81 | "index": 1, | 81 | "index": 1, |
| 82 | "dtype": "float32", | 82 | "dtype": "float32", |
| 83 | "format": "ND", | 83 | "format": "ND", |
| 84 | - "paramType": "required", | 84 | + "paramType": "optional", |
| 85 | "shape": [ | 85 | "shape": [ |
| 86 | -2 | 86 | -2 |
| 87 | ] | 87 | ] |