Skip to content

GitLab

Commit 19aca522 authored by Benjamin Thomas Graham's avatar Benjamin Thomas Graham
Browse files

Fix CPU IOLayers

parent c9e5e6cd
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
sparseconvnet/SCN/CPU/IOLayers.cpp
View file @ 19aca522
......@@ -15,10 +15,10 @@ void InputLayer_ForwardPass(T *input_features, T *output_features, Int nRows,
Int row;
#pragma omp parallel for private(row)
for (row = 0; row < nRows; row++) {
auto nActive = rules[0];
auto r = rules + row * (1 + maxActive);
auto nActive = r[0];
T multiplier = (average and nActive > 0) ? (T)1 / nActive : (T)1;
auto out_f = output_features + row * nPlanes;
auto r = rules + row * (1 + maxActive);
for (Int i = 1; i <= nActive; ++i) {
auto in_f = input_features + r[i] * nPlanes;
for (Int plane = 0; plane < nPlanes; plane++) {
......@@ -34,10 +34,10 @@ void InputLayer_BackwardPass(T *d_input_features, T *d_output_features,
Int row;
#pragma omp parallel for private(row)
for (row = 0; row < nRows; row++) {
auto nActive = rules[0];
auto r = rules + row * (1 + maxActive);
auto nActive = r[0];
T multiplier = (average and nActive > 0) ? (T)1 / nActive : (T)1;
auto d_out_f = d_output_features + row * nPlanes;
auto r = rules + row * (1 + maxActive);
for (Int i = 1; i <= nActive; ++i) {
auto d_in_f = d_input_features + r[i] * nPlanes;
for (Int plane = 0; plane < nPlanes; plane++)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment