核函数中如何使用临时变量?
收藏回复举报
核函数中如何使用临时变量?
t('forum.solved') 已解决
新人帖
发表于2023-09-13 16:48:06
0 查看
 

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?

我要发帖子