Fix test

Change-Id: I96c529abe7adb6def90a22f03b3432263ef12fda
apache · Aug 27, 2019 · 5ea26ed · 5ea26ed
1 parent bc4d3b5
commit 5ea26ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/unittest/test_operator.py b/tests/python/unittest/test_operator.py
@@ -291,7 +291,7 @@ def test_RNN_float64():
     args_grad = explicit_grad
     grad_req = 'write'
 
-    ex = sym.bind(mx.cpu(),
+    ex = sym.bind(default_context(),
         {
             'in': mx.nd.ones([2, 1, 2], dtype=dtype),
             'par': mx.nd.ones([12], dtype=dtype),