已合并
修复TensorEqual整形数据边界用例bug（商分） #2931
Wangdongxu_mj007创建于 5月25日
修复TensorEqual整形数据边界用例bug（商分） #2931
已合并
共 1 个文件变更+2-1
| @@ -68,7 +68,8 @@ __aicore__ inline void TensorEqualKernel<T>::Init(GM_ADDR input_x, GM_ADDR input | |||
| 68 | return; | 68 | return; |
| 69 | } | 69 | } |
| 70 | 70 | ||
| 71 | - blockOffset_ = blockIdx_ * ((tilingData_.perCoreLoopTimes - 1) * tilingData_.ubFactor + tilingData_.perCoreTailFactor); | 71 | + blockOffset_ = blockIdx_ * (sizeof(T) / sizeof(InputType)) * |
| 72 | + ((tilingData_.perCoreLoopTimes - 1) * tilingData_.ubFactor + tilingData_.perCoreTailFactor); | ||
| 72 | bufferSize_ = tilingData_.ubFactor * sizeof(T); | 73 | bufferSize_ = tilingData_.ubFactor * sizeof(T); |
| 73 | 74 | ||
| 74 | inputXGm_.SetGlobalBuffer((__gm__ InputType *)(input_x) + blockOffset_); | 75 | inputXGm_.SetGlobalBuffer((__gm__ InputType *)(input_x) + blockOffset_); |