fix unittests, testing 4 GPU