From 4548200ec9680aa927a93bd157b5f06af287d03f Mon Sep 17 00:00:00 2001 From: shoubhik Date: Mon, 20 Jan 2020 16:39:33 -0800 Subject: [PATCH 1/3] [WIP] end to end test case for mxnet qnn parser --- docker/install/ubuntu_install_mxnet.sh | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/docker/install/ubuntu_install_mxnet.sh b/docker/install/ubuntu_install_mxnet.sh index 0ce2e8d9361f..4c5ba1b8006d 100755 --- a/docker/install/ubuntu_install_mxnet.sh +++ b/docker/install/ubuntu_install_mxnet.sh @@ -19,5 +19,6 @@ set -e set -u set -o pipefail - -pip3 install mxnet==1.5.0 +# We need dependency on mkl version to be able to +# test quantized models end-to-end. +pip3 install mxnet-mkl==1.5.1 From dcf329c0249a8a76b0d2b853d3220a787462b38f Mon Sep 17 00:00:00 2001 From: shoubhik Date: Tue, 21 Jan 2020 09:32:26 -0800 Subject: [PATCH 2/3] Updating the version numbers for the docker gpu image. --- Jenkinsfile | 2 +- docker/Dockerfile.ci_gpu | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/Jenkinsfile b/Jenkinsfile index 6b92f969142c..f7ed14ca3ab4 100644 --- a/Jenkinsfile +++ b/Jenkinsfile @@ -45,7 +45,7 @@ // ci_lint = "tvmai/ci-lint:v0.51" -ci_gpu = "tvmai/ci-gpu:v0.56" +ci_gpu = "tvmai/ci-gpu:v0.57" ci_cpu = "tvmai/ci-cpu:v0.55" ci_i386 = "tvmai/ci-i386:v0.52" diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu index 7a65a70061b7..72df12d55e13 100644 --- a/docker/Dockerfile.ci_gpu +++ b/docker/Dockerfile.ci_gpu @@ -16,7 +16,7 @@ # under the License. # CI docker GPU env -# tag: v0.56 +# tag: v0.57 FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04 # Base scripts From 3f25b87b814635e1e80e0b07a3cfb2292b3bf6a5 Mon Sep 17 00:00:00 2001 From: shoubhik Date: Tue, 21 Jan 2020 11:14:32 -0800 Subject: [PATCH 3/3] Reverting the update of docker image tag for ci_gpu --- Jenkinsfile | 2 +- docker/Dockerfile.ci_gpu | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/Jenkinsfile b/Jenkinsfile index f7ed14ca3ab4..6b92f969142c 100644 --- a/Jenkinsfile +++ b/Jenkinsfile @@ -45,7 +45,7 @@ // ci_lint = "tvmai/ci-lint:v0.51" -ci_gpu = "tvmai/ci-gpu:v0.57" +ci_gpu = "tvmai/ci-gpu:v0.56" ci_cpu = "tvmai/ci-cpu:v0.55" ci_i386 = "tvmai/ci-i386:v0.52" diff --git a/docker/Dockerfile.ci_gpu b/docker/Dockerfile.ci_gpu index 72df12d55e13..7a65a70061b7 100644 --- a/docker/Dockerfile.ci_gpu +++ b/docker/Dockerfile.ci_gpu @@ -16,7 +16,7 @@ # under the License. # CI docker GPU env -# tag: v0.57 +# tag: v0.56 FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04 # Base scripts