Lunderberg opened a new pull request #10816: URL: https://github.com/apache/tvm/pull/10816
Follow-up from https://github.com/apache/tvm/pull/9727. When using `Stage.tensorize`, the element offset of the array view has one offset for each physical dimension of the buffer. This change replaces `BufferNode::elem_offset` and `BufferNode::offset_factor` with arrays. The commits in this PR are separated by category, to help in review. Updates to Buffer class, to handle multiple elem_offsets - Includes size checks to ensure that each parameter that depends on the number of physical dimensions (`elem_offsets`, `offset_factors`, and `axis_separators`) are consistent. - Support previous Buffer constructor with a single `elem_offset` and `offset_factor`, delegating to the array-based constructor. Update python calls, with backwards compatibility - `Buffer.elem_offset` and `Buffer.offset_factor` are implemented as properties, to avoid breakage of calling code. - `decl_buffer` handles single or multiple `elem_offset` and `offset_factor` parameters. The parameter name is unchanged, to avoid breaking call sites that use named arguments. Updated TVMScript parser and printer Other C++ changes - Several other locations that referenced either `elem_offsets` or `offset_factors`, which required only minor updates to maintain existing behavior. -- 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]
