diff options
Diffstat (limited to 'PKGBUILD')
| -rw-r--r-- | PKGBUILD | 108 |
1 files changed, 54 insertions, 54 deletions
@@ -1,37 +1,40 @@ # Maintainer : Daniel Bermond < gmail-com: danielbermond > -_catch_commit='0a34cc201ef28bf25c88b0062f331369596cb7b7' _pybind11_commit='5c8746ff135abb390bf95944be593e895a586a50' _cub_commit='285aeebaa34b0e8a7670867a2e66c1a52d998d6a' -_eigen_commit='cafae68f33f7f41270b2e8c2dd181f510aa4d918' +_eigen_commit='f59336cee358f92b959de6a0daf07c4ab2318022' _googletest_commit='2fe3bd994b3189899d93f1d5a881e725e046fdc2' -_nervanagpu_commit='d4eefd50fbd7d34a17dddbc829888835d67b5f4a' _benchmark_commit='505be96ab23056580a3a2315abba048f4428b04e' _protobuf_commit='2761122b810fe8861004ae785cc3ab39f384d342' _ios_cmake_commit='8abaed637d56f1337d6e1d2c4026e25c1eade724' -_nnpack_commit='af40ea7d12702f8ae55aeb13701c09cad09334c3' -_gloo_commit='aa0d2e3f8aa4f9cee5ffa46070491cf1ed6aae70' -_pthreadpool_commit='2b06b31f6a315162348e1f3c24325eedaf6cc559' -_fxdiv_commit='811b482bcd9e8d98ad80c6c78d5302bb830184b0' -_fp16_commit='43d6d17df48ebf622587e7ed9472ea76573799b9' -_psimd_commit='4ac61b112252778b174575931c641bef661ab3cd' +_nnpack_commit='1e005b0c2777f39972a4ac15bea03e0e315a3d92' +_gloo_commit='670b4d4aa46886cc66874e2a4dc846f5cfc2a285' +_pthreadpool_commit='13da0b4c21d17f94150713366420baaf1b5a46f4' +_fxdiv_commit='b742d1143724d646cd0f914646f1240eacf5bd73' +_fp16_commit='34d4bf01bbf7376f2baa71b8fa148b18524d45cf' +_psimd_commit='90a938f30ba414ada2f4b00674ee9631d7d85e19' _zstd_commit='aec56a52fbab207fc639a1937d1e708a282edca8' -_cpuinfo_commit='4e8f04355892c5deb64a51731a6afdb544a4294d' +_cpuinfo_commit='89fe1695edf9ee14c22f815f24bac45577a4f135' _python_enum_commit='4cfedc426c4e2fc52e3f5c2b4297e15ed8d6b8c7' _python_peachpy_commit='07d8fde8ac45d7705129475c0f94ed8925b93473' _python_six_commit='15e31431af97e5e64b80af0a3f598d382bcdd49a' _computelibrary_commit='292227986edb37b01061afcad6df18ba9d6ccbeb' -_onnx_commit='c4734c6200cb42c1aa36eb1f0160041d2401644d' -_cereal_commit='51cbda5f30e56c801c07fe3d3aba5d7fb9e6cca4' +_onnx_commit='42804705bdbf179d1a98394008417e1392013547' _onnx_tensorrt_commit='fa0964e8477fc004ee2f49ee77ffce0bf7f711a9' _sleef_commit='6ff7a135a1e31979d1e1844a2e7171dfbd34f54f' -_ideep_commit='dedff8fb8193fe3a1ea893d4bc852f8ea395b6b3' +_ideep_commit='7b8f9d025a675522305a1c5d1328d83f11f205e8' +_nccl_commit='3c6e25210bb1b544748937e5db74db0b9679b95e' +_gemmlowp_commit='8416bab644641a5c0a81ecf91a5cda804af0aee1' +_qnnpack_commit='ef05e87cef6b8e719989ce875b5e1c9fdb304c05' +_arm_neon_commit='97a126f08ce318023be604d03f88bf0820a9464a' +_fbgemm_commit='79333308f5e2fc242727879dcd3de3536b6ffc39' +_asmjit_commit='1e550aa568da993acb742ac19f2ab691a2806ad5' -_pytorchver=1.0rc1 # pytorch stable release version +_pytorchver=1.0.1 # pytorch stable release version pkgname=caffe2 pkgver="0.8.2.pytorch.${_pytorchver}" -pkgrel=8 +pkgrel=1 pkgdesc='A new lightweight, modular, and scalable deep learning framework' arch=('i686' 'x86_64') url='https://caffe2.ai/' @@ -42,7 +45,7 @@ depends=( 'google-glog' 'protobuf' 'lapack' 'python' 'python-numpy' 'python-protobuf' # not required but enabled in build: 'gflags' 'gtest' 'openmp' 'leveldb' 'lmdb' 'numactl' 'openmpi' 'snappy' - 'zeromq' 'hiredis' 'opencv' 'gtk3' 'ffmpeg' + 'zeromq' 'hiredis' 'gtk3' 'ffmpeg' # python: 'python-flask' 'python-future' 'graphviz' 'python-hypothesis' 'python-jupyter_core' 'python-matplotlib' 'python-pydot' 'python-yaml' @@ -63,13 +66,10 @@ options=('!emptydirs') source=( # main source: "pytorch-${_pytorchver}.tar.gz"::"https://github.com/pytorch/pytorch/archive/v${_pytorchver}.tar.gz" - # third party: - "git+https://github.com/catchorg/Catch2.git#commit=${_catch_commit}" - "git+https://github.com/pybind/pybind11.git#commit=${_pybind11_commit}" + "git+https://github.com/pybind/pybind11.git#commit=${_pybind11_commit}" "git+https://github.com/NVlabs/cub.git#commit=${_cub_commit}" "git+https://github.com/eigenteam/eigen-git-mirror.git#commit=${_eigen_commit}" "git+https://github.com/google/googletest.git#commit=${_googletest_commit}" - "git+https://github.com/NervanaSystems/nervanagpu.git#commit=${_nervanagpu_commit}" "git+https://github.com/google/benchmark.git#commit=${_benchmark_commit}" "git+https://github.com/google/protobuf.git#commit=${_protobuf_commit}" "git+https://github.com/Yangqing/ios-cmake.git#commit=${_ios_cmake_commit}" @@ -86,15 +86,18 @@ source=( "git+https://github.com/benjaminp/six.git#commit=${_python_six_commit}" "git+https://github.com/ARM-software/ComputeLibrary.git#commit=${_computelibrary_commit}" "git+https://github.com/onnx/onnx.git#commit=${_onnx_commit}" - "git+https://github.com/USCiLab/cereal.git#commit=${_cereal_commit}" - "git+https://github.com/onnx/onnx-tensorrt.git#commit=${_onnx_tensorrt_commit}" - "git+https://github.com/shibatch/sleef.git#commit=${_sleef_commit}" + "git+https://github.com/bddppq/onnx-tensorrt.git#commit=${_onnx_tensorrt_commit}" + "git+https://github.com/zdevito/sleef.git#commit=${_sleef_commit}" "git+https://github.com/intel/ideep.git#commit=${_ideep_commit}" - # patches: - 'caffe2-1.0rc1-add-full-relro.patch' - 'caffe2-1.0rc1-opencv4-fix.patch' + "git+https://github.com/NVIDIA/nccl.git#commit=${_nccl_commit}" + "git+https://github.com/google/gemmlowp.git#commit=${_gemmlowp_commit}" + "git+https://github.com/pytorch/QNNPACK.git#commit=${_qnnpack_commit}" + "git+https://github.com/intel/ARM_NEON_2_x86_SSE.git#commit=${_arm_neon_commit}" + "git+https://github.com/pytorch/fbgemm.git#commit=${_fbgemm_commit}" + # others: + "git+https://github.com/asmjit/asmjit.git#commit=${_asmjit_commit}" ) -sha256sums=('473cd4af032ddec4279cf3a90dd9508b6fa0be5cd89c842945f88b5a576a4231' +sha256sums=('210561583378132d07287518710251bcf861f9b7d625eb67258ed2e9d567cea1' 'SKIP' 'SKIP' 'SKIP' @@ -121,25 +124,28 @@ sha256sums=('473cd4af032ddec4279cf3a90dd9508b6fa0be5cd89c842945f88b5a576a4231' 'SKIP' 'SKIP' 'SKIP' - 'c41dd34bdf0bfbe62bba8527e950a14b356bef136edc53ea40c6469644502583' - '058d05d3fdf9e4d688dc29475265c69f99892ccc5f9926d45b26da9616eb936c') + 'SKIP' + 'SKIP' + 'SKIP') prepare() { cd "pytorch-${_pytorchver}/third_party" local _component - local _thirdparty_list="pybind11 cub googletest nervanagpu benchmark \ - protobuf ios-cmake NNPACK pthreadpool FXdiv FP16 \ - psimd gloo zstd cpuinfo ComputeLibrary onnx cereal \ - onnx-tensorrt sleef ideep" + local _thirdparty_list="pybind11 cub googletest benchmark protobuf ios-cmake \ + NNPACK pthreadpool FXdiv FP16 psimd gloo zstd cpuinfo \ + ComputeLibrary onnx onnx-tensorrt sleef ideep QNNPACK \ + fbgemm" # third_party which clone dir does not coincide with the third_party name - rm -rf catch eigen python-enum python-peachpy python-six - ln -sf "${srcdir}/Catch2" catch - ln -sf "${srcdir}/eigen-git-mirror" eigen - ln -sf "${srcdir}/enum34" python-enum - ln -sf "${srcdir}/PeachPy" python-peachpy - ln -sf "${srcdir}/six" python-six + rm -r eigen python-enum python-peachpy python-six nccl/nccl gemmlowp/gemmlowp neon2sse + ln -s "${srcdir}/eigen-git-mirror" eigen + ln -s "${srcdir}/enum34" python-enum + ln -s "${srcdir}/PeachPy" python-peachpy + ln -s "${srcdir}/six" python-six + ln -s "${srcdir}/nccl" nccl/nccl + ln -s "${srcdir}/gemmlowp" gemmlowp/gemmlowp + ln -s "${srcdir}/ARM_NEON_2_x86_SSE" neon2sse # third_party which clone dir coincide with the third_party name for _component in $_thirdparty_list @@ -147,13 +153,6 @@ prepare() { rm -rf "$_component" ln -sf "${srcdir}/${_component}" "${_component}" done - - # add full relro - cd "${srcdir}/pytorch-${_pytorchver}" - patch -Np1 -i "${srcdir}/caffe2-1.0rc1-add-full-relro.patch" - - # opencv 4.0 fix - patch -Np1 -i "${srcdir}/caffe2-1.0rc1-opencv4-fix.patch" } build() { @@ -179,6 +178,7 @@ build() { -DGLOO_STATIC_OR_SHARED:STRING='STATIC' \ \ -DOpenCV_DIR:PATH='/usr/share/OpenCV' \ + -DASMJIT_SRC_DIR:STRING="${srcdir}/asmjit" \ \ -DPYTHON_EXECUTABLE:FILEPATH="/usr/bin/python${_pythonver}" \ -DPYTHON_INCLUDE_DIR:PATH="/usr/include/python${_pythonver}m" \ @@ -189,32 +189,31 @@ build() { -DUSE_CUDA:BOOL='OFF' \ -DUSE_CUDNN:BOOL='OFF' \ -DUSE_DISTRIBUTED:BOOL='ON' \ + -DUSE_FBGEMM:BOOL='ON' \ -DUSE_FFMPEG:BOOL='ON' \ -DUSE_GFLAGS:BOOL='ON' \ -DUSE_GLOG:BOOL='ON' \ -DUSE_GLOO:BOOL='ON' \ -DUSE_GLOO_IBVERBS:BOOL='ON' \ -DUSE_IBVERBS:BOOL='ON' \ - -DUSE_IDEEP:BOOL='ON' \ -DUSE_LEVELDB:BOOL='ON' \ -DUSE_LITE_PROTO:BOOL='OFF' \ -DUSE_LMDB:BOOL='ON' \ -DUSE_METAL:BOOL='OFF' \ -DUSE_MKLDNN:BOOL='OFF' \ - -DUSE_MKLML:BOOL='OFF' \ - -DUSE_MOBILE_OPENGL:BOOL='OFF' \ -DUSE_MPI:BOOL='ON' \ -DUSE_NCCL:BOOL='OFF' \ - -DUSE_NERVANA_GPU:BOOL='OFF' \ - -DUSE_NNAPI:BOOL='OFF' \ + -DUSE_NNAPI:BOOL='ON' \ -DUSE_NNPACK:BOOL='ON' \ -DUSE_NUMA:BOOL='ON' \ + -DUSE_NUMPY:BOOL='ON' \ -DUSE_NVRTC:BOOL='OFF' \ -DUSE_OBSERVERS:BOOL='ON' \ -DUSE_OPENCL:BOOL='OFF' \ - -DUSE_OPENCV:BOOL='ON' \ + -DUSE_OPENCV:BOOL='OFF' \ -DUSE_OPENMP:BOOL='ON' \ -DUSE_PROF:BOOL='OFF' \ + -DUSE_QNNPACK:BOOL='ON' \ -DUSE_REDIS:BOOL='ON' \ -DUSE_ROCKSDB:BOOL='OFF' \ -DUSE_ROCM:BOOL='OFF' \ @@ -224,6 +223,7 @@ build() { -DUSE_TENSORRT:BOOL='OFF' \ -DUSE_ZMQ:BOOL='ON' \ -DUSE_ZSTD:BOOL='ON' \ + -DWITH_OPENMP:BOOL='ON' \ \ -Wno-dev \ .. @@ -243,11 +243,11 @@ package() { rm "$pkgdir"/usr/bin/{protoc,unzstd,zstd{cat,mt,}} rm "$pkgdir"/usr/include/{*.h,*.py} rm "$pkgdir"/usr/lib/*.a - rm "$pkgdir"/usr/lib/lib{zstd,onnxifi}* + rm "$pkgdir"/usr/lib/lib{onnxifi,zstd}* rm -r "$pkgdir"/usr/lib/cmake/protobuf rm "$pkgdir"/usr/lib/pkgconfig/{protobuf-lite,protobuf}.pc rm "$pkgdir"/usr/share/pkgconfig/libzstd.pc - rm -r "$pkgdir"/usr/share/{ATen,cmake/{ATen,ONNX}} + rm -r "$pkgdir"/usr/share/{ATen,cmake/{ATen,Gloo,ONNX}} rm "$pkgdir"/usr/share/man/man1/{unzstd,zstd{cat,}}.1 for _entry in "${_exclude_dirs[@]}" do |