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