add stdlib headers

apache · Aug 30, 2018 · e72cb3e · e72cb3e
1 parent d8b4982
commit e72cb3e
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tests/cpp/include/test_mkldnn.h b/tests/cpp/include/test_mkldnn.h
@@ -23,11 +23,14 @@
  *  \author Alex Zai
  */
 
-#ifndef TESTS_MKLDNN_H_
-#define TESTS_MKLDNN_H_
+#ifndef TEST_MKLDNN_H_
+#define TEST_MKLDNN_H_
 
 #if MXNET_USE_MKLDNN == 1
 
+#include <set>
+#include <string>
+#include <vector>
 #include "../../../3rdparty/mkldnn/include/mkldnn_types.h"
 #include "../../../../../../Applications/Xcode.app/Contents/Developer/Toolchains/XcodeDefault.xctoolchain/usr/include/c++/v1/cmath"
 #include "../../../../../../Applications/Xcode.app/Contents/Developer/Toolchains/XcodeDefault.xctoolchain/usr/include/c++/v1/climits"
@@ -578,4 +581,4 @@ void VerifySumResult(const std::vector<NDArray *> &in_arrs,
     ASSERT_EQ(d1[i] + d2[i], o[i]);
 }
 
-#endif
+#endif  // TEST_MKLDNN_H_