Join GitHub today
GitHub is home to over 50 million developers working together to host and review code, manage projects, and build software together.
Sign upGitHub is where the world builds software
Millions of developers and companies build, ship, and maintain their software on GitHub — the largest and most advanced development platform in the world.
Enable running the mnist_training sample without cuda #6085
+7
−4
Conversation
Signed-off-by: George Nash <[email protected]>
939cc9b
into
microsoft:master
21 checks passed
21 checks passed
license/cla
All CLA requirements met.
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Signed-off-by: George Nash [email protected]
Description:
Place the cuda specific import into a try block.
Moved other cuda specific code into the
if use_cudablockMotivation and Context
Why is this change required? What problem does it solve?
While working on CPU training code this sample is useful to quickly judge if there is an issue with the training build. However, the cuda specific code prevents running the sample.
If it fixes an open issue, please link to the issue here.
#6084