Skip to content

SPI-2704: Add A100 support and align CUDA architectures across PyTorch-related packages.

Dmitri Konstantinov requested to merge pytorch-arch-80 into master
  • configure TORCH_CUDA_ARCH_LIST using the environment variable common_cuda_arch for consistency.
  • add support for 8.0 (A100) to the list of CUDA architectures.
  • increment the REVISION to include these changes.

Merge request reports