From 2b61d61a9b3db810d2fb091b99944d957b7b6daa Mon Sep 17 00:00:00 2001
From: Noam ZERAH <noam.zerah@cea.fr>
Date: Wed, 16 Oct 2024 08:13:04 +0000
Subject: [PATCH] fix: Test Metaoperator scaling fix
---
unit_tests/operator/Test_ScalingMeta.cpp | 15 +++++++++------
1 file changed, 9 insertions(+), 6 deletions(-)
diff --git a/unit_tests/operator/Test_ScalingMeta.cpp b/unit_tests/operator/Test_ScalingMeta.cpp
index 592531b8..ae39ee0a 100644
--- a/unit_tests/operator/Test_ScalingMeta.cpp
+++ b/unit_tests/operator/Test_ScalingMeta.cpp
@@ -63,13 +63,16 @@ TEST_CASE("ScalingNodeMeta", "[ScalingMeta][CPU]") {
auto scal = MulPTQ(2.001);
auto scalop = std::static_pointer_cast<OperatorTensor>(scal->getOperator());
-
t0->setBackend("cpu");
- scalop->associateInput(0,t0);
- scalop->setBackend("cpu");
- scalop->forwardDims();
- //scalop->forward();
- scal->forward();
+ scal->getOperator()->associateInput(0,t0);
+
+ auto g = getConnectedGraphView(scal);
+ g->setDataType(DataType::Float32);
+ g->setBackend("cpu");
+
+ auto scheduler = SequentialScheduler(g);
+ scheduler.forward();
+
auto out0 = scalop->getOutput(0);
auto in0 = scalop->getInput(0);
auto in1 = scalop->getInput(1);
--
GitLab