Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Rework grouping to deal with corner cases better #4

Open
wants to merge 10 commits into
base: multi_gpu
Choose a base branch
from

Conversation

thatguymike
Copy link

Fix issues found on EC2 (no p2p support), boards with more than one ASIC, and fix p2p grouping a little better.

cypof and others added 10 commits May 18, 2015 17:24
- Interrupt the thread before waiting on join
- Provide a method for looping threads to exit on demand
- CHECK if start and stop succeed instead of returning an error
- Makes sure each solver accesses a different subset of the data
- Sequential reading of DB for performance
- Prefetches a configurable amount of data to host memory
- Distributes data to solvers in round-robin way for determinism
- Split batches between GPUs, and tree-reduce the gradients
- Detects machine topology (twin-GPU boards, P2P connectivity)
- Inserts a callbak in the solver for minimal code change
- Added gpus flags to caffe.cpp, e.g. '-gpus 0,1' or '-gpus all'
- Deterministic architecture for reproducible runs
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants