From f6fd4fe3c040e7a1d090c273d6bc29c98a97e78c Mon Sep 17 00:00:00 2001
From: lz <lizheng53@huawei.com>
Date: Tue, 13 Oct 2020 10:08:56 +0800
Subject: [PATCH] fix compile training

---
 .../runtime/kernel/arm/fp32_grad/pooling_grad_fp32_tests.cc  | 1 +
 .../runtime/kernel/arm/fp32_grad/softmax_grad_fp32_tests.cc  | 5 +----
 2 files changed, 2 insertions(+), 4 deletions(-)

diff --git 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
index d895dc8122..2ae6ec1af5 100644
--- 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"
diff --git 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
index 608cb12a0b..ad9914cf91 100644
--- 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"