Fixed scheduling

aidge_backend_cpu/unit_tests/test_scheduler.py

@@ -55,6 +55,8 @@ class test_scheduler(unittest.TestCase):
         graph_view.set_datatype(aidge_core.DataType.Float32)
         graph_view.set_backend("cpu")
 
+        graph_view.forward_dims()
+
         scheduler = aidge_core.SequentialScheduler(graph_view)
         scheduler.generate_scheduling()
 
@@ -80,6 +82,8 @@ class test_scheduler(unittest.TestCase):
         graph_view.set_datatype(aidge_core.DataType.Float32)
         graph_view.set_backend("cpu")
 
+        graph_view.forward_dims()
+
         scheduler = aidge_core.SequentialScheduler(graph_view)
         scheduler.generate_scheduling()
 

include/aidge/backend/cpu/operator/ProducerImpl.hpp

@@ -28,7 +28,8 @@ public:
         return std::make_unique<ProducerImpl_cpu>(op);
     }
 
-    void forward() override {};
+    NbElts_t getNbProducedData(const IOIndex_t outputIdx) const override final;
+    void forward() override;
 };
 
 namespace {

src/operator/ProducerImpl.cpp

+/********************************************************************************
+ * 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 <cassert>
+#include <numeric> // std::accumulate
+#include <vector>
+
+#include "aidge/data/Tensor.hpp"
+#include "aidge/operator/Producer.hpp"
+#include "aidge/utils/Types.h"
+
+#include "aidge/backend/cpu/operator/ProducerImpl.hpp"
+
+Aidge::DimSize_t Aidge::ProducerImpl_cpu::getNbProducedData(
+    Aidge::IOIndex_t outputIdx) const
+{
+    // Requires the whole tensors, regardless of available data on inputs
+    assert(outputIdx == 0 && "operator has only one output");
+    (void) outputIdx;
+
+    return std::static_pointer_cast<Tensor>(mOp.getOutput(0))->size();
+}
+
+void Aidge::ProducerImpl_cpu::forward()
+{
+}