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

[Bug] Cannot lower cross thread reduction after emit binary search block #16

Closed yzh119 closed 2 years ago

yzh119 commented 2 years ago

Error message:

  File "/home/expye/repos/tvm/src/tir/transforms/lower_cross_thread_reduction.cc", line 476
ValueError: Check failed: n_deepest_reduction_loops == reduction_loops.size() (0 vs. 1) : Cross-thread reduction requires all the reduction-related loops to be the deepest among all statements outside the desired block. However, block csrmm0 needs cross-thread reduction, while the reduction-related loops outside of it are not the deepest statements, which violates the condition.
yzh119 commented 2 years ago

Solution, make binary search outermost scope #21 .