[CustomDevice]fix == error with place (#55173)

engineer1109 · web-flow · commit 70df3aa4ea81 · 2023-07-07T10:32:40.000+08:00
diff --git a/paddle/fluid/framework/new_executor/new_ir_interpreter.cc b/paddle/fluid/framework/new_executor/new_ir_interpreter.cc
@@ -1150,7 +1150,8 @@ void NewIRInterpreter::RecordStreamForGC(const Instruction& instr) {
       instr.KernelType() != OpFuncType::kGpuAsync) {
     return;
   }
-  if (instr.DeviceContext().GetPlace() == phi::CustomPlace()) {
+  if (instr.DeviceContext().GetPlace().GetType() ==
+      phi::AllocationType::CUSTOM) {
     return;
   }
   platform::RecordEvent record(
diff --git a/paddle/fluid/framework/new_executor/program_interpreter.cc b/paddle/fluid/framework/new_executor/program_interpreter.cc
@@ -1136,7 +1136,8 @@ void ProgramInterpreter::RecordStreamForGC(const Instruction& instr) {
     return;
   }
 
-  if (instr.DeviceContext().GetPlace() == phi::CustomPlace()) {
+  if (instr.DeviceContext().GetPlace().GetType() ==
+      phi::AllocationType::CUSTOM) {
     return;
   }
 

Original file line number	Diff line number	Diff line change
`@@ -1150,7 +1150,8 @@ void NewIRInterpreter::RecordStreamForGC(const Instruction& instr) {`
`1150`	`1150`	`instr.KernelType() != OpFuncType::kGpuAsync) {`
`1151`	`1151`	`return;`
`1152`	`1152`	`}`
`1153`		`- if (instr.DeviceContext().GetPlace() == phi::CustomPlace()) {`
	`1153`	`+ if (instr.DeviceContext().GetPlace().GetType() ==`
	`1154`	`+ phi::AllocationType::CUSTOM) {`
`1154`	`1155`	`return;`
`1155`	`1156`	`}`
`1156`	`1157`	`platform::RecordEvent record(`
Original file line number	Diff line number	Diff line change
`@@ -1136,7 +1136,8 @@ void ProgramInterpreter::RecordStreamForGC(const Instruction& instr) {`
`1136`	`1136`	`return;`
`1137`	`1137`	`}`
`1138`	`1138`
`1139`		`- if (instr.DeviceContext().GetPlace() == phi::CustomPlace()) {`
	`1139`	`+ if (instr.DeviceContext().GetPlace().GetType() ==`
	`1140`	`+ phi::AllocationType::CUSTOM) {`
`1140`	`1141`	`return;`
`1141`	`1142`	`}`
`1142`	`1143`