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

[gpubox fix] benchmark (#49587)

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