已合并
argmax算子冗余代码修复 #3203
huangyuqian创建于 6月9日
argmax算子冗余代码修复 #3203
已合并
从已删除 :master合入到cann/ops-mathmaster
共 2 个文件变更+2-2
| @@ -111,7 +111,7 @@ aclnnStatus aclnnMaxDim( | |||
| 111 | <td>输出</td> | 111 | <td>输出</td> |
| 112 | <td>最大值索引输出张量。</td> | 112 | <td>最大值索引输出张量。</td> |
| 113 | <td>-</td> | 113 | <td>-</td> |
| 114 | - <td>BOOL、FLOAT32、FLOAT16、INT8、INT16、UINT16、UINT8、INT32、INT64、UINT32、UINT64、DOUBLE、COMPLEX64、COMPLEX128、BFLOAT16</td> | 114 | + <td>INT32、INT64</td> |
| 115 | <td>ND</td> | 115 | <td>ND</td> |
| 116 | <td>-</td> | 116 | <td>-</td> |
| 117 | <td>√</td> | 117 | <td>√</td> |
| @@ -372,7 +372,7 @@ __aicore__ inline void ArgMaxWithValueBase<T1, T2, T3, isMin>::ArgMaxRaInt64(__l | |||
| 372 | for (uint16_t i = 0; i < loopA; i++) { | 372 | for (uint16_t i = 0; i < loopA; i++) { |
| 373 | auto tmpPtr = (__local_mem__ T4*&)src + alignA; | 373 | auto tmpPtr = (__local_mem__ T4*&)src + alignA; |
| 374 | mask = AscendC::MicroAPI::UpdateMask<uint32_t>(processA); | 374 | mask = AscendC::MicroAPI::UpdateMask<uint32_t>(processA); |
| 375 | - AscendC::MicroAPI::DataCopyUnAlignPre(uSrc, (__local_mem__ T4*&)src);AscendC::MicroAPI::DataCopyUnAlignPre(uSrc, (__local_mem__ T4*&)src); | 375 | + AscendC::MicroAPI::DataCopyUnAlignPre(uSrc, (__local_mem__ T4*&)src); |
| 376 | AscendC::MicroAPI::DataCopyUnAlign<T4, AscendC::MicroAPI::PostLiteral::POST_MODE_UPDATE>(vregValue0, uSrc, | 376 | AscendC::MicroAPI::DataCopyUnAlign<T4, AscendC::MicroAPI::PostLiteral::POST_MODE_UPDATE>(vregValue0, uSrc, |
| 377 | (__local_mem__ T4*&)src, offsetA); | 377 | (__local_mem__ T4*&)src, offsetA); |
| 378 | AscendC::MicroAPI::Duplicate(vregIndices0, VALUE_ZERO, fullMask); | 378 | AscendC::MicroAPI::Duplicate(vregIndices0, VALUE_ZERO, fullMask); |