search

apache / tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators
https://tvm.apache.org/
Apache License 2.0
11.46k stars 3.41k forks source link

[TIR] Fix Shuffle rewrite #17030

Closed Hzfengsy closed 2 months ago

Hzfengsy commented 2 months ago

This PR fixes the shuffle rewrite pass to handle the case where the vector lanes are larger than the data type of the input vector.