@@ -39,7 +39,7 @@ namespace { | |||
} \ | |||
ge_tensor = MakeShared<GeTensor>(out_desc); \ | |||
GE_CHECK_NOTNULL(ge_tensor); \ | |||
GELOGI("node:%s allocate output %zu success, size=%lld", op_desc->GetName().c_str(), i, data_num * sizeof(TYPE));\ | |||
GELOGD("node:%s allocate output %zu success, size=%lld", op_desc->GetName().c_str(), i, data_num * sizeof(TYPE));\ | |||
if (ge_tensor->SetData(reinterpret_cast<uint8_t *>(buf.get()), data_num * sizeof(TYPE)) != GRAPH_SUCCESS) { \ | |||
GELOGE(MEMALLOC_FAILED, "Set data for output %zu of node %s failed.", i, op_desc->GetName().c_str()); \ | |||
return MEMALLOC_FAILED; \ | |||
@@ -50,8 +50,7 @@ namespace { | |||
} else { \ | |||
ge_tensor = outputs[i]; \ | |||
GE_CHECK_NOTNULL(ge_tensor); \ | |||
GELOGI("node:%s existed output %zu, addr=%p, size=%lld", op_desc->GetName().c_str(), i, \ | |||
reinterpret_cast<const uint8_t *>(ge_tensor->GetData().data()), ge_tensor->GetData().size()); \ | |||
GELOGD("node:%s existed output %zu", op_desc->GetName().c_str(), i); \ | |||
} \ | |||
auto tensor = TensorAdapter::AsTensor(*ge_tensor); \ | |||
auto tensor_name = op_desc->GetOutputNameByIndex(i); \ | |||
@@ -23,25 +23,15 @@ | |||
#include <sstream> | |||
#include <string> | |||
#include <thread> | |||
#include <utility> | |||
#include "common/ge/ge_util.h" | |||
#include "common/math/math_util.h" | |||
#include "common/thread_pool.h" | |||
#include "common/util.h" | |||
#include "external/graph/types.h" | |||
#include "framework/common/debug/ge_log.h" | |||
#include "framework/common/ge_inner_error_codes.h" | |||
#include "framework/common/ge_types.h" | |||
#include "analyzer/analyzer.h" | |||
#include "graph/common/ge_call_wrapper.h" | |||
#include "graph/common/local_context.h" | |||
#include "graph/common/transop_util.h" | |||
#include "graph/debug/ge_attr_define.h" | |||
#include "graph/ge_context.h" | |||
#include "graph/ge_global_options.h" | |||
#include "graph/ge_local_context.h" | |||
#include "graph/manager/graph_mem_allocator.h" | |||
#include "graph/manager/util/rt_context_util.h" | |||
#include "graph/partition/dynamic_shape_partition.h" | |||
#include "graph/passes/enter_pass.h" | |||
@@ -61,8 +51,6 @@ | |||
#include "graph/passes/dimension_adjust_pass.h" | |||
#include "graph/passes/dimension_compute_pass.h" | |||
#include "graph/passes/flow_ctrl_pass.h" | |||
#include "graph/passes/hccl_group_pass.h" | |||
#include "graph/passes/hccl_memcpy_pass.h" | |||
#include "graph/passes/identity_pass.h" | |||
#include "graph/passes/input_output_connection_identify_pass.h" | |||
#include "graph/passes/iterator_op_pass.h" | |||
@@ -76,7 +64,6 @@ | |||
#include "graph/passes/permute_pass.h" | |||
#include "graph/passes/prune_pass.h" | |||
#include "graph/passes/ref_identity_delete_op_pass.h" | |||
#include "graph/passes/replace_with_empty_const_pass.h" | |||
#include "graph/passes/reshape_recovery_pass.h" | |||
#include "graph/passes/reshape_remove_pass.h" | |||
#include "graph/passes/same_transdata_breadth_fusion_pass.h" | |||
@@ -86,13 +73,11 @@ | |||
#include "graph/passes/switch_logic_remove_pass.h" | |||
#include "graph/passes/switch_to_stream_switch_pass.h" | |||
#include "graph/passes/transop_breadth_fusion_pass.h" | |||
#include "graph/passes/transop_depth_fusion_pass.h" | |||
#include "graph/passes/transop_nearby_allreduce_fusion_pass.h" | |||
#include "graph/passes/transop_symmetry_elimination_pass.h" | |||
#include "graph/passes/transop_without_reshape_fusion_pass.h" | |||
#include "graph/passes/transpose_transdata_pass.h" | |||
#include "graph/passes/variable_op_pass.h" | |||
#include "graph/passes/variable_prepare_op_pass.h" | |||
#include "graph/passes/variable_ref_delete_op_pass.h" | |||
#include "graph/passes/variable_ref_useless_control_out_delete_pass.h" | |||
#include "graph/passes/end_of_sequence_add_control_pass.h" | |||
@@ -103,9 +88,6 @@ | |||
#include "graph/passes/memcpy_addr_async_pass.h" | |||
#include "graph/build/label_allocator.h" | |||
#include "graph/utils/tensor_adapter.h" | |||
#include "graph/utils/type_utils.h" | |||
#include "graph/graph_util.h" | |||
#include "graph/types.h" | |||
#include "inc/pass_manager.h" | |||
#include "init/gelib.h" | |||
#include "ir_build/atc_ir_common.h" | |||
@@ -16,10 +16,7 @@ | |||
#include "graph/manager/graph_mem_allocator.h" | |||
#include <set> | |||
#include <string> | |||
#include "framework/common/debug/ge_log.h" | |||
#include "graph/manager/graph_caching_allocator.h" | |||
#include "graph/manager/rdma_pool_allocator.h" | |||
@@ -17,13 +17,8 @@ | |||
#include "graph/passes/switch_to_stream_switch_pass.h" | |||
#include <stack> | |||
#include "common/ge/ge_util.h" | |||
#include "framework/common/debug/ge_log.h" | |||
#include "framework/common/debug/log.h" | |||
#include "framework/common/ge_inner_error_codes.h" | |||
#include "framework/common/types.h" | |||
#include "ge/ge_api_types.h" | |||
#include "graph/common/omg_util.h" | |||
#include "graph/debug/ge_attr_define.h" | |||
#include "graph/ge_context.h" | |||
#include "graph/utils/type_utils.h" | |||
@@ -125,12 +120,13 @@ void SwitchToStreamSwitchPass::MarkCycleDependence( | |||
if (visited.count(tmp_node) > 0) { | |||
continue; | |||
} | |||
GELOGD("MarkCycleDependence: tmp_node=%s.", tmp_node->GetName().c_str()); | |||
for (const NodePtr &out_node : tmp_node->GetOutAllNodes()) { | |||
if (switch_nodes.find(out_node) == switch_nodes.end()) { | |||
out_nodes.push(out_node); | |||
continue; | |||
} | |||
GELOGD("MarkCycleDependence: tmp_node=%s, switch_node=%s.", | |||
tmp_node->GetName().c_str(), out_node->GetName().c_str()); | |||
GE_IF_BOOL_EXEC(SetCyclicDependenceFlag(out_node) != SUCCESS, | |||
GELOGW("set cyclic dependence attr failed."); return ); | |||
auto map_iter = switch_cyclic_map_.find(out_node); | |||
@@ -602,7 +598,7 @@ Status SwitchToStreamSwitchPass::AddConstNode(const ComputeGraphPtr &graph, cons | |||
/// | |||
Status SwitchToStreamSwitchPass::ModifySwitchInCtlEdges(const NodePtr &switch_node, const NodePtr &cast_node, | |||
const std::set<NodePtr> &same_cond_switch) { | |||
GELOGI("ModifySwitchInCtlEdges: switch_node=%s, active_node=%s", switch_node->GetName().c_str(), | |||
GELOGD("ModifySwitchInCtlEdges: switch_node=%s, active_node=%s", switch_node->GetName().c_str(), | |||
cast_node->GetName().c_str()); | |||
std::string orig_switch_name = switch_node->GetName(); | |||
OpDescPtr switch_desc = switch_node->GetOpDesc(); | |||
@@ -653,7 +649,7 @@ Status SwitchToStreamSwitchPass::ModifySwitchInCtlEdges(const NodePtr &switch_no | |||
/// | |||
Status SwitchToStreamSwitchPass::ModifySwitchOutCtlEdges(const NodePtr &switch_node, const NodePtr &stream_switch, | |||
const NodePtr &active_node) { | |||
GELOGI("ModifySwitchOutCtlEdges: switch_node=%s, stream_switch=%s, active_node=%s", switch_node->GetName().c_str(), | |||
GELOGD("ModifySwitchOutCtlEdges: switch_node=%s, stream_switch=%s, active_node=%s", switch_node->GetName().c_str(), | |||
stream_switch->GetName().c_str(), active_node->GetName().c_str()); | |||
auto find_res = switch_node_map_.find(switch_node); | |||
GE_IF_BOOL_EXEC(find_res == switch_node_map_.end(), { | |||
@@ -18,7 +18,6 @@ | |||
#include <map> | |||
#include <set> | |||
#include <string> | |||
#include <utility> | |||
#include "common/formats/format_transfers/format_transfer_fractal_nz.h" | |||
#include "common/formats/format_transfers/format_transfer_fractal_z.h" | |||
#include "common/formats/format_transfers/format_transfer_nchw_nc1hwc0.h" | |||
@@ -28,13 +27,9 @@ | |||
#include "common/helper/model_helper.h" | |||
#include "common/math/math_util.h" | |||
#include "common/op/ge_op_utils.h" | |||
#include "common/util/error_manager/error_manager.h" | |||
#include "common/formats/utils/formats_trans_utils.h" | |||
#include "framework/common/debug/ge_log.h" | |||
#include "graph/common/ge_call_wrapper.h" | |||
#include "graph/common/local_context.h" | |||
#include "graph/common/transop_util.h" | |||
#include "graph/debug/ge_attr_define.h" | |||
#include "graph/ge_context.h" | |||
#include "graph/shape_refiner.h" | |||
#include "graph/manager/graph_var_manager.h" | |||
@@ -44,29 +39,21 @@ | |||
#include "graph/passes/aicpu_constant_folding_pass.h" | |||
#include "graph/passes/assert_pass.h" | |||
#include "graph/passes/assign_pass.h" | |||
#include "graph/passes/base_pass.h" | |||
#include "graph/passes/common_subexpression_elimination_pass.h" | |||
#include "graph/passes/cond_pass.h" | |||
#include "graph/passes/cond_remove_pass.h" | |||
#include "graph/passes/constant_folding_pass.h" | |||
#include "graph/passes/constant_fuse_same_pass.h" | |||
#include "graph/passes/control_trigger_pass.h" | |||
#include "graph/passes/dimension_adjust_pass.h" | |||
#include "graph/passes/dimension_compute_pass.h" | |||
#include "graph/passes/dropout_pass.h" | |||
#include "graph/passes/enter_pass.h" | |||
#include "graph/passes/flow_ctrl_pass.h" | |||
#include "graph/passes/for_pass.h" | |||
#include "graph/passes/get_original_format_pass.h" | |||
#include "graph/passes/guarantee_const_pass.h" | |||
#include "graph/passes/hccl_group_pass.h" | |||
#include "graph/passes/hccl_memcpy_pass.h" | |||
#include "graph/passes/identity_pass.h" | |||
#include "graph/passes/infershape_pass.h" | |||
#include "graph/passes/iterator_op_pass.h" | |||
#include "graph/passes/merge_pass.h" | |||
#include "graph/passes/net_output_pass.h" | |||
#include "graph/passes/next_iteration_pass.h" | |||
#include "graph/passes/no_use_reshape_remove_pass.h" | |||
#include "graph/passes/parallel_concat_start_op_pass.h" | |||
#include "graph/passes/placeholder_with_default_pass.h" | |||
@@ -81,45 +68,18 @@ | |||
#include "graph/passes/shape_operate_op_remove_pass.h" | |||
#include "graph/passes/snapshot_pass.h" | |||
#include "graph/passes/stop_gradient_pass.h" | |||
#include "graph/passes/subgraph_pass.h" | |||
#include "graph/passes/switch_data_edges_bypass.h" | |||
#include "graph/passes/switch_dead_branch_elimination.h" | |||
#include "graph/passes/switch_logic_remove_pass.h" | |||
#include "graph/passes/merge_to_stream_merge_pass.h" | |||
#include "graph/passes/switch_to_stream_switch_pass.h" | |||
#include "graph/passes/attach_stream_label_pass.h" | |||
#include "graph/passes/unused_const_pass.h" | |||
#include "graph/passes/unused_op_remove_pass.h" | |||
#include "graph/passes/var_is_initialized_op_pass.h" | |||
#include "graph/passes/variable_prepare_op_pass.h" | |||
#include "graph/preprocess/insert_op/util_insert_aipp_op.h" | |||
#include "graph/types.h" | |||
#include "graph/utils/tensor_utils.h" | |||
#include "graph/utils/type_utils.h" | |||
#include "inc/pass_manager.h" | |||
#include "init/gelib.h" | |||
#include "multi_batch_copy_graph.h" | |||
#include "runtime/dev.h" | |||
#include "graph/passes/dimension_adjust_pass.h" | |||
#include "graph/passes/link_gen_mask_nodes_pass.h" | |||
#include "graph/passes/permute_pass.h" | |||
#include "graph/passes/reshape_remove_pass.h" | |||
#include "graph/passes/same_transdata_breadth_fusion_pass.h" | |||
#include "graph/passes/transop_breadth_fusion_pass.h" | |||
#include "graph/passes/transop_depth_fusion_pass.h" | |||
#include "graph/passes/transop_nearby_allreduce_fusion_pass.h" | |||
#include "graph/passes/cast_remove_pass.h" | |||
#include "graph/passes/data_pass.h" | |||
#include "graph/passes/transop_without_reshape_fusion_pass.h" | |||
#include "graph/passes/transpose_transdata_pass.h" | |||
#include "graph/passes/variable_op_pass.h" | |||
#include "graph/passes/variable_prepare_op_pass.h" | |||
#include "graph/passes/variable_ref_delete_op_pass.h" | |||
#include "graph/passes/mark_agnostic_pass.h" | |||
namespace ge { | |||
namespace { | |||
static std::map<std::string, ge::DataType> output_type_str_to_datatype = { | |||