From 9c5c8aae98a27f6bb67a3c1baf2f13458001523c Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Wed, 3 Jul 2024 10:48:14 +0200
Subject: [PATCH] Fixed convToMatMul + unit test
---
unit_tests/recipies/Test_ConvToMatMul.cpp | 66 +++++++++++++++++++++++
1 file changed, 66 insertions(+)
create mode 100644 unit_tests/recipies/Test_ConvToMatMul.cpp
diff --git a/unit_tests/recipies/Test_ConvToMatMul.cpp b/unit_tests/recipies/Test_ConvToMatMul.cpp
new file mode 100644
index 00000000..ef3920f0
--- /dev/null
+++ b/unit_tests/recipies/Test_ConvToMatMul.cpp
@@ -0,0 +1,66 @@
+/********************************************************************************
+ * Copyright (c) 2023 CEA-List
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License 2.0 which is available at
+ * http://www.eclipse.org/legal/epl-2.0.
+ *
+ * SPDX-License-Identifier: EPL-2.0
+ *
+ ********************************************************************************/
+
+#include <catch2/catch_test_macros.hpp>
+
+#include "aidge/recipes/Recipes.hpp"
+#include "aidge/operator/Conv.hpp"
+#include "aidge/operator/Producer.hpp"
+#include "aidge/scheduler/SequentialScheduler.hpp"
+#include "aidge/filler/Filler.hpp"
+#include "aidge/graph/OpArgs.hpp"
+#include <cstddef>
+
+using namespace Aidge;
+
+TEST_CASE("[ConvToMatMul] conv") {
+ auto conv1 = Conv(3, 4, {3, 3}, "conv1");
+ auto conv2 = Conv(4, 7, {3, 3}, "conv2", {1, 1}, {1, 1}, true);
+ auto conv3 = Conv(7, 10, {1, 1}, "conv3", {2, 2});
+
+ auto g1 = Sequential({
+ Producer({2, 3, 13, 24}, "dataProvider"),
+ conv1,
+ conv2,
+ conv3
+ });
+
+ g1->setBackend("cpu");
+ g1->forwardDims();
+
+ // Random initialization of input and weights
+ uniformFiller<float>(std::static_pointer_cast<OperatorTensor>(conv1->getOperator())->getInput(0), -10.0, 10.0);
+ uniformFiller<float>(std::static_pointer_cast<OperatorTensor>(conv1->getOperator())->getInput(1), -10.0, 10.0);
+ uniformFiller<float>(std::static_pointer_cast<OperatorTensor>(conv1->getOperator())->getInput(2), -10.0, 10.0);
+ uniformFiller<float>(std::static_pointer_cast<OperatorTensor>(conv2->getOperator())->getInput(1), -10.0, 10.0);
+ uniformFiller<float>(std::static_pointer_cast<OperatorTensor>(conv3->getOperator())->getInput(1), -10.0, 10.0);
+ uniformFiller<float>(std::static_pointer_cast<OperatorTensor>(conv3->getOperator())->getInput(2), -10.0, 10.0);
+
+ auto s1 = SequentialScheduler(g1);
+ s1.forward();
+
+ g1->save("convToMatMul_before");
+
+ auto g2 = g1->clone();
+ g2->forwardDims();
+ REQUIRE(convToMatMul(g2) == 3);
+
+ g2->setBackend("cpu");
+
+ auto s2 = SequentialScheduler(g2);
+ s2.forward();
+
+ g2->save("convToMatMul_after");
+
+ auto g1OutOp = std::static_pointer_cast<OperatorTensor>((*g1->outputNodes().cbegin())->getOperator());
+ auto g2OutOp = std::static_pointer_cast<OperatorTensor>((*g1->outputNodes().cbegin())->getOperator());
+ REQUIRE(*(g1OutOp->getOutput(0)) == *(g2OutOp->getOutput(0)));
+}
--
GitLab