From afe0d5a7b745ed4850a48ca9b200055ade7d4540 Mon Sep 17 00:00:00 2001
From: Michal Tarnowski <michal.tarnowski@tcl.com>
Date: Mon, 4 Feb 2019 09:20:50 +0100
Subject: [PATCH] Additional tests added for GPU Eltwise

---
 mace/ops/eltwise_test.cc | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

diff --git a/mace/ops/eltwise_test.cc b/mace/ops/eltwise_test.cc
index 47b96bd0..d6a7b449 100644
--- a/mace/ops/eltwise_test.cc
+++ b/mace/ops/eltwise_test.cc
@@ -803,6 +803,28 @@ TEST_F(EltwiseOpTest, TensorGeneralBroadcastGPU) {
   TensorGeneralBroadcastEltwise<DeviceType::GPU, float, float>(
       ops::EltwiseType::SQR_DIFF, {1, 1, 2, 3}, {1, 2, 3, 4, 5, 6},
       {1, 1, 2, 1}, {1, 2}, {1, 1, 2, 3}, {0, 1, 4, 4, 9, 16});
+
+  TensorGeneralBroadcastEltwise<DeviceType::GPU, float, float>(
+      ops::EltwiseType::SUM, {1, 1, 2, 5}, {0, 1, 2, 3, 4, 5, 6, 7, 8, 9}, 
+      {1, 1, 2, 1}, {1, 2}, {1, 1, 2, 5}, {1, 2, 3, 4, 5, 7, 8, 9, 10, 11});
+  TensorGeneralBroadcastEltwise<DeviceType::GPU, float, float>(
+      ops::EltwiseType::SUB, {1, 1, 2, 5}, {0, 1, 2, 3, 4, 5, 6, 7, 8, 9}, 
+      {1, 1, 2, 1}, {1, 2}, {1, 1, 2, 5}, {-1, 0, 1, 2, 3, 3, 4, 5, 6, 7});
+  TensorGeneralBroadcastEltwise<DeviceType::GPU, float, float>(
+      ops::EltwiseType::PROD, {1, 1, 2, 5}, {0, 1, 2, 3, 4, 5, 6, 7, 8, 9}, 
+      {1, 1, 2, 1}, {1, 2}, {1, 1, 2, 5}, {0, 1, 2, 3, 4, 10, 12, 14, 16, 18});
+  TensorGeneralBroadcastEltwise<DeviceType::GPU, float, float>(
+      ops::EltwiseType::DIV, {1, 1, 2, 5}, {0, 1, 2, 3, 4, 5, 6, 7, 8, 9}, 
+      {1, 1, 2, 1}, {1, 2}, {1, 1, 2, 5}, {0, 1, 2, 3, 4, 2.5, 3, 3.5, 4, 4.5});
+  TensorGeneralBroadcastEltwise<DeviceType::GPU, float, float>(
+      ops::EltwiseType::MIN, {1, 1, 2, 5}, {0, 1, 2, 3, 4, 5, 6, 7, 8, 9}, 
+      {1, 1, 2, 1}, {3, 4}, {1, 1, 2, 5}, {0, 1, 2, 3, 3, 4, 4, 4, 4, 4});
+  TensorGeneralBroadcastEltwise<DeviceType::GPU, float, float>(
+      ops::EltwiseType::MAX, {1, 1, 2, 5}, {0, 1, 2, 3, 4, 5, 6, 7, 8, 9}, 
+      {1, 1, 2, 1}, {3, 4}, {1, 1, 2, 5}, {3, 3, 3, 3, 4, 5, 6, 7, 8, 9});
+  TensorGeneralBroadcastEltwise<DeviceType::GPU, float, float>(
+      ops::EltwiseType::SQR_DIFF, {1, 1, 2, 5}, {0, 1, 2, 3, 4, 5, 6, 7, 8, 9}, 
+      {1, 1, 2, 1}, {2, 3}, {1, 1, 2, 5}, {4, 1, 0, 1, 4, 4, 9, 16, 25, 36});
 }
 
 TEST_F(EltwiseOpTest, QuantizedSum) {
-- 
GitLab