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

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

上级 0685b3ec
......@@ -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(
......
......@@ -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;
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册