华为计算微信公众号
昇腾AI开发者公众号
华为计算微博
华为计算今日头条
在operator/MatMulLeakyReluCustomSample/FrameworkLaunch/MatmulLeakyReluCustom/op_kernel/matmul_leakyrelu_custom.cpp的MatmulLeakyKernel::Process中
if constexpr (setTmpSpace) { TBuf<> tmpMMFormatUb; LocalTensor<uint8_t> mmformatUb; pipe->InitBuffer(tmpMMFormatUb, tiling.baseM * tiling.baseN * sizeof(cType)); mmformatUb = tmpMMFormatUb.Get<uint8_t>(tiling.baseM * tiling.baseN * sizeof(cType)); matmulObj.SetLocalWorkspace(mmformatUb); }
代码有什么作用?删掉有影响吗?
我要发帖子
在operator/MatMulLeakyReluCustomSample/FrameworkLaunch/MatmulLeakyReluCustom/op_kernel/matmul_leakyrelu_custom.cpp的MatmulLeakyKernel::Process中
if constexpr (setTmpSpace) { TBuf<> tmpMMFormatUb; LocalTensor<uint8_t> mmformatUb; pipe->InitBuffer(tmpMMFormatUb, tiling.baseM * tiling.baseN * sizeof(cType)); mmformatUb = tmpMMFormatUb.Get<uint8_t>(tiling.baseM * tiling.baseN * sizeof(cType)); matmulObj.SetLocalWorkspace(mmformatUb); }代码有什么作用?删掉有影响吗?