Hello @hijiang,
I have followed your work and made some extensions to support such splitting as I mentioned in the previous discussions https://discuss.tvm.apache.org/t/setting-the-cpu-affinity-and-number-of-cores-locally-without-rpc-remote/11306/4?u=popojames. https://discuss.tvm.apache.org/t/can-tvm-split-work-into-different-layers-and-assign-layers-into-different-cores/11161/10?u=popojames. I was able to split the network and with my own CPU affinity setting. Now I try to assign **param** of the pre-trained model into the pipeline module, I noticed the function of PipelineModule.set_input support such setting: https://github.com/apache/tvm/blob/fba531e55f032367acc28090605c5533b2e88f55/python/tvm/contrib/pipeline_executor.py#L222 However, this function will induce the following error: `AttributeError: 'Module' object has no attribute 'set_input'` This is because of the type of **self.graph_modules_[mod_idx]** is something like **Module(GraphExecutor, 27517208)** instead of **tvm.contrib.graph_executor.GraphModule**. In this case, the latter can access set_input but the former cannot. Is there any other way to assign parameters or can I transfer from Module(GraphExecutor, XXX) to tvm.contrib.graph_executor.GraphModule in this case to set the parameters? Thanks for your help in advance and many thanks for developing this TVM function. --- [Visit Topic](https://discuss.tvm.apache.org/t/how-to-set-parameter-into-pipeline-module/11375/1) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.tvm.apache.org/email/unsubscribe/72ba1c0886e2a25d30b8d9e20513905384583a7928c4c2b7cbe102f84022b569).
