Skip to content
This repository has been archived by the owner on Sep 1, 2024. It is now read-only.

move x to device for gaussian mlp tests #179

Merged

Conversation

raghavauppuluri13
Copy link
Contributor

@raghavauppuluri13 raghavauppuluri13 commented Feb 9, 2023

Types of changes

  • Docs change / refactoring / dependency upgrade
  • 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

  • The documentation is up-to-date with the changes I made.
  • I have read the CONTRIBUTING document and completed the CLA (see CONTRIBUTING).
  • All tests passed, and additional code has been covered with new tests.

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Feb 9, 2023
@luisenp luisenp merged commit 3c6553a into facebookresearch:main Feb 14, 2023
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants