• X
    Run Python OP tests in a single Python process to improve test time. (#8362) · cde6241a
    Xin Pan 提交于
    Currently, our tests run with 2 GPUs, the init time is absurdly long:
    about 4s for each process.  Currently, we run each OP test on
    different processes. This PR:
    
    1. create cmake function py_test_modules which will generate the
    Makefile that runs a list of Python unittest module in a single Python
    process.
    
    2. move all "python unittest compatible" (e.g., used the unittest
    package, not just a regular python file). from fluid/tests to
    fluid/tests/unittests.
    
    3. cmake now will run all OP tests in fluid/tests/unittests in a
    single process, except the time-consuming tests, they are separated
    into different processes to utilize parallelism. Please make sure to
    use the unittest package if you put the python test file in
    fluid/tests/unittests
    
    4. remove all exit(0) from fluid/tests/unittests/*.py, exit(0) is used
    to disable unittest, we can not do it when running all tests in a
    single process since it will terminate the process without running the
    other tests. Instead, the test is disabled in
    fluid/tests/unittests/CMakeLists.txt. FIXME is added for each disabled
    item. Please disable the unittest from
    fluid/tests/unittests/CMakeLists.txt, instead of adding exit(0) to the
    Python file, for all Python file in fluid/tests/unittests/.
    
    5. add an option WITH_FAST_BUNDLE_TEST. When OFF, will run the unit
    tests in separate process so that they can be tested individually.
    cde6241a
test_adam_op.py 10.6 KB