fix clear gpu mem (PaddlePaddle#119)

Co-authored-by: root <root@yq01-inf-hic-k8s-a100-ab2-0009.yq01.baidu.com>
danleifeng · Sep 17, 2022 · 357a451 · 357a451
1 parent 04110a5
commit 357a451
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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
   }