ant-research / Pyraformer

Apache License 2.0
264 stars 41 forks source link

What should I do if I want to compile using tvm on cuda 12.1? #31

Open DaozeZhang opened 2 weeks ago

DaozeZhang commented 2 weeks ago

Thanks for this awesome work. I admit that I am new about the TVM, so I met some troubles. For some reasons about my server, I can only use cuda 12.1 version. I installed tvm from source code.

When I was trying to run the code, this error occurred: Check failed: (bf != nullptr) is false: target.build.cuda is not enabled

From these two refs (1, 2), there are two ways to solve this problem. However, they all need a lower cuda version like cuda 11.6 or 11.8

So could you please tell me how can i run the code on cuda 12.1 ?