From ab859789ed705b54db81e49cefb4c2b1ef895795 Mon Sep 17 00:00:00 2001
From: hlu1 <14827759+hlu1@users.noreply.github.com>
Date: Tue, 19 Mar 2019 15:25:16 -0700
Subject: [PATCH] [Docker] Enable NNPACK for ci_gpu

---
 Jenkinsfile | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/Jenkinsfile b/Jenkinsfile
index a56296c98c54a..26d5c96993c9e 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -21,7 +21,7 @@
 // - Periodically cleanup the old versions on local workers
 //
 ci_lint = "tvmai/ci-lint:v0.50"
-ci_gpu = "tvmai/ci-gpu:v0.51"
+ci_gpu = "tvmai/ci-gpu:v0.52"
 ci_cpu = "tvmai/ci-cpu:v0.41"
 ci_i386 = "tvmai/ci-i386:v0.50"
 
@@ -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