[Primitive] New schedule primitive: `reorder_block_itervar`

uwsampl / SparseTIR

SparseTIR: Sparse Tensor Compiler for Deep Learning

https://sampl.cs.washington.edu/SparseTIR/

Apache License 2.0

131 stars 13 forks source link

[Primitive] New schedule primitive: `reorder_block_itervar` #89

Closed yzh119 closed 1 year ago

yzh119 commented 1 year ago

Currently there is no schedule primitive to reorder itervars inside blocks (reorder primitive only manipulate loop orders), this PR solves the issue.

This PR would be upstream to TVM later.