Skip to content
This repository has been archived by the owner on May 12, 2024. It is now read-only.

Commit

Permalink
remove nnpack
Browse files Browse the repository at this point in the history
  • Loading branch information
PINTO0309 committed Mar 28, 2022
1 parent 915ef5f commit d0a2bd5
Showing 1 changed file with 15 additions and 14 deletions.
29 changes: 15 additions & 14 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -184,19 +184,20 @@ RUN wget https://apt.llvm.org/llvm.sh \
&& apt clean \
&& rm -rf /var/lib/apt/lists/*

# NNPACK
RUN git clone --recursive https://github.com/Maratyszcza/NNPACK.git \
&& cd NNPACK \
&& git checkout c07e3a0400713d546e0dea2d5466dd22ea389c73 \
&& sed -i "s|gnu99|gnu99 -fPIC|g" CMakeLists.txt \
&& sed -i "s|gnu++11|gnu++11 -fPIC|g" CMakeLists.txt \
&& mkdir build \
&& cd build \
&& cmake -G Ninja -D BUILD_SHARED_LIBS=ON .. \
&& ninja \
&& ninja install \
&& sh -c "echo '/usr/local/lib' > /etc/ld.so.conf.d/nnpack.conf" \
&& ldconfig
# # NNPACK
# RUN git clone --recursive https://github.com/Maratyszcza/NNPACK.git \
# && cd NNPACK \
# && git checkout c07e3a0400713d546e0dea2d5466dd22ea389c73 \
# && git submodule update --init --recursive \
# && sed -i "s|gnu99|gnu99 -fPIC|g" CMakeLists.txt \
# && sed -i "s|gnu++11|gnu++11 -fPIC|g" CMakeLists.txt \
# && mkdir build \
# && cd build \
# && cmake -G Ninja -D BUILD_SHARED_LIBS=ON .. \
# && ninja \
# && ninja install \
# && sh -c "echo '/usr/local/lib' > /etc/ld.so.conf.d/nnpack.conf" \
# && ldconfig

# Clear caches
RUN apt clean \
Expand Down Expand Up @@ -248,7 +249,7 @@ RUN git clone --recursive https://github.com/apache/tvm \
-DUSE_LLVM=ON \
-DUSE_MKLDNN=ON \
-DUSE_OPENMP=ON \
-DUSE_NNPACK=ON \
-DUSE_NNPACK=OFF \
# -DUSE_TFLITE=/usr/local/lib/libtensorflow-lite.a \
# -DUSE_EDGETPU=OFF \
-DUSE_CUDNN=ON \
Expand Down

0 comments on commit d0a2bd5

Please sign in to comment.