fix RecordStreamForGC nullptr exception (#54606)

engineer1109 · web-flow · commit 5f92cc54b361 · 2023-06-16T14:40:46.000+08:00
changed
diff --git 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);
 

Original file line number	Diff line number	Diff line change
`@@ -1278,6 +1278,9 @@ void InterpreterCore::RecordStreamForGC(const Instruction& instr) {`
`1278`	`1278`	`instr.KernelType() != OpFuncType::kGpuAsync) {`
`1279`	`1279`	`return;`
`1280`	`1280`	`}`
	`1281`	`+ if (instr.DeviceContext().GetPlace() == phi::CustomPlace()) {`
	`1282`	`+ return;`
	`1283`	`+ }`
`1281`	`1284`	`platform::RecordEvent record(`
`1282`	`1285`	`"RecordStreamForGC", platform::TracerEventType::UserDefined, 10);`
`1283`	`1286`