From b88de43d2761dfbf3123670e476d9a6a639e7c84 Mon Sep 17 00:00:00 2001 From: Tianqi Chen Date: Thu, 6 Feb 2020 11:58:11 -0800 Subject: [PATCH] [CI][DOCKER] Update ci-gpu to v0.60 (#4827) --- Jenkinsfile | 2 +- tests/python/frontend/onnx/test_forward.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/Jenkinsfile b/Jenkinsfile index 6b92f96..e65f73e 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.60" ci_cpu = "tvmai/ci-cpu:v0.55" ci_i386 = "tvmai/ci-i386:v0.52" diff --git a/tests/python/frontend/onnx/test_forward.py b/tests/python/frontend/onnx/test_forward.py index fc05e7a..f9e8aac 100644 --- a/tests/python/frontend/onnx/test_forward.py +++ b/tests/python/frontend/onnx/test_forward.py @@ -16,6 +16,8 @@ # under the License. import numpy as np import math +import onnx +from onnx import helper, TensorProto, mapping import torch import torchvision import topi @@ -24,8 +26,6 @@ import tvm from tvm import relay from tvm.contrib import graph_runtime from tvm.relay.testing.config import ctx_list -import onnx -from onnx import helper, TensorProto, mapping import scipy -- 2.7.4