From: Haichen Shen <shenhaichen@gmail.com>
Date: Mon, 6 Apr 2020 06:38:24 +0000 (-0700)
Subject: [CI] Update MxNet to 1.6.0 with MKL (#5240)
X-Git-Tag: upstream/0.7.0~978
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=41b8fd1ead60e6f41594f6b3690cb1fff6feeef9;p=platform%2Fupstream%2Ftvm.git

[CI] Update MxNet to 1.6.0 with MKL (#5240)
---

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')