search

uwsampl / SparseTIR

SparseTIR: Sparse Tensor Compiler for Deep Learning
https://sampl.cs.washington.edu/SparseTIR/
Apache License 2.0
131 stars 14 forks source link

[Tracking Issue] Note on passes reorder issues #28

Closed yzh119 closed 2 years ago

yzh119 commented 2 years ago
  1. Storage Rewrite pass was moved to mixed pass list because some memory allocations would be created after lowering allreduce.
  2. Infer Fragment was moved after StorageRewrite because InferFragment insert StmtAttr for wmma allocations, which would not be moved along with allocation statements in StorageRewrite pass.
yzh119 commented 2 years ago

This issue was created for reminding me.