Browse Source

!216 Feature:add new hcom ops kernel (by hccl)

Merge pull request !216 from zhaoxinxin/development
tags/v1.1.0
mindspore-ci-bot Gitee 4 years ago
parent
commit
1d508ab94b
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      ge/opskernel_manager/ops_kernel_builder_manager.cc

+ 2
- 1
ge/opskernel_manager/ops_kernel_builder_manager.cc View File

@@ -30,7 +30,8 @@ const std::vector<std::string> kBasicBuilderLibs = {

const std::vector<std::string> kHcclBuilderLibs = {
"libhcom_opskernel_builder.so",
"libhvd_opskernel_builder.so"
"libhvd_opskernel_builder.so",
"libhcom_gradtune_opskernel_builder.so"
};
} // namespace
OpsKernelBuilderManager::~OpsKernelBuilderManager() {


Loading…
Cancel
Save