Fix gpu test

apache · Aug 3, 2020 · 0b45794 · 0b45794
1 parent 644c730
commit 0b45794
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/python/unittest/test_gluon.py b/tests/python/unittest/test_gluon.py
@@ -313,7 +313,7 @@ def hybrid_forward(self, F, x):
     if compute_before_cast:
         # Compute before casting to catch bugs where symbol dtype isn't casted correctly GH-18843
         net_fp32.initialize()
-        net_fp32(mx.nd.zeros((1,3,224,224)))
+        net_fp32(mx.nd.zeros((1,3,224,224)), ctx=ctx)
     net_fp32.cast('float64')
     net_fp32.hybridize()
     data = mx.nd.zeros((1,3,224,224), dtype='float64', ctx=ctx)