punica icon indicating copy to clipboard operation
punica copied to clipboard

Support for H100 GPUs?

Open LorrinWWW opened this issue 11 months ago • 1 comments

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.

LorrinWWW avatar Feb 29 '24 18:02 LorrinWWW