TBuf<TPosition::VECIN> calcBuf;
pipe.InitBuffer(calcBuf, 1024);
LocalTensor<half> constValue = calcBuf.Get<half>(TILE_LENGTH);
for (int32_t i = 0; i < TILE_LENGTH; i++) {
constValue.SetValue(i, 1);
}
LocalTensor<half> tempValue = calcBuf.Get<half>(TILE_LENGTH);
//Add(tempValue, xLocal, constValue, TILE_LENGTH);
Add(yLocal, xLocal, constValue, TILE_LENGTH);
想在代码中用一下临时变量,发现注释掉的那行代码打开以后就不符合预期了,这边是哪个地方用错了?
还有constValue的构造显得有点粗暴,有没有什么办法可以初始化一个全为1的tensor?
TBuf<TPosition::VECIN> calcBuf;
pipe.InitBuffer(calcBuf, 1024);
LocalTensor<half> constValue = calcBuf.Get<half>(TILE_LENGTH);
for (int32_t i = 0; i < TILE_LENGTH; i++) {
constValue.SetValue(i, 1);
}
LocalTensor<half> tempValue = calcBuf.Get<half>(TILE_LENGTH);
//Add(tempValue, xLocal, constValue, TILE_LENGTH);
Add(yLocal, xLocal, constValue, TILE_LENGTH);
想在代码中用一下临时变量,发现注释掉的那行代码打开以后就不符合预期了,这边是哪个地方用错了?
还有constValue的构造显得有点粗暴,有没有什么办法可以初始化一个全为1的tensor?