|
|
|
@@ -62,6 +62,7 @@ |
|
|
|
#include "transform/graph_ir/convert.h" |
|
|
|
#include "transform/graph_ir/df_graph_manager.h" |
|
|
|
#include "transform/graph_ir/op_adapter_map.h" |
|
|
|
#include "runtime/device/ascend/profiling/profiling_manager.h" |
|
|
|
#endif |
|
|
|
#ifdef ENABLE_DUMP_IR |
|
|
|
#include "debug/rdr/running_data_recorder.h" |
|
|
|
@@ -79,6 +80,10 @@ using mindspore::abstract::AbstractTensorPtr; |
|
|
|
using mindspore::abstract::AbstractTuple; |
|
|
|
using mindspore::abstract::AbstractTuplePtr; |
|
|
|
|
|
|
|
#if (ENABLE_GE || ENABLE_D) |
|
|
|
using mindspore::device::ascend::ProfilingManager; |
|
|
|
#endif |
|
|
|
|
|
|
|
const char IR_TYPE_ANF[] = "anf_ir"; |
|
|
|
const char IR_TYPE_ONNX[] = "onnx_ir"; |
|
|
|
const char IR_TYPE_MINDIR[] = "mind_ir"; |
|
|
|
@@ -1078,6 +1083,11 @@ void InitHccl() { |
|
|
|
(void)context::OpenTsd(ms_context); |
|
|
|
} |
|
|
|
#endif |
|
|
|
#if (ENABLE_GE || ENABLE_D) |
|
|
|
if (!ProfilingManager::GetInstance().IsProfiling()) { |
|
|
|
ProfilingManager::GetInstance().SetHcclEnabledBefProfilingEnabled(); |
|
|
|
} |
|
|
|
#endif |
|
|
|
} |
|
|
|
|
|
|
|
void FinalizeHccl() { |
|
|
|
|