fix RecordStreamForGC nullptr exception (#54606)

changed

fix RecordStreamForGC nullptr exception (#54606)
changed
5f92cc54 · engineer1109 · GitHub · 584ae4d7 · 5f92cc54
隐藏空白更改
内联并排

Showing with 3 addition and 0 deletion

paddle/fluid/framework/new_executor/interpretercore.cc paddle/fluid/framework/new_executor/interpretercore.cc +3 -0

未找到文件。
--- a/paddle/fluid/framework/new_executor/interpretercore.cc
+++ b/paddle/fluid/framework/new_executor/interpretercore.cc
@@ -1278,6 +1278,9 @@ void InterpreterCore::RecordStreamForGC(const Instruction& instr) {
      instr.KernelType() != OpFuncType::kGpuAsync) {
    return;
  }
+  if (instr.DeviceContext().GetPlace() == phi::CustomPlace()) {
+    return;
+  }
  platform::RecordEvent record(
      "RecordStreamForGC", platform::TracerEventType::UserDefined, 10);