Can TVM compile a given Python function using a decorator syntax, like this
@tvm.jit
def f(a, b, c):
return a @ b + c
does something like this exist? Or in general how do I compile a Python function? There’s no DNN model here, just a Python function. Thanks!
TVM is not designed to compile a generic python program. Matxscript, which is built with similar ideas of TVM’s PackedFunc system, broadly expand the territory of what python program it could compile, and perhaps it’s worth a try: https://github.com/bytedance/matxscript
That’s very helpful! Thanks for the suggestions!