add include (#29952)

4 years ago · 0ca6de171f
parent 898486dd46
commit 0ca6de171f
1 changed files with 1 additions and 0 deletions
--- a/paddle/fluid/framework/fleet/heter_ps/heter_ps_base.h
+++ b/paddle/fluid/framework/fleet/heter_ps/heter_ps_base.h
@ -15,6 +15,7 @@ limitations under the License. */
 #pragma once
 #include <vector>
 #include "paddle/fluid/framework/fleet/heter_ps/feature_value.h"
+#include "paddle/fluid/framework/fleet/heter_ps/heter_resource.h"

 #ifdef PADDLE_WITH_PSLIB