root
/
MNN
mirror of https://github.com/alibaba/MNN.git
1
0
Fork 0
Code Issues Actions 6 Packages Projects Releases Wiki Activity

sunset ddk v2 interface

This commit is contained in:
liuchao19891114 2024-09-24 14:27:09 +08:00
parent aed41a82de
commit 9e64e885a9
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
source/backend/hiai/backend/NPUBackend.cpp
View File

@ -211,7 +211,7 @@ namespace MNN {
return false; return false;
} }
uint32_t write_size = (uint32_t)fwrite(om_model_buff.data, 1 , om_model_buff.length, fp); uint32_t write_size = (uint32_t)fwrite(om_model_buff.data, 1, om_model_buff.length, fp);
if (write_size != om_model_buff.length) { if (write_size != om_model_buff.length) {
fclose(fp); fclose(fp);
printf("write om file failed !!!"); printf("write om file failed !!!");
@ -434,7 +434,6 @@ namespace MNN {
auto index = mInputMap.find((unsigned long)(const_cast<Tensor*>(dstTensor))); auto index = mInputMap.find((unsigned long)(const_cast<Tensor*>(dstTensor)));
MNN_ASSERT(index != mInputMap.end()); MNN_ASSERT(index != mInputMap.end());
shared_ptr<hiai::AiTensor> input = mInputTensors[index->second]; shared_ptr<hiai::AiTensor> input = mInputTensors[index->second];
memcpy(input->GetBuffer(), srcTensor->host<void>(), (size_t)input->GetSize()); memcpy(input->GetBuffer(), srcTensor->host<void>(), (size_t)input->GetSize());
} else if(isOutputCopy){ } else if(isOutputCopy){
int index; int index;