punica
punica copied to clipboard
Support for H100 GPUs?
When I set TORCH_CUDA_ARCH_LIST="8.0 8.6 8.9 9.0"
, I got compiling errors. And then I found: https://github.com/punica-ai/punica/blob/591b59899f0a20760821785d06b331c8a2e5cb86/.github/workflows/release_wheel.yml#L15
Is there something we does not support yet? Thank you in advance!
Update: Adding +PTX
works. But it would be great to have 9.0 supported in the future.