Sven-Hendrik Haase pushed to branch main at Arch Linux / Packaging / Packages /
python-pytorch
Commits:
bdd43389 by Sven-Hendrik Haase at 2024-11-15T05:41:08+01:00
upgpkg: 2.5.1-4: Compile opt for x86-64-v3 instead of haswell
Fixes #15.
- - - - -
2 changed files:
- .SRCINFO
- PKGBUILD
Changes:
=====================================
.SRCINFO
=====================================
@@ -1,7 +1,7 @@
pkgbase = python-pytorch
pkgdesc = Tensors and Dynamic neural networks in Python with strong GPU
acceleration
pkgver = 2.5.1
- pkgrel = 3
+ pkgrel = 4
url = https://pytorch.org
arch = x86_64
license = BSD
=====================================
PKGBUILD
=====================================
@@ -8,7 +8,7 @@ pkgname=("${pkgbase}" "${pkgbase}-opt" "${pkgbase}-cuda"
"${pkgbase}-opt-cuda" "
# When updating pytorch, also check the compatibility table for torchvision
# https://github.com/pytorch/vision?tab=readme-ov-file#installation
pkgver=2.5.1
-pkgrel=3
+pkgrel=4
_pkgdesc='Tensors and Dynamic neural networks in Python with strong GPU
acceleration'
pkgdesc="${_pkgdesc}"
arch=('x86_64')
@@ -285,7 +285,7 @@ build() {
export USE_CUDA=0
export USE_CUDNN=0
export USE_ROCM=0
- echo "add_definitions(-march=haswell)" >> cmake/MiscCheck.cmake
+ echo "add_definitions(-march=x86-64-v3)" >> cmake/MiscCheck.cmake
# same horrible hack as above
python setup.py build || python setup.py build
@@ -308,7 +308,7 @@ build() {
export USE_ROCM=0
export MAGMA_HOME=/opt/cuda/targets/x86_64-linux
_prepare
- echo "add_definitions(-march=haswell)" >> cmake/MiscCheck.cmake
+ echo "add_definitions(-march=x86-64-v3)" >> cmake/MiscCheck.cmake
# same horrible hack as above
python setup.py build || python setup.py build
@@ -336,7 +336,7 @@ build() {
export USE_CUDNN=0
export USE_ROCM=1
export MAGMA_HOME=/opt/rocm
- echo "add_definitions(-march=haswell)" >> cmake/MiscCheck.cmake
+ echo "add_definitions(-march=x86-64-v3)" >> cmake/MiscCheck.cmake
# Conversion of CUDA to ROCm source files
python tools/amd_build/build_amd.py
patch -Np1 -i "$srcdir/pytorch-rocm-jit.patch"
View it on GitLab:
https://gitlab.archlinux.org/archlinux/packaging/packages/python-pytorch/-/commit/bdd43389e91491fe7e5249fb85f64a3ebed2dfd3
--
View it on GitLab:
https://gitlab.archlinux.org/archlinux/packaging/packages/python-pytorch/-/commit/bdd43389e91491fe7e5249fb85f64a3ebed2dfd3
You're receiving this email because of your account on gitlab.archlinux.org.
