apache / tvm

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

[Bugfix][Relax] Set purity=false for LazySetOutput #17119

Closed Lunderberg closed 1 week ago

Lunderberg commented 1 week ago

The relax.transform.LazySetOutput transformation updates a Relax function to produce output from a fset_output callback. In the initial implementation, the fset_output was marked as a pure function, which allowed it to be erroneously removed from a function.

This commit updates the relax::FuncStructInfo used to annotate fset_output, marking it as an impure function.