Please build release wheels for arch 8.6, 8.9 and arch 9.0
Closed this issue · 0 comments
jon-chuang commented
It may slow down wheel build but it should be better perf for newer architectures.
current: TORCH_CUDA_ARCH_LIST: "6.0 6.1 7.0 7.5 8.0+PTX"