[Docker] Enable NNPACK for ci_gpu (apache#2856)

Laurawly · Mar 22, 2019 · 831521f · 831521f
1 parent 7ab3876
commit 831521f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Jenkinsfile b/Jenkinsfile
@@ -115,6 +115,8 @@ stage('Build') {
            echo set\\(USE_CUDA ON\\) >> config.cmake
            echo set\\(USE_OPENGL ON\\) >> 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_SORT ON\\) >> config.cmake
            echo set\\(USE_GRAPH_RUNTIME ON\\) >> config.cmake