fix lint

apache · Aug 30, 2018 · 5ce1d4f · 5ce1d4f
1 parent 8daf66d
commit 5ce1d4f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/tests/cpp/include/test_mkldnn.h b/tests/cpp/include/test_mkldnn.h
@@ -23,6 +23,9 @@
  *  \author Alex Zai
  */
 
+#ifndef TESTS_MKLDNN_H_
+#define TESTS_MKLDNN_H_
+
 #if MXNET_USE_MKLDNN == 1
 
 #include "../../../3rdparty/mkldnn/include/mkldnn_types.h"
@@ -36,9 +39,6 @@
 #include "../../../src/operator/nn/mkldnn/mkldnn_pooling-inl.h"
 #include "../../../src/operator/nn/pooling-inl.h"
 
-#ifndef MXNET_STORAGE_H_
-#define MXNET_STORAGE_H_
-
 using namespace mxnet;
 
 static mkldnn::memory::primitive_desc GetMemPD(const TShape s, int dtype,

diff --git a/tests/cpp/operator/mkldnn_test.cc b/tests/cpp/operator/mkldnn_test.cc
@@ -291,7 +291,7 @@ TEST(MKLDNN_BASE, CreateMKLDNNMem) {
     InitMKLDNNArray(&orig_arr.arr, input_mem->get_primitive_desc());
     orig_arr.arr.CopyFrom(*input_mem);
     auto output_mem_t = CreateMKLDNNMem(in_arr.arr,
-                                        input_mem->get_primitive_desc(), kWriteInplace, &in_arr.arr);
+        input_mem->get_primitive_desc(), kWriteInplace, &in_arr.arr);
     op::MKLDNNSum(*input_mem, *input_mem2, *output_mem_t.second);
     CommitOutput(in_arr.arr, output_mem_t);
     stream->Submit();
@@ -395,4 +395,4 @@ TEST(MKLDNN_NDArray, GetTestOutputArraysConcat) {
   }
 }
 
-#endif
+#endif