lebeg commented on a change in pull request #12726: Fix failing GPU test on single GPU host (kvstore) #10977 URL: https://github.com/apache/incubator-mxnet/pull/12726#discussion_r222113617
########## File path: tests/python/gpu/test_kvstore_gpu.py ########## @@ -21,7 +21,7 @@ import mxnet as mx import numpy as np import unittest -from mxnet.test_utils import assert_almost_equal, default_context +from mxnet.test_utils import assert_almost_equal, default_context, list_gpus Review comment: list_gpus in utils is a bad way to identify the GPU's, better use [context.num_gpus](https://github.com/apache/incubator-mxnet/blob/master/python/mxnet/context.py#L244) ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services