Fix architectures for CUDA 12 and 11.6

k2-fsa / k2

FSA/FST algorithms, differentiable, with PyTorch compatibility.

https://k2-fsa.github.io/k2

Apache License 2.0

1.1k stars 214 forks source link

Fix architectures for CUDA 12 and 11.6 #1200

Closed artbataev closed 1 year ago

artbataev commented 1 year ago

add support for Hopper (90) architecture for CUDA 11.6+
remove compute architecture 35 for CUDA 12.0+ (no longer supported)

Fixes issue #1199

csukuangfj commented 1 year ago

Thanks for your contribution!

MartinKocour commented 1 year ago

We have an issue with this update in our cluster... is Hopper really supported with CUDA 11.6? Shouldn't it be from 11.8? https://docs.nvidia.com/cuda/hopper-compatibility-guide/index.html

artbataev commented 1 year ago

We have an issue with this update in our cluster... is Hopper really supported with CUDA 11.6? Shouldn't it be from 11.8? https://docs.nvidia.com/cuda/hopper-compatibility-guide/index.html

I'm sorry, it seems to be my mistake, CUDA should be "11.8" for Hopper architecture. I will provide a fix. @MartinKocour, Did you try to modify the version? Does it fully fix your problem?