!7221 fix compile training

Merge pull request !7221 from lz/master
5 years ago · 6ce86c2074
parent 2cb27c59e0 f6fd4fe3c0
commit 6ce86c2074
2 changed files with 2 additions and 4 deletions
--- a/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/pooling_grad_fp32_tests.cc
+++ b/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/pooling_grad_fp32_tests.cc
@ -16,6 +16,7 @@

 #include <iostream>
 #include <memory>
+#include "src/ops/primitive_c.h"
 #include "mindspore/lite/include/context.h"
 #include "src/common/log_adapter.h"
 #include "common/common_test.h"
--- a/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/softmax_grad_fp32_tests.cc
+++ b/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32_grad/softmax_grad_fp32_tests.cc
@ -17,10 +17,7 @@
 #include <iostream>
 #include <memory>
 #include <vector>
-
-// #include "mindspore/lite/src/ir/tensor.h"
-// #include "mindspore/lite/src/lite_kernel.h"
-
+#include "src/ops/primitive_c.h"
 #include "mindspore/lite/include/context.h"
 #include "src/common/log_adapter.h"
 #include "common/common_test.h"