Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
WIP
Browse files Browse the repository at this point in the history
  • Loading branch information
KellenSunderland committed Nov 18, 2018
1 parent cd64c32 commit e410dd8
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion ci/docker/runtime_functions.sh
Original file line number Diff line number Diff line change
Expand Up @@ -489,6 +489,8 @@ build_ubuntu_gpu_tensorrt() {

build_ccache_wrappers

export LDFLAGS="-Wl,--as-needed"

# Build ONNX
pushd .
echo "Installing ONNX."
Expand All @@ -498,7 +500,7 @@ build_ubuntu_gpu_tensorrt() {
cd build
cmake \
-DCMAKE_CXX_FLAGS=-I/usr/include/python${PYVER}\
-DBUILD_SHARED_LIBS=ON ..\
-DBUILD_SHARED_LIBS=OFF ..\
-G Ninja
ninja -j 1 -v onnx/onnx.proto
ninja -j 1 -v
Expand All @@ -518,6 +520,7 @@ build_ubuntu_gpu_tensorrt() {

mkdir -p /work/mxnet/lib/
cp 3rdparty/onnx-tensorrt/third_party/onnx/build/*.so /work/mxnet/lib/
cp 3rdparty/onnx-tensorrt/third_party/onnx/build/*.a /work/mxnet/lib/
cp -L 3rdparty/onnx-tensorrt/build/libnvonnxparser_runtime.so.0 /work/mxnet/lib/
cp -L 3rdparty/onnx-tensorrt/build/libnvonnxparser.so.0 /work/mxnet/lib/

Expand Down

0 comments on commit e410dd8

Please sign in to comment.