manupa-arm commented on a change in pull request #8019:
URL: https://github.com/apache/tvm/pull/8019#discussion_r631995595
##########
File path: src/tir/analysis/calculate_workspace.cc
##########
@@ -46,6 +48,10 @@ size_t WorkspaceCalculator::operator()(const PrimFunc& func)
{
return this->max_size;
}
+size_t WorkspaceCalculator::GetByteAlignedSize(size_t non_aligned_size) {
+ return ((non_aligned_size + byte_alignment - 1) / byte_alignment) *
byte_alignment;
Review comment:
Well; I've made sure it defaults to 1 wherever its accessed, so should
not affect anything else if its not specified.
This enables us to see where we are in-terms of memory usage (by actually
running models using the calculated value) -- that motivates the unified memory
planning work even more.
--
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.
For queries about this service, please contact Infrastructure at:
[email protected]
