microsoft/nnfusion

[BUG] does not support nvidia A30

zhaohb opened this issue · 2 comments

I converted a model on the A30 card, but was stuck in cuda_init when main_test was executed.
It's stuck on:

...
 CUDNN_SAFE_CALL(cudnnCreate(&cudnn_handle_0))
...

how to fix it?

Hi, @zhaohb,

Thank you for your attention. You may need to add the compilation information (i.e., -gencode arch=compute_80,code=sm_80) to the compilation flags to support the A30 GPU, and re-build the NNFusion.

Hi, @xysmlx,

If I want to try Roller on A100, which part of the code should I modify?