From 357a45128dba99b6a5ef4b0e44ced509b4991837 Mon Sep 17 00:00:00 2001 From: miaoli06 <106585574+miaoli06@users.noreply.github.com> Date: Sat, 17 Sep 2022 20:55:39 +0800 Subject: [PATCH] fix clear gpu mem (#119) Co-authored-by: root --- paddle/fluid/framework/data_feed.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/framework/data_feed.h b/paddle/fluid/framework/data_feed.h index 4b99ad0164c11..35774170d13f0 100644 --- a/paddle/fluid/framework/data_feed.h +++ b/paddle/fluid/framework/data_feed.h @@ -1068,7 +1068,7 @@ class DataFeed { #endif virtual void clear_gpu_mem() { -#if defined(PADDLE_WITH_GPU_GRAPH) && defined(PADDLE_WITH_HETEPS) +#if defined(PADDLE_WITH_GPU_GRAPH) && defined(PADDLE_WITH_HETERPS) gpu_graph_data_generator_.clear_gpu_mem(); #endif }