|
|
|
@@ -65,6 +65,7 @@ void RegisterAiCoreEngine() { |
|
|
|
DNNEnginePtr aicore_engine_ptr = MakeShared<AICoreDNNEngine>(attr_aicore); |
|
|
|
if (aicore_engine_ptr == nullptr) { |
|
|
|
GELOGE(ge::FAILED, "[Register][AiCoreEngine] failed, as malloc shared_ptr failed."); |
|
|
|
REPORT_INNER_ERROR("E19999", "RegisterAiCoreEngine failed for new DNNEnginePtr failed."); |
|
|
|
return; |
|
|
|
} |
|
|
|
if (EngineManager::RegisterEngine(ai_core, aicore_engine_ptr) != SUCCESS) { |
|
|
|
@@ -81,6 +82,7 @@ void RegisterVectorEngine() { |
|
|
|
DNNEnginePtr vectorcore_engine_ptr = MakeShared<VectorCoreDNNEngine>(attr_vector_core); |
|
|
|
if (vectorcore_engine_ptr == nullptr) { |
|
|
|
GELOGE(ge::FAILED, "[Register][VectorEngine] failed, as malloc shared_ptr failed."); |
|
|
|
REPORT_INNER_ERROR("E19999", "RegisterVectorEngine failed for new DNNEnginePtr failed."); |
|
|
|
return; |
|
|
|
} |
|
|
|
if (EngineManager::RegisterEngine(vector_core, vectorcore_engine_ptr) != SUCCESS) { |
|
|
|
@@ -96,6 +98,7 @@ void RegisterAiCpuEngine() { |
|
|
|
DNNEnginePtr vm_engine_ptr = MakeShared<AICpuDNNEngine>(attr_aicpu); |
|
|
|
if (vm_engine_ptr == nullptr) { |
|
|
|
GELOGE(ge::FAILED, "[Register][AiCpuEngine] failed, as malloc shared_ptr failed."); |
|
|
|
REPORT_INNER_ERROR("E19999", "RegisterAiCpuEngine failed for new DNNEnginePtr failed."); |
|
|
|
return; |
|
|
|
} |
|
|
|
if (EngineManager::RegisterEngine(vm_aicpu, vm_engine_ptr) != SUCCESS) { |
|
|
|
@@ -112,6 +115,7 @@ void RegisterAiCpuTFEngine() { |
|
|
|
DNNEnginePtr vm_engine_ptr = MakeShared<AICpuTFDNNEngine>(attr_aicpu_tf); |
|
|
|
if (vm_engine_ptr == nullptr) { |
|
|
|
GELOGE(ge::FAILED, "[Register][AiCpuTFEngine] failed, as malloc shared_ptr failed."); |
|
|
|
REPORT_INNER_ERROR("E19999", "RegisterAiCpuTFEngine failed for new DNNEnginePtr failed."); |
|
|
|
return; |
|
|
|
} |
|
|
|
if (EngineManager::RegisterEngine(vm_aicpu_tf, vm_engine_ptr) != SUCCESS) { |
|
|
|
@@ -128,6 +132,7 @@ void RegisterGeLocalEngine() { |
|
|
|
DNNEnginePtr ge_local_engine = MakeShared<GeLocalDNNEngine>(attr_ge_local); |
|
|
|
if (ge_local_engine == nullptr) { |
|
|
|
GELOGE(ge::FAILED, "[Register][GeLocalEngine] failed, as malloc shared_ptr failed."); |
|
|
|
REPORT_INNER_ERROR("E19999", "RegisterGeLocalEngine failed for new DNNEnginePtr failed."); |
|
|
|
return; |
|
|
|
} |
|
|
|
if (EngineManager::RegisterEngine(vm_ge_local, ge_local_engine) != SUCCESS) { |
|
|
|
@@ -145,6 +150,7 @@ void RegisterHostCpuEngine() { |
|
|
|
DNNEnginePtr host_cpu_engine = MakeShared<HostCpuDNNEngine>(attr_host_cpu); |
|
|
|
if (host_cpu_engine == nullptr) { |
|
|
|
GELOGE(ge::FAILED, "[Register][HostCpuEngine] failed, as malloc shared_ptr failed."); |
|
|
|
REPORT_INNER_ERROR("E19999", "RegisterHostCpuEngine failed for new DNNEnginePtr failed."); |
|
|
|
return; |
|
|
|
} |
|
|
|
if (EngineManager::RegisterEngine(vm_host_cpu, host_cpu_engine) != SUCCESS) { |
|
|
|
@@ -160,6 +166,7 @@ void RegisterRtsEngine() { |
|
|
|
DNNEnginePtr rts_engine = MakeShared<RtsDNNEngine>(attr_rts); |
|
|
|
if (rts_engine == nullptr) { |
|
|
|
GELOGE(ge::FAILED, "[Register][RtsEngine] failed, as malloc shared_ptr failed."); |
|
|
|
REPORT_INNER_ERROR("E19999", "RegisterRtsEngine failed for new DNNEnginePtr failed."); |
|
|
|
return; |
|
|
|
} |
|
|
|
if (EngineManager::RegisterEngine(vm_rts, rts_engine) != SUCCESS) { |
|
|
|
@@ -175,6 +182,7 @@ void RegisterHcclEngine() { |
|
|
|
DNNEnginePtr hccl_engine = MakeShared<HcclDNNEngine>(attr_hccl); |
|
|
|
if (hccl_engine == nullptr) { |
|
|
|
GELOGE(ge::FAILED, "[Register][HcclEngine] failed, as malloc shared_ptr failed."); |
|
|
|
REPORT_INNER_ERROR("E19999", "RegisterHcclEngine failed for new DNNEnginePtr failed."); |
|
|
|
return; |
|
|
|
} |
|
|
|
if (EngineManager::RegisterEngine(dnn_hccl, hccl_engine) != SUCCESS) { |
|
|
|
|