From 81c4c40f8a72d8ea6a1989624728c0e052b3b824 Mon Sep 17 00:00:00 2001 From: tqchen Date: Tue, 4 Feb 2020 12:51:01 -0800 Subject: [PATCH] [CI][DOCKER] Update ci-gpu to v0.60 --- Jenkinsfile | 2 +- tests/python/frontend/onnx/test_forward.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/Jenkinsfile b/Jenkinsfile index 6b92f969142c..e65f73e8c6d5 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 fc05e7a011a9..f9e8aac4e389 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 @@ 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