diff options
author | Stephen Zhang | 2017-09-12 19:06:54 +0800 |
---|---|---|
committer | Stephen Zhang | 2017-09-13 00:52:36 +0800 |
commit | f0454a6fd997f856c6eb6cc1b3e99cf492f1f377 (patch) | |
tree | 431f9e96b9083d1924ac72ff85faedaa63f8bd38 /PKGBUILD | |
parent | 534517aed4bb2032f61a2d89e2459c15b72a58bd (diff) | |
download | aur-f0454a6fd997f856c6eb6cc1b3e99cf492f1f377.tar.gz |
add _BITS_FLOATN_H to make cuda comfortable, remove git version in setup.py and fix MKL/magma support
Diffstat (limited to 'PKGBUILD')
-rw-r--r-- | PKGBUILD | 21 |
1 files changed, 11 insertions, 10 deletions
@@ -4,7 +4,7 @@ pkgbase="python-pytorch" pkgname=("python-pytorch" "python2-pytorch") _pkgname="pytorch" pkgver=0.2.0 -pkgrel=2 +pkgrel=3 pkgdesc="Tensors and Dynamic neural networks in Python with strong GPU acceleration" arch=('x86_64') url="http://pytorch.org" @@ -14,17 +14,14 @@ makedepends=('python' 'python-setuptools' 'python2' 'python2-setuptools' 'gcc5' 'cmake') depends=('cuda' 'cudnn') source=("https://github.com/pytorch/pytorch/archive/v${pkgver}.tar.gz") -sha256sums=('22b30638536d20d387e1adff62aa4b9ddebd8ac7ab812a36c699d72df9f7f570') - +sha256sums=('b76d61aaa8fc18b928ca3c910c398687be08f5661d6615884c4faba3e8742a26') prepare() { cd "$srcdir/" cd "${_pkgname}-${pkgver}" - # Uncomment and modify these lines to enable Intel MKL support - #sed -i -e '59i-DINTEL_MKL_DIR="/opt/intel/mkl" \\' torch/lib/build_all.sh - #sed -i -e '86d' -e '89d' torch/lib/TH/cmake/FindMKL.cmake sed -i -e '144icp -r nccl gloo/third-party/' torch/lib/build_all.sh + sed -i -e '470,475d' setup.py cd .. cp -a "${_pkgname}-${pkgver}" "${_pkgname}-${pkgver}-py2" @@ -33,10 +30,12 @@ prepare() { build() { msg "Building Python 2" cd "$srcdir/${_pkgname}-${pkgver}-py2" + # Uncomment and modify the following line to enable Intel MKL and magma support + #CMAKE_PREFIX_PATH=/opt/intel/mkl/include:/opt/intel/mkl/lib/intel64:/opt/magma \ CC=gcc-5 \ CXX=g++-5 \ - CFLAGS="${CFLAGS/-fno-plt/}" \ - CXXFLAGS="${CXXFLAGS/-fno-plt/}" \ + CFLAGS="${CFLAGS/-fno-plt/} -D_BITS_FLOATN_H" \ + CXXFLAGS="${CFLAGS/-fno-plt/} -D_BITS_FLOATN_H" \ WITH_CUDA=1 \ CUDA_HOME=/opt/cuda \ WITH_CUDNN=1 \ @@ -46,10 +45,12 @@ build() { msg "Building Python 3" cd "$srcdir/${_pkgname}-${pkgver}" + # Uncomment and modify the following line to enable Intel MKL and magma support + #CMAKE_PREFIX_PATH=/opt/intel/mkl/include:/opt/intel/mkl/lib/intel64:/opt/magma \ CC=gcc-5 \ CXX=g++-5 \ - CFLAGS="${CFLAGS/-fno-plt/}" \ - CXXFLAGS="${CXXFLAGS/-fno-plt/}" \ + CFLAGS="${CFLAGS/-fno-plt/} -D_BITS_FLOATN_H" \ + CXXFLAGS="${CFLAGS/-fno-plt/} -D_BITS_FLOATN_H" \ WITH_CUDA=1 \ CUDA_HOME=/opt/cuda \ WITH_CUDNN=1 \ |