stepan
/
dynare
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Dynare++: small optimization in unfolded tensor indexing

time-shift
Sébastien Villemot 2019-02-20 13:05:04 +01:00
parent 30b8681731
commit c0224a640c
No known key found for this signature in database
GPG Key ID: 2CECE9350ECEBE4A
1 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
dynare++/tl/cc/tensor.cc
View File

@ -83,12 +83,11 @@ UTensor::decrement(IntSequence &v, const IntSequence &nvmx)
int
UTensor::getOffset(const IntSequence &v, int nv)
{
int pow = 1;
int res = 0;
for (int i = v.size()-1; i >= 0; i--)
for (int i = 0; i < v.size(); i++)
{
res += v[i]*pow;
pow *= nv;
res *= nv;
res += v[i];
}
return res;
}
@ -98,12 +97,11 @@ UTensor::getOffset(const IntSequence &v, int nv)
int
UTensor::getOffset(const IntSequence &v, const IntSequence &nvmx)
{
int pow = 1;
int res = 0;
for (int i = v.size()-1; i >= 0; i--)
for (int i = 0; i < v.size(); i++)
{
res += v[i]*pow;
pow *= nvmx[i];
res *= nvmx[i];
res += v[i];
}
return res;
}