I am working on a case: compile a pytorch model using tvm on g4dn.
But I encountered a problem that the tensor conversion between tvm and pytorch is too slow, since some pre-process and post-process are implemented using pytorch. I found that DLPACK maybe helpful in this case, thus I tried the example here (https://tvm.apache.org/2018/08/10/DLPack-Bridge). The example works well, but when it comes to my pytorch model, I got the error when calling the converted model: [AttributeError: Module has no function ‘__tvm_main__’]. Can anybody help me on this case? Thank you! --- [Visit Topic](https://discuss.tvm.apache.org/t/dlpack-error-attributeerror-module-has-no-function-tvm-main/8029/1) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.tvm.apache.org/email/unsubscribe/a7694f71b182d6891b09a4b32d28e52b5386df72e15ee8178cb957621ddff372).
