fix travis-ci

test=develop
6 years ago · 487624e15d
parent 1e9fd40777
commit 487624e15d
1 changed files with 2 additions and 0 deletions
--- a/paddle/fluid/operators/cross_entropy_op.cc
+++ b/paddle/fluid/operators/cross_entropy_op.cc
@ -13,7 +13,9 @@ See the License for the specific language governing permissions and
 limitations under the License. */

 #include "paddle/fluid/operators/cross_entropy_op.h"
+#include <memory>
 #include <string>
+#include <unordered_map>

 namespace paddle {
 namespace operators {