diff options
-rw-r--r-- | .SRCINFO | 17 | ||||
-rw-r--r-- | PKGBUILD | 26 |
2 files changed, 24 insertions, 19 deletions
@@ -1,9 +1,9 @@ # Generated by mksrcinfo v8 -# Fri Jul 21 01:53:55 UTC 2017 +# Sun Aug 6 15:40:38 UTC 2017 pkgbase = caffe2 pkgdesc = A new lightweight, modular, and scalable deep learning framework (gpu enabled) - pkgver = 0.7.0 - pkgrel = 13 + pkgver = 0.8.0 + pkgrel = 1 url = http://caffe2.ai/ arch = x86_64 license = BSD @@ -19,7 +19,8 @@ pkgbase = caffe2 depends = python2-numpy depends = python2-protobuf depends = cuda - depends = cudnn + depends = cudnn>=5.1 + depends = cudnn<7.0 depends = gflags depends = gtest depends = openmp @@ -57,16 +58,16 @@ pkgbase = caffe2 noextract = thirdparty-eigen-3.3.2.tar.gz noextract = thirdparty-protobuf-3.1.0.tar.gz options = !emptydirs - source = caffe2-0.7.0.tar.gz::https://github.com/caffe2/caffe2/archive/v0.7.0.tar.gz + source = caffe2-0.8.0.tar.gz::https://github.com/caffe2/caffe2/archive/v0.8.0.tar.gz source = external-nnpack-fix.patch source = thirdparty-android-cmake-git::git+https://github.com/taka-no-me/android-cmake.git#commit=556cc14296c226f753a3778d99d8b60778b7df4f source = thirdparty-benchmark-git::git+https://github.com/google/benchmark.git#commit=4bf28e611b55de8a2d4eece3c335e014f8b0f630 source = thirdparty-cnmem-git::git+https://github.com/NVIDIA/cnmem.git#commit=28a182d49529da49f4ac4e3941cec3edf16b3540 source = thirdparty-cub-git::git+https://github.com/NVlabs/cub.git#commit=89de7ab20167909bc2c4f8acd397671c47cf3c0d source = thirdparty-eigen-3.3.2.tar.gz::https://github.com/RLovelett/eigen/archive/3.3.2.tar.gz - source = thirdparty-gloo-git::git+https://github.com/facebookincubator/gloo.git#commit=f67ab32d3439061946ce2c1bed8baa0e1b5d367f + source = thirdparty-gloo-git::git+https://github.com/facebookincubator/gloo.git#commit=21a5c8ea5e02edca03068790df3d7f7ba4e2d75b source = thirdparty-googletest-git::git+https://github.com/google/googletest.git#commit=5e7fd50e17b6edf1cadff973d0ec68966cf3265e - source = thirdparty-ios-cmake-git::git+https://github.com/Yangqing/ios-cmake.git#commit=e3a7695d1d68ef4eca716031f94e8475b1589b1f + source = thirdparty-ios-cmake-git::git+https://github.com/Yangqing/ios-cmake.git#commit=e24081928d9ceec4f4adfcf12293f1e2a20eaedc source = thirdparty-nccl-git::git+https://github.com/NVIDIA/nccl.git#commit=2a974f5ca2aa12b178046b2206b43f1fd69d9fae source = thirdparty-nervanagpu-git::git+https://github.com/NervanaSystems/nervanagpu.git#commit=d4eefd50fbd7d34a17dddbc829888835d67b5f4a source = thirdparty-NNPACK-git::git+https://github.com/Maratyszcza/NNPACK.git#commit=02bfa475d64040cd72b7c01daa9e862523ae87da @@ -76,7 +77,7 @@ pkgbase = caffe2 source = thirdparty-NNPACK_deps-FXdiv-git::git+https://github.com/Maratyszcza/FXdiv.git#commit=8f85044fb41e560508cd69ed26c9afb9cc120e8a source = thirdparty-NNPACK_deps-psimd-git::git+https://github.com/Maratyszcza/psimd.git#commit=0b26a3fb98dd6af7e1f4e0796c56df6b32b1c016 source = thirdparty-NNPACK_deps-pthreadpool-git::git+https://github.com/Maratyszcza/pthreadpool.git#commit=9e17903a3fc963fe86b151aaddae7cf1b1d34815 - sha256sums = b8f266ed283efc172fa96c06c878ed7f125755f89cde480580b754c1f03c0bab + sha256sums = 3c46d2324046774e96601bad737cdce05084cab80b9109c48a09f332438b921c sha256sums = 1c94a1ecc0fe2a52c50c9d1a7bfca655d60c08d48995b27c73aa22b6c375da54 sha256sums = SKIP sha256sums = SKIP @@ -1,13 +1,13 @@ # Maintainer : Daniel Bermond < yahoo-com: danielbermond > -# third_party with no stable release at the needed commit +# third_party with no stable release at the target commit _android_cmake_commit='556cc14296c226f753a3778d99d8b60778b7df4f' _benchmark_commit='4bf28e611b55de8a2d4eece3c335e014f8b0f630' _cnmem_commit='28a182d49529da49f4ac4e3941cec3edf16b3540' _cub_commit='89de7ab20167909bc2c4f8acd397671c47cf3c0d' -_gloo_commit='f67ab32d3439061946ce2c1bed8baa0e1b5d367f' +_gloo_commit='21a5c8ea5e02edca03068790df3d7f7ba4e2d75b' _googletest_commit='5e7fd50e17b6edf1cadff973d0ec68966cf3265e' -_ios_cmake_commit='e3a7695d1d68ef4eca716031f94e8475b1589b1f' +_ios_cmake_commit='e24081928d9ceec4f4adfcf12293f1e2a20eaedc' _nccl_commit='2a974f5ca2aa12b178046b2206b43f1fd69d9fae' _nervanagpu_commit='d4eefd50fbd7d34a17dddbc829888835d67b5f4a' _nnpack_commit='02bfa475d64040cd72b7c01daa9e862523ae87da' @@ -17,13 +17,19 @@ _nnpackdeps_fxdiv_commit='8f85044fb41e560508cd69ed26c9afb9cc120e8a' _nnpackdeps_psimd_commit='0b26a3fb98dd6af7e1f4e0796c56df6b32b1c016' _nnpackdeps_pthreadpool_commit='9e17903a3fc963fe86b151aaddae7cf1b1d34815' - # third_party with stable release at the needed commit + # third_party with stable release at the target commit _eigen_version='3.3.2' # commit 'ae9889a130bd0a9d3007f41d015563c2e8ac605f' is version '3.3.2' _protobuf_version='3.1.0' # commit 'a428e42072765993ff674fda72863c9f1aa2d268' is version '3.1.0' +# note about modified gloo commit: +# the gloo target commit is 7ea9d9af4e82d20c7c6cee5edd3c52f9bcb42821, but +# it's necessary to use a newer commit to avoid compile errors if using +# '-march=native' in a cpu with AVX2: +# https://github.com/facebookincubator/gloo/issues/43 + pkgname=caffe2 -pkgver=0.7.0 -pkgrel=13 +pkgver=0.8.0 +pkgrel=1 pkgdesc='A new lightweight, modular, and scalable deep learning framework (gpu enabled)' arch=('x86_64') url='http://caffe2.ai/' @@ -32,7 +38,7 @@ depends=( # official repositories: # required: 'google-glog' 'protobuf' 'python2' 'python2-numpy' 'python2-protobuf' 'cuda' - 'cudnn' + 'cudnn>=5.1' 'cudnn<7.0' # not required but enabled in build: 'gflags' 'gtest' 'openmp' 'leveldb' 'lmdb' 'openmpi' 'snappy' 'zeromq' 'hiredis' @@ -83,7 +89,7 @@ source=( ) noextract=("thirdparty-eigen-${_eigen_version}.tar.gz" "thirdparty-protobuf-${_protobuf_version}.tar.gz") -sha256sums=('b8f266ed283efc172fa96c06c878ed7f125755f89cde480580b754c1f03c0bab' +sha256sums=('3c46d2324046774e96601bad737cdce05084cab80b9109c48a09f332438b921c' '1c94a1ecc0fe2a52c50c9d1a7bfca655d60c08d48995b27c73aa22b6c375da54' 'SKIP' 'SKIP' @@ -135,7 +141,7 @@ prepare() { unset _component # avoid compile errors with nnpack if system library is found - # https://github.com/caffe2/caffe2/pull/808 + # https://github.com/caffe2/caffe2/commit/f070eb99a4a36f12abbdb9af98bb68ff485c840b cd "${srcdir}/${pkgname}-${pkgver}" patch -Np1 -i "${srcdir}/external-nnpack-fix.patch" } @@ -181,7 +187,6 @@ build() { -DCUDA_TOOLKIT_ROOT_DIR:PATH='/opt/cuda' \ -DCUDA_USE_STATIC_CUDA_RUNTIME:BOOL='OFF' \ -DCUDA_VERBOSE_BUILD:BOOL='OFF' \ - -DCUDA_VERSION:STRING='8.0' \ -DCUDNN_INCLUDE_DIR:PATH='/opt/cuda/include' \ -DCUDNN_ROOT_DIR:PATH='/opt/cuda' \ \ @@ -236,7 +241,6 @@ package() { mkdir -p "${pkgdir}/usr/share/licenses/${pkgname}" # move/rename folders to the right location - mv -f "${pkgdir}/usr/binaries" "${pkgdir}/usr/bin" mv -f "${pkgdir}/usr/caffe" "${pkgdir}/usr/lib/python2.7/site-packages" mv -f "${pkgdir}/usr/caffe2" "${pkgdir}/usr/lib/python2.7/site-packages" |