tvm-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From haic...@apache.org
Subject [incubator-tvm] branch master updated: [CI] Update MxNet to 1.6.0 with MKL (#5240)
Date Mon, 06 Apr 2020 06:38:34 GMT
This is an automated email from the ASF dual-hosted git repository.

haichen pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git


The following commit(s) were added to refs/heads/master by this push:
     new 41b8fd1  [CI] Update MxNet to 1.6.0 with MKL (#5240)
41b8fd1 is described below

commit 41b8fd1ead60e6f41594f6b3690cb1fff6feeef9
Author: Haichen Shen <shenhaichen@gmail.com>
AuthorDate: Sun Apr 5 23:38:24 2020 -0700

    [CI] Update MxNet to 1.6.0 with MKL (#5240)
---
 docker/install/ubuntu_install_mxnet.sh      | 2 +-
 tests/python/frontend/mxnet/test_forward.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/docker/install/ubuntu_install_mxnet.sh b/docker/install/ubuntu_install_mxnet.sh
index 0ce2e8d..d587843 100755
--- a/docker/install/ubuntu_install_mxnet.sh
+++ b/docker/install/ubuntu_install_mxnet.sh
@@ -20,4 +20,4 @@ set -e
 set -u
 set -o pipefail
 
-pip3 install mxnet==1.5.0
+pip3 install mxnet-mkl==1.6.0
diff --git a/tests/python/frontend/mxnet/test_forward.py b/tests/python/frontend/mxnet/test_forward.py
index f015447..eb308c5 100644
--- a/tests/python/frontend/mxnet/test_forward.py
+++ b/tests/python/frontend/mxnet/test_forward.py
@@ -125,7 +125,7 @@ def test_forward_rrelu():
     data = mx.sym.var('data')
     data = mx.sym.concat(data, -data, dim=1)  # negative part explicitly
     mx_sym = mx.sym.LeakyReLU(data, act_type='rrelu', lower_bound=0.3, upper_bound=0.7)
-    verify_mxnet_frontend_impl(mx_sym, (1, 3, 100, 100), (1, 6, 100, 100))
+    verify_mxnet_frontend_impl(mx_sym[0], (1, 3, 100, 100), (1, 6, 100, 100))
 
 def test_forward_prelu():
     data = mx.sym.var('data')


Mime
View raw message