Date: Friday, June 25, 2021 @ 18:54:46 Author: kgizdov Revision: 967450
upgpkg: python-pytorch 1.9.0-2: add latest PTX version for future compat Modified: python-pytorch/trunk/PKGBUILD ----------+ PKGBUILD | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) Modified: PKGBUILD =================================================================== --- PKGBUILD 2021-06-25 16:58:49 UTC (rev 967449) +++ PKGBUILD 2021-06-25 18:54:46 UTC (rev 967450) @@ -6,7 +6,7 @@ _pkgname="pytorch" pkgver=1.9.0 _pkgver=1.9.0 -pkgrel=1 +pkgrel=2 pkgdesc="Tensors and Dynamic neural networks in Python with strong GPU acceleration" arch=('x86_64') url="https://pytorch.org" @@ -197,7 +197,7 @@ export CUDNN_LIB_DIR=/usr/lib export CUDNN_INCLUDE_DIR=/usr/include export TORCH_NVCC_FLAGS="-Xfatbin -compress-all" - export TORCH_CUDA_ARCH_LIST="5.2;5.3;6.0;6.1;6.2;7.0;7.2;7.5;8.0;8.6" + export TORCH_CUDA_ARCH_LIST="5.2;5.3;6.0;6.1;6.2;7.0;7.2;7.5;8.0;8.6;8.6+PTX" #include latest PTX for future compat export OVERRIDE_TORCH_CUDA_ARCH_LIST="${TORCH_CUDA_ARCH_LIST}" }