marcoabreu commented on a change in pull request #10354: Expose the number of
GPUs.
URL: https://github.com/apache/incubator-mxnet/pull/10354#discussion_r187772910
##
File path: python/mxnet/context.py
##
@@ -212,6 +216,25 @@ def gpu(device_id=0):
return
marcoabreu commented on a change in pull request #10354: Expose the number of
GPUs.
URL: https://github.com/apache/incubator-mxnet/pull/10354#discussion_r182601933
##
File path: include/mxnet/base.h
##
@@ -307,6 +312,21 @@ inline Context Context::GPU(int32_t dev_id) {