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