|
|
|
@@ -20,6 +20,7 @@ |
|
|
|
|
|
|
|
namespace ge { |
|
|
|
namespace { |
|
|
|
#ifdef ENABLE_OPEN_SRC |
|
|
|
const std::vector<std::string> kBasicBuilderLibs = { |
|
|
|
"libge_local_opskernel_builder.so", |
|
|
|
"libhost_cpu_opskernel_builder.so", |
|
|
|
@@ -27,7 +28,15 @@ const std::vector<std::string> kBasicBuilderLibs = { |
|
|
|
"libaicpu_ascend_builder.so", |
|
|
|
"libaicpu_tf_builder.so" |
|
|
|
}; |
|
|
|
|
|
|
|
#else |
|
|
|
const std::vector<std::string> kBasicBuilderLibs = { |
|
|
|
"libge_local_opskernel_builder.so", |
|
|
|
"libhost_cpu_opskernel_builder.so", |
|
|
|
"librts_engine.so", |
|
|
|
"libaicpu_ascend_engine.so", |
|
|
|
"libaicpu_tf_engine.so" |
|
|
|
}; |
|
|
|
#endif |
|
|
|
const std::vector<std::string> kHcclBuilderLibs = { |
|
|
|
"libhcom_opskernel_builder.so", |
|
|
|
"libhvd_opskernel_builder.so", |
|
|
|
|