Commit 3ee13fc5 by Andrew Tulloch Committed by Tianqi Chen

Add NNPACK to CI (#2085)

parent fc83c7f2
...@@ -35,4 +35,8 @@ RUN bash /install/ubuntu_install_redis.sh ...@@ -35,4 +35,8 @@ RUN bash /install/ubuntu_install_redis.sh
COPY install/ubuntu_install_golang.sh /install/ubuntu_install_golang.sh COPY install/ubuntu_install_golang.sh /install/ubuntu_install_golang.sh
RUN bash /install/ubuntu_install_golang.sh RUN bash /install/ubuntu_install_golang.sh
# NNPACK deps
COPY install/ubuntu_install_nnpack.sh /install/ubuntu_install_nnpack.sh
RUN bash /install/ubuntu_install_nnpack.sh
ENV PATH $PATH:$CARGO_HOME/bin:/usr/lib/go-1.10/bin ENV PATH $PATH:$CARGO_HOME/bin:/usr/lib/go-1.10/bin
apt-get update && apt-get install -y --no-install-recommends --force-yes git cmake
git clone https://github.com/Maratyszcza/NNPACK NNPACK
cd NNPACK
# TODO: specific tag?
git checkout 1e005b0c2
cd -
mkdir -p NNPACK/build
cd NNPACK/build
cmake -DCMAKE_INSTALL_PREFIX:PATH=. -DNNPACK_INFERENCE_ONLY=OFF -DNNPACK_CONVOLUTION_ONLY=OFF -DNNPACK_BUILD_TESTS=OFF -DCMAKE_POSITION_INDEPENDENT_CODE=ON .. && make -j4 && make install
cd -
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment