未验证 提交 70df3aa4 编写于 作者: E engineer1109 提交者: GitHub

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

上级 0685b3ec
...@@ -1150,7 +1150,8 @@ void NewIRInterpreter::RecordStreamForGC(const Instruction& instr) { ...@@ -1150,7 +1150,8 @@ void NewIRInterpreter::RecordStreamForGC(const Instruction& instr) {
instr.KernelType() != OpFuncType::kGpuAsync) { instr.KernelType() != OpFuncType::kGpuAsync) {
return; return;
} }
if (instr.DeviceContext().GetPlace() == phi::CustomPlace()) { if (instr.DeviceContext().GetPlace().GetType() ==
phi::AllocationType::CUSTOM) {
return; return;
} }
platform::RecordEvent record( platform::RecordEvent record(
......
...@@ -1136,7 +1136,8 @@ void ProgramInterpreter::RecordStreamForGC(const Instruction& instr) { ...@@ -1136,7 +1136,8 @@ void ProgramInterpreter::RecordStreamForGC(const Instruction& instr) {
return; return;
} }
if (instr.DeviceContext().GetPlace() == phi::CustomPlace()) { if (instr.DeviceContext().GetPlace().GetType() ==
phi::AllocationType::CUSTOM) {
return; return;
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册