[arch-commits] Commit in python-pytorch/trunk (PKGBUILD)

[Konstantin Gizdov]

    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}"
 }