Commit 7ef32cdc by hlu1 Committed by Tianqi Chen

[Docker] Enable NNPACK for ci_gpu (#2856)

parent d5b34220
...@@ -115,6 +115,8 @@ stage('Build') { ...@@ -115,6 +115,8 @@ stage('Build') {
echo set\\(USE_CUDA ON\\) >> config.cmake echo set\\(USE_CUDA ON\\) >> config.cmake
echo set\\(USE_OPENGL ON\\) >> config.cmake echo set\\(USE_OPENGL ON\\) >> config.cmake
echo set\\(USE_LLVM llvm-config-6.0\\) >> config.cmake echo set\\(USE_LLVM llvm-config-6.0\\) >> config.cmake
echo set\\(USE_NNPACK ON\\) >> config.cmake
echo set\\(NNPACK_PATH /NNPACK/build/\\) >> config.cmake
echo set\\(USE_RPC ON\\) >> config.cmake echo set\\(USE_RPC ON\\) >> config.cmake
echo set\\(USE_SORT ON\\) >> config.cmake echo set\\(USE_SORT ON\\) >> config.cmake
echo set\\(USE_GRAPH_RUNTIME ON\\) >> config.cmake echo set\\(USE_GRAPH_RUNTIME ON\\) >> config.cmake
......
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