[x] Bug fix (non-breaking change which fixes an issue)
[ ] New feature (non-breaking change which adds functionality)
[ ] Breaking change (fix or feature that would cause existing functionality to change)
Motivation and Context / Related issue
The tests test_gaussian_mlp_ensemble_random_model_propagation and test_gaussian_mlp_ensemble_fixed_model_propagation fail when using dual-gpu linux system.
How Has This Been Tested (if it applies)
Ran previously failed tests on dual-gpu linux machine and they pass.
Checklist
[x] The documentation is up-to-date with the changes I made.
[x] I have read the CONTRIBUTING document and completed the CLA (see CONTRIBUTING).
[x] All tests passed, and additional code has been covered with new tests.
Types of changes
Motivation and Context / Related issue
The tests
test_gaussian_mlp_ensemble_random_model_propagation
andtest_gaussian_mlp_ensemble_fixed_model_propagation
fail when using dual-gpu linux system.How Has This Been Tested (if it applies)
Ran previously failed tests on dual-gpu linux machine and they pass.
Checklist