Skip to content

Conversation

@jinhongyii
Copy link
Contributor

Currently the transform_params function lifted by LiftTransformParams will first load all inputs into memory, and transform. That may exceed memory limit. The better way is to load the input to memory on demand, and immediately free it after the last use. This PR implements such method. It replaces all params[i] with get_item(i) and inserts set_item(i) and kill_object after the last use except in return stmt.

cc: @tqchen @MasterJH5574

@tvm-bot
Copy link
Collaborator

tvm-bot commented May 4, 2023

Thanks for contributing to TVM! Please refer to the contributing guidelines https://tvm.apache.org/docs/contribute/ for useful information and tips. Please request code reviews from Reviewers by @-ing them in a comment.

Generated by tvm-bot

@tqchen tqchen merged commit 98a25df into apache:unity May 7, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants