It is actually possible to get multiple versions of function compiled together. Just need a bit of extra effort to do so.
One way to achieve that now is to call tvm.lower(instead of buld) to get List[LoweredFuncs] for each of the function you care about(give them different names), then concat these lists together to get one list, then feed it to tvm.build to get you a single runtime module which which will contain all the functions you need. Then we can call mod.export_library to export a single so file that contains all the function variants. This is the approach we use to export multiple functions in a neural networks into a single shared library Also we might have another way(by importing other modules into a single one) which @FrozenGene recently enabled. --- [Visit Topic](https://discuss.tvm.ai/t/optimizing-matrix-multiplication-for-gpu/4212/21) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.tvm.ai/email/unsubscribe/2116b6406a5483635761368b84b1546a3b00a9aab3265df180ce18c112f8b5c3).