jinhongyii opened a new pull request, #16093: URL: https://github.com/apache/tvm/pull/16093
In DistIR compilation flow, shard loading is implemented in a Relax func, like ``` a = LoadWholeParamOnWorker0(loader, index=0) b = broadcast(a) c = LoadWholeParamOnWorker0(loader, index=1) d = scatter_from_worker0(c) ``` LoadWholeParamOnWorker0 loads the unsharded param on worker0, and for other workers returns an empty array. This PR implements LoadWholeParamOnWorker0 -- 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]
