modify Dockerfile to compile tritonbackend (#305)

Co-authored-by: zhoubofan <[email protected]>
bytedance · May 5, 2022 · 3c1c506 · 3c1c506
1 parent b5bc246
commit 3c1c506
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/Tritonserver/Dockerfile b/docker/Tritonserver/Dockerfile
@@ -55,7 +55,7 @@ ENV PATH /usr/local/cuda/bin:/usr/local/hdf5:/opt/tritonserver/bin:${PATH}
 # compile lightseq
 RUN cd /opt && git clone --recursive https://github.com/bytedance/lightseq.git && \
     cd lightseq && mkdir build && cd build && \
-    cmake -DCMAKE_BUILD_TYPE=Release -DFP16_MODE=ON -DDEBUG_MODE=OFF -DDYNAMIC_API=ON .. && \
+    cmake -DCMAKE_BUILD_TYPE=Release -DFP16_MODE=ON -DDEBUG_MODE=OFF -DDYNAMIC_API=ON -DUSE_TRITONBACKEND=ON .. && \
     make -j${nproc}
 
 # organize lightseq-tritonbackend dynamic link library