huajsj commented on a change in pull request #9751:
URL: https://github.com/apache/tvm/pull/9751#discussion_r770232372
##########
File path: src/runtime/pipeline/pipeline_executor.h
##########
@@ -115,35 +123,25 @@ class TVM_DLL PipelineExecutor : public ModuleNode {
/*!\brief The class used to execute and schedule the pipeline logic.*/
PipelineScheduler pipeline_scheduler_;
/*!\brief The dependency information of each graph runtime module of the
pipeline.*/
- PipelineConfig pipeline_config_;
+ ConfigPipelineExecution pipeline_config_;
+ /*!\brief The map of global input and subgraph input.*/
+ InputConnectionConfig input_connection_config;
/*!\brief The module information used to create the graph runtimes.*/
ModuleConfig mod_config_;
/*!\brief How many outputs are in this pipeline executor.*/
size_t num_outputs_ = 0;
/*!\brief Json loader.*/
- PipelineConfig& LoadPipelineConfig(dmlc::JSONReader* reader) {
- reader->BeginArray();
- while (reader->NextArrayItem()) {
- std::string key;
- reader->BeginObject();
- int mod_idx = -1;
- OutputMap output;
- std::string dev;
- while (reader->NextObjectItem(&key)) {
- if (key == "mod_idx") {
- reader->Read(&mod_idx);
- } else if (key == "dev") {
- reader->Read(&dev);
- } else if (key == "output") {
- reader->Read(&output);
- } else {
- LOG(FATAL) << "do not support key " << key;
- }
+ ConfigPipelineExecution& LoadConfigPipelineExecution(dmlc::JSONReader*
reader) {
Review comment:
changed the name into "LoadConfig" to avoid confuse.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
