Skip to content
Snippets Groups Projects
Commit adf85efb authored by Cyril Moineau's avatar Cyril Moineau
Browse files

Add getGraphview in scheduler.

parent 4b783082
No related branches found
No related tags found
No related merge requests found
...@@ -322,17 +322,17 @@ public: ...@@ -322,17 +322,17 @@ public:
/** /**
* @brief Insert a node (newParentNode) as a parent of the passed node (childNode). * @brief Insert a node (newParentNode) as a parent of the passed node (childNode).
* *
* @param childNode Node that gets a new parent. * @param childNode Node that gets a new parent.
* @param newParentNode Inserted Node. * @param newParentNode Inserted Node.
* @param childInputTensorIdx Index of the input Tensor for the childNode linked to the inserted Node output. * @param childInputTensorIdx Index of the input Tensor for the childNode linked to the inserted Node output.
* @param newParentInputTensorIdx Index of the input Tensor for the newParentNode linked to the former parent of childNode. * @param newParentInputTensorIdx Index of the input Tensor for the newParentNode linked to the former parent of childNode.
* @param newParentOutputTensorIdx Index of the output Tensor for the newParentNode linked to the childNode's input Tensor. * @param newParentOutputTensorIdx Index of the output Tensor for the newParentNode linked to the childNode's input Tensor.
*/ */
void insertParent(NodePtr childNode, void insertParent(NodePtr childNode,
NodePtr newParentNode, NodePtr newParentNode,
IOIndex_t childInputTensorIdx, IOIndex_t childInputTensorIdx,
IOIndex_t newParentInputTensorIdx, IOIndex_t newParentInputTensorIdx,
IOIndex_t newParentOutputTensorIdx); IOIndex_t newParentOutputTensorIdx);
/** /**
...@@ -421,4 +421,4 @@ private: ...@@ -421,4 +421,4 @@ private:
}; };
} // namespace Aidge } // namespace Aidge
#endif /* AIDGE_CORE_GRAPH_GRAPHVIEW_H_ */ #endif /* AIDGE_CORE_GRAPH_GRAPHVIEW_H_ */
\ No newline at end of file
...@@ -187,7 +187,7 @@ public: ...@@ -187,7 +187,7 @@ public:
IOIndex_t getNbFreeDataInputs() const; IOIndex_t getNbFreeDataInputs() const;
/** /**
* @brief List input ids of children liked to outputs of the node * @brief List input ids of children linked to outputs of the node
* @return std::vector<std::vector<std::pair<std::shared_ptr<Node>, * @return std::vector<std::vector<std::pair<std::shared_ptr<Node>,
* IOIndex_t>>> * IOIndex_t>>>
*/ */
...@@ -402,7 +402,7 @@ public: ...@@ -402,7 +402,7 @@ public:
/** /**
* @brief Get the set of pointers to connected node at a distance of a delta. * @brief Get the set of pointers to connected node at a distance of a delta.
* @details the recution are cut * @details the recution are cut
* Return a nullptr is nofing found. * Return a nullptr is nofing found.
* @param delta Input delta. * @param delta Input delta.
* @return std::shared_ptr<Node> * @return std::shared_ptr<Node>
......
...@@ -64,6 +64,9 @@ public: ...@@ -64,6 +64,9 @@ public:
std::vector<std::shared_ptr<Node>> getStaticScheduling(){ std::vector<std::shared_ptr<Node>> getStaticScheduling(){
return mStaticSchedule; return mStaticSchedule;
} }
std::shared_ptr<GraphView> getGraphView(){
return mGraphView;
}
private: private:
/** /**
......
...@@ -20,7 +20,9 @@ void init_Operator(py::module& m){ ...@@ -20,7 +20,9 @@ void init_Operator(py::module& m){
py::class_<Operator, std::shared_ptr<Operator>>(m, "Operator") py::class_<Operator, std::shared_ptr<Operator>>(m, "Operator")
.def("output", &Operator::output, py::arg("outputIdx")) .def("output", &Operator::output, py::arg("outputIdx"))
.def("input", &Operator::input, py::arg("inputIdx")) .def("input", &Operator::input, py::arg("inputIdx"))
.def("nb_inputs", &Operator::nbInputs)
.def("nb_data_inputs", &Operator::nbDataInputs) .def("nb_data_inputs", &Operator::nbDataInputs)
.def("nb_outputs", &Operator::nbOutputs)
.def("associate_input", &Operator::associateInput, py::arg("inputIdx"), py::arg("data")) .def("associate_input", &Operator::associateInput, py::arg("inputIdx"), py::arg("data"))
.def("set_datatype", &Operator::setDatatype, py::arg("datatype")) .def("set_datatype", &Operator::setDatatype, py::arg("datatype"))
.def("set_backend", &Operator::setBackend, py::arg("name")) .def("set_backend", &Operator::setBackend, py::arg("name"))
......
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