Skip to content

T
tic
Unverified Commit b88de43d by Tianqi Chen Committed by GitHub
Options

[CI][DOCKER] Update ci-gpu to v0.60 (#4827)

parent 62543d49
Showing with 3 additions and 3 deletions
Jenkinsfile
...@@ -45,7 +45,7 @@ ...@@ -45,7 +45,7 @@
// //
ci_lint = "tvmai/ci-lint:v0.51" 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_cpu = "tvmai/ci-cpu:v0.55"
ci_i386 = "tvmai/ci-i386:v0.52" ci_i386 = "tvmai/ci-i386:v0.52"
......
tests/python/frontend/onnx/test_forward.py
...@@ -16,6 +16,8 @@ ...@@ -16,6 +16,8 @@
# under the License. # under the License.
import numpy as np import numpy as np
import math import math
import onnx
from onnx import helper, TensorProto, mapping
import torch import torch
import torchvision import torchvision
import topi import topi
...@@ -24,8 +26,6 @@ import tvm ...@@ -24,8 +26,6 @@ import tvm
from tvm import relay from tvm import relay
from tvm.contrib import graph_runtime from tvm.contrib import graph_runtime
from tvm.relay.testing.config import ctx_list from tvm.relay.testing.config import ctx_list
import onnx
from onnx import helper, TensorProto, mapping
import scipy import scipy
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment