@tqchen, thanks! This is exactly what we are expecting. However, last time I tried to bring my own tuner into mlc-llm
, I encountered an issue:
import tvm # upstream
relax_mod = relax_transform(relax_mod)
import welder
relax_mod = welder.tune(relax_mod)
# something bad happened
The problem was that when welder
is imported, it also imports in its own version of TVM, which then invokes load_dlls
(for example, to load libcutlass
). This process ends up overwriting the upstream cutlass lib and lead to some bugs.