summarylogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--.SRCINFO3
-rw-r--r--PKGBUILD6
2 files changed, 4 insertions, 5 deletions
diff --git a/.SRCINFO b/.SRCINFO
index 36e408e0ef74..d934a9bfd446 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,11 +1,10 @@
pkgbase = python-apex-git
pkgdesc = A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
- pkgver = 22.03.r180.g8b7a1ff1
+ pkgver = 22.03.r231.g141bbf1c
pkgrel = 1
url = https://github.com/NVIDIA/apex
arch = x86_64
license = BSD
- makedepends = gcc11
makedepends = git
makedepends = numactl
makedepends = python-setuptools
diff --git a/PKGBUILD b/PKGBUILD
index ef7cce301d74..c5a274c50ce2 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -2,7 +2,7 @@
pkgname=python-apex-git
_pkgname=apex
-pkgver=22.03.r180.g8b7a1ff1
+pkgver=22.03.r231.g141bbf1c
pkgrel=1
pkgdesc="A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch"
arch=('x86_64')
@@ -10,7 +10,7 @@ url="https://github.com/NVIDIA/apex"
_github='NVIDIA/apex'
license=('BSD')
depends=('python' 'cuda' 'python-pytorch-cuda' 'python-sympy')
-makedepends=('gcc11' 'git' 'numactl' 'python-setuptools' 'python-pip')
+makedepends=('git' 'numactl' 'python-setuptools' 'python-pip')
provides=('python-apex')
conflicts=('python-apex')
source=("git+$url")
@@ -23,7 +23,7 @@ pkgver () {
build() {
cd "$srcdir/${_pkgname}"
- CC=gcc-11 CXX=g++-11 \
+ CC=/opt/cuda/bin/gcc CXX=/opt/cuda/bin/g++ \
python setup.py build --cuda_ext --cpp_ext
}