diff options
| author | Laurens van der Maaten <lvdmaaten@gmail.com> | 2017-04-05 23:15:11 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2017-04-05 23:15:11 +0300 |
| commit | 24eb1d26310c51690c0a7f3ba6a571553ffad4e9 (patch) | |
| tree | 2e7cd0033b4e8d7871cb747afb14954baa088b68 | |
| parent | f225bb44c2cbccf90a48ffc4f25ee9056958ac63 (diff) | |
Remove global `rangeStart`lvdmaaten-patch-1
| -rw-r--r-- | DataParallelTable.lua | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/DataParallelTable.lua b/DataParallelTable.lua index 22c7739..a662b76 100644 --- a/DataParallelTable.lua +++ b/DataParallelTable.lua @@ -470,10 +470,10 @@ local function sliceRange(nElem, idx, splits) local eltsPerMod = math.floor(nElem / splits) local numExtra = nElem - eltsPerMod * splits if idx <= numExtra then - rangeStart = (idx - 1) * (eltsPerMod + 1) + 1 + local rangeStart = (idx - 1) * (eltsPerMod + 1) + 1 return rangeStart, eltsPerMod + 1 else - rangeStart = numExtra * (eltsPerMod + 1) + (idx - 1 - numExtra) * eltsPerMod + 1 + local rangeStart = numExtra * (eltsPerMod + 1) + (idx - 1 - numExtra) * eltsPerMod + 1 return rangeStart, eltsPerMod end end |