wrongtest opened a new pull request, #11308: URL: https://github.com/apache/tvm/pull/11308
Hi there, the PR propose to represent `tir.RampNode` in buffer access index just as python slice. I believe this could be a more user friendly way for script users on SIMD-based target, since it just mimic usages in various python numerical libraries like numpy. For example: `A[T.ramp(0, 1, 1024)]` => `A[0:1024]` `A[T.ramp(i*32, 1, 32)]` => `A[i*32:i*32+32]` `A[T.ramp(i*4, 4, 32)]` => `A[i*4:i*4+128:4]` The syntax sugar is only available for constant stride and lanes. Also please correct me on two points: - TIR lowering actually do not support vectorized index in non-last axes, but it is not checked by current change. - `Slice` with non-trivial step is only allowed in buffer indices, we may want to ensure all other usages should check step==1. cc @junrushao1994 @Hzfengsy @Lunderberg -- 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]
