[PIR] add kejibu log (#68382)

wanghuancoder · web-flow · commit 37fc1541d527 · 2024-09-24T18:33:43.000+08:00
* add kejibu log

* refine
diff --git a/paddle/fluid/framework/new_executor/instruction/phi_kernel_instruction.cc b/paddle/fluid/framework/new_executor/instruction/phi_kernel_instruction.cc
@@ -33,6 +33,11 @@
 #include "paddle/pir/include/core/value.h"
 
 #include "paddle/fluid/framework/new_executor/instruction/instruction_util.h"
+
+PHI_DEFINE_EXPORTED_bool(print_kernel_run_info,
+                         false,
+                         "Whether print kernel run info.");
+
 namespace paddle {
 namespace framework {
 
@@ -180,6 +185,32 @@ PhiKernelInstruction::PhiKernelInstruction(
 PhiKernelInstruction::~PhiKernelInstruction() { delete phi_kernel_; }
 
 void PhiKernelInstruction::Run() {
+  if (FLAGS_print_kernel_run_info) {
+#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
+    auto place =
+        kernel_context_.GetDeviceContext<phi::DeviceContext>().GetPlace();
+    if (phi::is_gpu_place(place)) {
+      std::string use_cudnn =
+          op_->attributes()
+                      .at("kernel_key")
+                      .dyn_cast<paddle::dialect::KernelAttribute>()
+                      .data()
+                      .backend() == phi::Backend::GPUDNN
+              ? "true"
+              : "false";
+      std::cout << "run " << phi_op_name_
+                << ": thread_id=" << std::this_thread::get_id()
+                << ", backend=" << place << ", use_cudnn=" << use_cudnn
+                << ", stream="
+                << kernel_context_.GetDeviceContext<phi::GPUContext>().stream()
+                << std::endl;
+    } else {
+      std::cout << "run " << phi_op_name_
+                << ": thread_id=" << std::this_thread::get_id()
+                << ", backend=" << place << std::endl;
+    }
+#endif
+  }
   VLOG(6) << "Begin run op " << phi_op_name_ << " infer meta.";
   if (infer_meta_interface_) {
     phi::RecordEvent record_event("PhiKernelInstruction::infermeta",
