Lunderberg opened a new pull request, #15563: URL: https://github.com/apache/tvm/pull/15563
Prior to this commit, any symbolic variables found in the transform_params function that are not provided by input tensors result in undefined symbolic variables. This commit updates LazyTransformParam to only delegate tensor parameters to `"get_item"` and `"set_item"`. This can be useful when producing sharded transformed parameters in a multi-GPU setup, where a `rank` symbolic variable may be provided to specify a subset of the weights to be used. -- 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]
