marcoabreu commented on issue #10354: Expose the number of GPUs. URL: https://github.com/apache/incubator-mxnet/pull/10354#issuecomment-380253789 I think we can't use assertRaises because we don't have CPU-only tests (besides the MKL ones) as of now. Instead, we'd have to go with a try catch approach. If it stays in the try clause, we're probably on a GPU instance and it should be > 0 and in the catch clause we should be on a CPU instance.
---------------------------------------------------------------- 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
