areusch commented on a change in pull request #8019:
URL: https://github.com/apache/tvm/pull/8019#discussion_r631978186
##########
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:
okay, i think that makes sense to me. i do wonder if it makes sense to
upstream this given we're just going to change it later, but given the target
parameter does not have a default, it's probably fine to do this as an interim
solution until we address the broader memory planning challenges.
--
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]
