huajsj commented on a change in pull request #9108:
URL: https://github.com/apache/tvm/pull/9108#discussion_r718043485
##########
File path: tests/python/relay/test_pipeline_executor.py
##########
@@ -231,9 +231,19 @@ def test_pipeline():
with tvm.transform.PassContext(opt_level=3):
pipeline_mod_factory = pipeline_executor.build(pipe_config)
+ # Export parameter configuration to file.
+ config_file_name = pipeline_mod_factory.export_library()
+
+ # The test is to create and initialized class PipelineModule using
build return.
Review comment:
fixed.
##########
File path: tests/python/relay/test_pipeline_executor.py
##########
@@ -231,9 +231,19 @@ def test_pipeline():
with tvm.transform.PassContext(opt_level=3):
pipeline_mod_factory = pipeline_executor.build(pipe_config)
+ # Export parameter configuration to file.
+ config_file_name = pipeline_mod_factory.export_library()
+
+ # The test is to create and initialized class PipelineModule using
build return.
pipeline_module =
pipeline_executor.PipelineModule(pipeline_mod_factory)
assert pipeline_module
+ # The test is to initialized class PipelineModule using import
function.
Review comment:
fixed.
--
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]
