From 009dbe8e0a07fc63e06eed33f20c34d81ddd8d6c Mon Sep 17 00:00:00 2001
From: Liangliang He <lliang.he@gmail.com>
Date: Mon, 23 Apr 2018 10:18:02 +0800
Subject: [PATCH] Fix batch noram unit test

---
 mace/ops/batch_norm_test.cc | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/mace/ops/batch_norm_test.cc b/mace/ops/batch_norm_test.cc
index 7f1bb037..8bb742f1 100644
--- a/mace/ops/batch_norm_test.cc
+++ b/mace/ops/batch_norm_test.cc
@@ -115,7 +115,7 @@ TEST_F(BatchNormOpTest, SimpleRandomOPENCL) {
   net.AddRandomInput<DeviceType::OPENCL, float>("Scale", {channels});
   net.AddRandomInput<DeviceType::OPENCL, float>("Offset", {channels});
   net.AddRandomInput<DeviceType::OPENCL, float>("Mean", {channels});
-  net.AddRandomInput<DeviceType::OPENCL, float>("Var", {channels}, true);
+  net.AddRandomInput<DeviceType::OPENCL, float>("Var", {channels});
 
   // run cpu
   net.RunOp();
@@ -176,7 +176,7 @@ TEST_F(BatchNormOpTest, SimpleRandomHalfOPENCL) {
     .Input("Offset")
     .Input("Mean")
     .Input("Var")
-    .AddFloatArg("epsilon", 1e-3)
+    .AddFloatArg("epsilon", 1e-1)
     .Output("Output")
     .Finalize(net.NewOperatorDef());
 
@@ -186,7 +186,7 @@ TEST_F(BatchNormOpTest, SimpleRandomHalfOPENCL) {
   net.AddRandomInput<DeviceType::OPENCL, float>("Scale", {channels});
   net.AddRandomInput<DeviceType::OPENCL, float>("Offset", {channels});
   net.AddRandomInput<DeviceType::OPENCL, float>("Mean", {channels});
-  net.AddRandomInput<DeviceType::OPENCL, float>("Var", {channels}, true);
+  net.AddRandomInput<DeviceType::OPENCL, float>("Var", {channels});
 
   // run cpu
   net.RunOp();
@@ -213,7 +213,7 @@ TEST_F(BatchNormOpTest, SimpleRandomHalfOPENCL) {
     .Input("OffsetImage")
     .Input("MeanImage")
     .Input("VarImage")
-    .AddFloatArg("epsilon", 1e-3)
+    .AddFloatArg("epsilon", 1e-1)
     .Output("OutputImage")
     .AddIntArg("T", static_cast<int>(DataType::DT_HALF))
     .Finalize(net.NewOperatorDef());
@@ -258,7 +258,7 @@ TEST_F(BatchNormOpTest, ComplexRandomOPENCL) {
   net.AddRandomInput<DeviceType::OPENCL, float>("Scale", {channels});
   net.AddRandomInput<DeviceType::OPENCL, float>("Offset", {channels});
   net.AddRandomInput<DeviceType::OPENCL, float>("Mean", {channels});
-  net.AddRandomInput<DeviceType::OPENCL, float>("Var", {channels}, true);
+  net.AddRandomInput<DeviceType::OPENCL, float>("Var", {channels});
 
   // run cpu
   net.RunOp();
@@ -319,7 +319,7 @@ TEST_F(BatchNormOpTest, ComplexRandomHalfOPENCL) {
     .Input("Offset")
     .Input("Mean")
     .Input("Var")
-    .AddFloatArg("epsilon", 1e-3)
+    .AddFloatArg("epsilon", 1e-1)
     .Output("Output")
     .Finalize(net.NewOperatorDef());
 
@@ -329,7 +329,7 @@ TEST_F(BatchNormOpTest, ComplexRandomHalfOPENCL) {
   net.AddRandomInput<DeviceType::OPENCL, float>("Scale", {channels});
   net.AddRandomInput<DeviceType::OPENCL, float>("Offset", {channels});
   net.AddRandomInput<DeviceType::OPENCL, float>("Mean", {channels});
-  net.AddRandomInput<DeviceType::OPENCL, float>("Var", {channels}, true);
+  net.AddRandomInput<DeviceType::OPENCL, float>("Var", {channels});
 
   // run cpu
   net.RunOp();
@@ -356,7 +356,7 @@ TEST_F(BatchNormOpTest, ComplexRandomHalfOPENCL) {
     .Input("OffsetImage")
     .Input("MeanImage")
     .Input("VarImage")
-    .AddFloatArg("epsilon", 1e-3)
+    .AddFloatArg("epsilon", 1e-1)
     .Output("OutputImage")
     .AddIntArg("T", static_cast<int>(DataType::DT_HALF))
     .Finalize(net.NewOperatorDef());
@@ -403,7 +403,7 @@ TEST_F(BatchNormOpTest, NEONTest) {
   net.AddRandomInput<DeviceType::CPU, float>("Scale", {channels});
   net.AddRandomInput<DeviceType::CPU, float>("Offset", {channels});
   net.AddRandomInput<DeviceType::CPU, float>("Mean", {channels});
-  net.AddRandomInput<DeviceType::CPU, float>("Var", {channels}, true);
+  net.AddRandomInput<DeviceType::CPU, float>("Var", {channels});
 
   // run cpu
   net.RunOp();
-- 
GitLab