p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!12236 fix error of AICPU so name 

From: @simson_wu
Reviewed-by: @chujinjin,@zhoufeng54
Signed-off-by: @zhoufeng54
This commit is contained in:
mindspore-ci-bot 2021-02-19 17:00:25 +08:00 committed by Gitee
parent 26d944ab15 d6b7ee64b7
commit 7e34acc21c
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/ccsrc/backend/kernel_compiler/aicpu/aicpu_kernel_mod.cc
View File

@ -79,7 +79,9 @@ void AicpuOpKernelMod::CreateCpuKernelInfo(const std::vector<AddressPtr> &inputs
node_so_ = AICPU_OPS_SO_NAME;
node_name_ = kCustRunApi;
} else {
node_so_ = AICPU_OPS_SO_NAME;
if (node_so_ != CUST_AICPU_OPS_SO_NAME) {
node_so_ = AICPU_OPS_SO_NAME;
}
}
// InputOutputAddr
vector<void *> io_addrs;
@ -171,7 +173,9 @@ std::vector<TaskInfoPtr> AicpuOpKernelMod::GenTask(const std::vector<AddressPtr>
node_so_ = AICPU_OPS_SO_NAME;
node_name_ = kCustRunApi;
} else {
node_so_ = AICPU_OPS_SO_NAME;
if (node_so_ != CUST_AICPU_OPS_SO_NAME) {
node_so_ = AICPU_OPS_SO_NAME;
}
}
std::vector<void *> input_data_addrs;
(void)std::transform(std::begin(inputs), std::end(inputs), std::back_inserter(input_data_addrs),