[AutoParallel] Fix pipeline visualization tool (#71386)

AndSonder · web-flow · commit 96a8cf438a23 · 2025-03-05T15:49:52.000+08:00
* fix profile tool

* fix profile tool
diff --git a/paddle/fluid/framework/new_executor/pir_interpreter.cc b/paddle/fluid/framework/new_executor/pir_interpreter.cc
@@ -1938,7 +1938,8 @@ void PirInterpreter::RunInstructionBase(InstructionBase* instr_node) {
       std::string op_name = instr_node->Name();
       ::pir::Operation* op = instr_node->Operation();
       if (!calculate_stream_timer_->IsStarted() && op_name != "pd_op.feed" &&
-          !op->HasAttribute("ring_id")) {
+          !op->HasAttribute("ring_id") && op_name != "pd_op.shadow_feed" &&
+          op_name != "pd_op.full_int_array") {
         VLOG(3) << "Start calculated stream timer from op: " << op_name;
         calculate_stream_timer_->Start();
       }
diff --git a/python/paddle/distributed/auto_parallel/static/profiler_helper_static.py b/python/paddle/distributed/auto_parallel/static/profiler_helper_static.py
@@ -33,6 +33,8 @@
     "default": "thread_state_unknown",  # RGB: 199, 155, 125
 }
 
+ignore_job_type = ["recv_forward", "send_backward"]
+
 
 def parse_args():
     parser = ArgumentParser()
@@ -60,6 +62,9 @@ def process_job_log(log_data, device_id, multi_machine_idx=-1):
     for i, match in enumerate(matches):
         job_id, job_type, micro_batch_id, job_start_time, job_end_time = match
 
+        if job_type in ignore_job_type:
+            continue
+
         if job_type != "default" and start_job_type == "":
             start_job_type = job_type
 
