comaniac commented on a change in pull request #9846:
URL: https://github.com/apache/tvm/pull/9846#discussion_r784264593
##########
File path: python/tvm/contrib/pipeline_executor.py
##########
@@ -125,6 +130,39 @@ def get_input_pipeline_map(self, name):
"""
return self._get_input_pipeline_map(name)
+ def get_params_group_pipeline_map(self, name):
+ """Use the name of the parameters group to get the corresponding
runtime module index.
+
+ Parameters
+ ----------
+ name: str
+ The parameter group name.
+
+ Returns
+ -------
+ module_index: int
+ The index of the runtime module.
+ """
+ return self._get_params_group_pipeline_map(name)
+
+ def set_params(self, params_group_name, params_data):
+ """Pipeline executor uses a map of parameters groups and module
indexes to support
+ the "on the fly" parameters setting. For example, users can bind the
parameter group
+ "param0" to the parameters of the 1st module. This function is used to
set parameters
+ into to correct runtime module by using the parameters group name.
+ ----------
Review comment:
```suggestion
"""Set the parameter group value given the parameter group name.
Note that the parameter group
name is declared in the pipeline executor config.
Parameters
----------
```
--
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]
