Skip to content
Snippets Groups Projects
Commit 2b61d61a authored by Noam Zerah's avatar Noam Zerah
Browse files

fix: Test Metaoperator scaling fix

parent 71259085
No related branches found
No related tags found
No related merge requests found
Pipeline #57055 failed
......@@ -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);
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment