simson
4 years ago
1 changed files with 6 additions and 2 deletions
Unified View
Diff Options
+ 6
- 2
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_so_ = AICPU_OPS_SO_NAME; | ||||
| node_name_ = kCustRunApi; | node_name_ = kCustRunApi; | ||||
| } else { | } else { | ||||
| node_so_ = AICPU_OPS_SO_NAME; | |||||
| if (node_so_ != CUST_AICPU_OPS_SO_NAME) { | |||||
| node_so_ = AICPU_OPS_SO_NAME; | |||||
| } | |||||
| } | } | ||||
| // InputOutputAddr | // InputOutputAddr | ||||
| vector<void *> io_addrs; | vector<void *> io_addrs; | ||||
| @@ -171,7 +173,9 @@ std::vector<TaskInfoPtr> AicpuOpKernelMod::GenTask(const std::vector<AddressPtr> | |||||
| node_so_ = AICPU_OPS_SO_NAME; | node_so_ = AICPU_OPS_SO_NAME; | ||||
| node_name_ = kCustRunApi; | node_name_ = kCustRunApi; | ||||
| } else { | } 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; | std::vector<void *> input_data_addrs; | ||||
| (void)std::transform(std::begin(inputs), std::end(inputs), std::back_inserter(input_data_addrs), | (void)std::transform(std::begin(inputs), std::end(inputs), std::back_inserter(input_data_addrs), | ||||