未验证 提交 eba7b584 编写于 作者: P pangyoki 提交者: GitHub

change no_event GC to fast GC for xpu (#49871)

上级 48b3e869
......@@ -60,7 +60,16 @@ CreateInterpreterCoreGarbageCollector(
return std::unique_ptr<InterpreterCoreGarbageCollector>(
new InterpreterCoreEventGarbageCollector(vec_instruction));
}
} else if (platform::is_xpu_place(place) || platform::is_ipu_place(place)) {
} else if (platform::is_xpu_place(place)) {
// Because there is no multi-stream on XPU device, fast GC can
// be used.
// Previously, XPU used no_event GC. But `Wait` in no_event GC
// may cause GC delayed, causing no enough memory problem.
// TODO(pangyoki): Multi-stream allocator and multi-stream GC
// are needed to be adapted for XPU.
return std::unique_ptr<InterpreterCoreGarbageCollector>(
new InterpreterCoreFastGarbageCollector());
} else if (platform::is_ipu_place(place)) {
return std::unique_ptr<InterpreterCoreGarbageCollector>(
new InterpreterCoreNoEventGarbageCollector());
} else {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册