diff options
Diffstat (limited to 'PKGBUILD')
-rw-r--r-- | PKGBUILD | 37 |
1 files changed, 27 insertions, 10 deletions
@@ -7,20 +7,28 @@ pkgname=("python-pytorch-rocm" "python-pytorch-opt-rocm") _pkgname="pytorch" pkgver=1.7.0 _pkgver=1.7.0 -pkgrel=1 +pkgrel=2 pkgdesc="Tensors and Dynamic neural networks in Python with strong GPU acceleration" arch=('x86_64') url="https://pytorch.org" license=('BSD') depends=('google-glog' 'gflags' 'opencv' 'openmp' 'rccl' 'pybind11' 'python' 'python-yaml' 'libuv' 'python-numpy' 'protobuf' 'ffmpeg' 'python-future' 'qt5-base' 'onednn' 'intel-mkl') -makedepends=('python' 'python-setuptools' 'python-yaml' 'python-numpy' 'cmake' 'rocm' 'rocm-libs' 'miopen' - 'git' 'magma' 'ninja' 'pkgconfig' 'doxygen') +makedepends=('python' 'python-setuptools' 'python-yaml' 'python-numpy' 'cmake' 'rocm' + 'rocm-libs' 'miopen' 'git' 'magma' 'ninja' 'pkgconfig' 'doxygen') source=("${_pkgname}-${pkgver}::git+https://github.com/pytorch/pytorch.git#tag=v$_pkgver" fix_include_system.patch + use-system-libuv.patch + use-system-libuv2.patch + nccl_version.patch + disable_non_x86_64.patch "find-hsa-runtime.patch::https://patch-diff.githubusercontent.com/raw/pytorch/pytorch/pull/45550.patch") sha256sums=('SKIP' '147bdaeac8ec46ea46382e6146878bd8f8d51e05d5bd6f930dfd8e2b520859b9' + '26b1dd596f1e21a011ee18cab939924483d6c6d4d98e543bf76f5a9312d54d67' + '7b65c3b209fc39f92ba58a58be6d3da40799f1922910b1171ccd9209eda1f9eb' + 'e4a96887b41cbdfd4204ce5f16fcb16a23558d23126331794ab6aa30a66f2e0d' + 'd3ef8491718ed7e814fe63e81df2f49862fffbea891d2babbcb464796a1bd680' 'SKIP') prepare() { @@ -36,6 +44,12 @@ prepare() { # https://bugs.archlinux.org/task/64981 patch -N torch/utils/cpp_extension.py "${srcdir}"/fix_include_system.patch + # Use system libuv + patch -Np1 -i "${srcdir}"/use-system-libuv.patch + + # FindNCCL patch to export correct nccl version + # patch -Np1 -i "${srcdir}"/nccl_version.patch + # https://github.com/pytorch/pytorch/pull/45550 patch -Np1 -i "${srcdir}"/find-hsa-runtime.patch @@ -53,22 +67,23 @@ prepare() { # Check tools/setup_helpers/cmake.py, setup.py and CMakeLists.txt for a list of flags that can be set via env vars. export USE_MKLDNN=ON - # export BUILD_CUSTOM_PROTOBUF=OFF + export BUILD_CUSTOM_PROTOBUF=OFF # export BUILD_SHARED_LIBS=OFF export USE_FFMPEG=ON export USE_GFLAGS=ON export USE_GLOG=ON export BUILD_BINARY=ON - #export USE_OPENCV=ON + # export USE_OPENCV=ON export USE_SYSTEM_NCCL=ON + # export USE_SYSTEM_LIBS=ON export NCCL_VERSION=$(pkg-config nccl --modversion) export NCCL_VER_CODE=$(sed -n 's/^#define NCCL_VERSION_CODE\s*\(.*\).*/\1/p' /usr/include/nccl.h) - export CUDAHOSTCXX=g++-9 + export CUDAHOSTCXX=g++ export CUDA_HOME=/opt/cuda 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.0+PTX;6.1;6.1+PTX;6.2;6.2+PTX;7.0;7.0+PTX;7.2;7.2+PTX;7.5;7.5+PTX;8.0;8.0+PTX;" + # 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.0+PTX;7.2;7.2+PTX;7.5;7.5+PTX;8.0;8.0+PTX;8.6;8.6+PTX" } build() { @@ -76,6 +91,8 @@ build() { export USE_CUDA=OFF export USE_ROCM=ON cd "${srcdir}/${_pkgname}-${pkgver}-rocm" + patch -Np1 -i "${srcdir}/disable_non_x86_64.patch" + echo "add_definitions(-march=x86-64)" >> cmake/MiscCheck.cmake # Apply changes needed for ROCm python tools/amd_build/build_amd.py @@ -128,7 +145,7 @@ _package() { package_python-pytorch-rocm() { pkgdesc="Tensors and Dynamic neural networks in Python with strong GPU acceleration (with ROCM)" - depends+=(rocm magma) + depends+=(rocm rocm-libs miopen magma) conflicts=(python-pytorch) provides=(python-pytorch) @@ -138,7 +155,7 @@ package_python-pytorch-rocm() { package_python-pytorch-opt-rocm() { pkgdesc="Tensors and Dynamic neural networks in Python with strong GPU acceleration (with ROCM and CPU optimizations)" - depends+=(rocm magma) + depends+=(rocm rocm-libs miopen magma) conflicts=(python-pytorch) provides=(python-pytorch python-pytorch-rocm) |