已合并
fix GetVecTensor size get #6448
zc1110创建于 9月28日
fix GetVecTensor size get #6448
已合并
共 1 个文件变更+2-2
| @@ -2029,12 +2029,12 @@ private: | |||
| 2029 | TBuffAddr tbufOutTmp; | 2029 | TBuffAddr tbufOutTmp; |
| 2030 | tbufOutTmp.logicPos = (uint8_t)(TPosition::VECCALC); | 2030 | tbufOutTmp.logicPos = (uint8_t)(TPosition::VECCALC); |
| 2031 | tbufOutTmp.bufferAddr = addr; | 2031 | tbufOutTmp.bufferAddr = addr; |
| 2032 | -#if ASCENDC_CPU_DEBUG | 2032 | + if constexpr (IsTypeOneOfV<T, fp4x2_e1m2_t, fp4x2_e2m1_t>) { |
| 2033 | - if (IsTypeOneOfV<T, fp4x2_e1m2_t, fp4x2_e2m1_t>) { | ||
| 2034 | tbufOutTmp.dataLen = size / AscendC::Impl::FP4_TWO; | 2033 | tbufOutTmp.dataLen = size / AscendC::Impl::FP4_TWO; |
| 2035 | } else { | 2034 | } else { |
| 2036 | tbufOutTmp.dataLen = size * sizeof(T); | 2035 | tbufOutTmp.dataLen = size * sizeof(T); |
| 2037 | } | 2036 | } |
| 2037 | +#if ASCENDC_CPU_DEBUG | ||
| 2038 | tbufOutTmp.absAddr = reinterpret_cast<uint8_t*>(addr); | 2038 | tbufOutTmp.absAddr = reinterpret_cast<uint8_t*>(addr); |
| 2039 | #endif | 2039 | #endif |
| 2040 | cLocal.SetAddr(tbufOutTmp); | 2040 | cLocal.SetAddr(tbufOutTmp); |