[TIR] Fix Bug in VectorizeLoop

apache / tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators

https://tvm.apache.org/

Apache License 2.0

11.42k stars 3.4k forks source link

Closed CharlieFRuan closed 1 month ago

CharlieFRuan commented 1 month ago

This PR fixes a bug in vectorize loop introduced related to https://github.com/apache/tvm/pull/16966

The visit to condition can write need scalarize to true then the followup visit to then case can trigger an ICHECK.

The visit to let value can also write need scalarize flag in which case we need to immediately scalarize.

tqchen commented 1 month ago

@tvm-bot rerun

tqchen commented 1 month ago

@tvm-bot rerun

ekalda commented 1 month ago

Thanks @tqchen and @CharlieFRuan for the fix!