-
Notifications
You must be signed in to change notification settings - Fork 6
Open
Description
Description of the problem
- using layer->len_ops number of threads may cause an issue for a neural network where the input dimension is high.
- Threads need not be passed to _rr_scheduler in the present implementation.
BNN/bnn/autodiff/forward_impl.cpp
Lines 50 to 56 in 3d752fe
while(layer != NULL) { unsigned threads = layer->len_ops, i; thread* pool[threads]; op_queue<data_type>* jobs[threads][2]; _rr_scheduler<data_type>(layer, jobs, threads);
Example of the problem
References/Other comments
Metadata
Metadata
Assignees
Labels
No labels