[incubator-mxnet] branch master updated: Update test_operator_gpu.py (#8819)

[jx...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

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


The following commit(s) were added to refs/heads/master by this push:
     new 91ffd69  Update test_operator_gpu.py (#8819)
91ffd69 is described below

commit 91ffd691bf865833a51ca793eda125b7f9befc18
Author: Sheng Zha <sz...@users.noreply.github.com>
AuthorDate: Sun Nov 26 13:58:45 2017 -0800

    Update test_operator_gpu.py (#8819)
    
    change clip norm test to assume 1 gpu only
---
 tests/python/gpu/test_operator_gpu.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/python/gpu/test_operator_gpu.py b/tests/python/gpu/test_operator_gpu.py
index 15354b6..fad5940 100644
--- a/tests/python/gpu/test_operator_gpu.py
+++ b/tests/python/gpu/test_operator_gpu.py
@@ -1425,7 +1425,7 @@ def test_cuda_rtc():
 
 def test_global_norm_clip_multi_device():
     x1 = mx.nd.ones((3,3), ctx=mx.gpu(0))
-    x2 = mx.nd.ones((4,4), ctx=mx.gpu(1))
+    x2 = mx.nd.ones((4,4), ctx=mx.cpu(0))
     norm = gluon.utils.clip_global_norm([x1, x2], 1.0)
     assert norm == 5.0
     assert_almost_equal(x1.asnumpy(), np.ones((3,3))/5)

-- 
To stop receiving notification emails like this one, please contact
['"commits@mxnet.apache.org" <co...@mxnet.apache.org>'].