[BUG] does not support nvidia A30
zhaohb opened this issue · 2 comments
zhaohb commented
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?
xysmlx commented
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.
puddingfjz commented
Hi, @xysmlx,
If I want to try Roller on A100, which part of the code should I modify?