Mutli-nGraph Backend Unit tests

NervanaSystems / ngraph-mxnet

MXNet - nGraph integration

Apache License 2.0

34 stars 3 forks source link

Mutli-nGraph Backend Unit tests #311

Open mbrookhart opened 6 years ago

mbrookhart commented 6 years ago

Acceptance Criteria:

1) NGraph unit tests run the standard operator unit tests with the new context from https://github.com/NervanaSystems/ngraph-mxnet/issues/289 2) MXNet test files are not modified. 3) tests nGraph needs to skip are handled in the nGraph test files. 4) CI is updated to use new context tests.

cconvey commented 6 years ago

A few (mutually exclusive) ideas for how to solve this:

Add a feature to MXNet itself in which the default context can be overridden via an environment variable.
Modify the test scripts so that CMake's "configure" system tweaks which particular hardcoded context is used.

mbrookhart commented 6 years ago

Haven't gotten to fleshing this one out yet, mxnet.test_utils.set_default_context can do this. The basic idea is, say, make a test_ngraph_cpu.py file, have a mxnet.test_utils.set_default_context(mx.ngraph("CPU")), a from test_operator import *, and an if __name__=='__main__': import nose nose.runmodule()