未验证 提交 b7b2a527 编写于 作者: W wangzhen38 提交者: GitHub

[gpubox fix] benchmark (#49587)

上级 370b50f6
......@@ -1398,9 +1398,11 @@ void PSGPUWrapper::build_task() {
void PSGPUWrapper::BeginPass() {
platform::Timer timer;
#if defined(PADDLE_WITH_GPU_GRAPH) && defined(PADDLE_WITH_HETERPS)
if (FLAGS_gpugraph_storage_mode == GpuGraphStorageMode::WHOLE_HBM) {
return;
}
#endif
timer.Start();
if (current_task_) {
PADDLE_THROW(
......@@ -1426,9 +1428,11 @@ void PSGPUWrapper::BeginPass() {
}
void PSGPUWrapper::EndPass() {
#if defined(PADDLE_WITH_GPU_GRAPH) && defined(PADDLE_WITH_HETERPS)
if (FLAGS_gpugraph_storage_mode == GpuGraphStorageMode::WHOLE_HBM) {
return;
}
#endif
platform::Timer stagetime;
stagetime.Start();
HbmToSparseTable();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册