xxq250
/
homework-jianmu
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

refactor: do some internal refactor.

This commit is contained in:
Haojun Liao 2023-01-06 17:10:52 +08:00
parent 0966528a2b
commit 7f29a4a62d
1 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
source/util/src/tcompression.c
View File

@ -286,12 +286,9 @@ int32_t tsDecompressINTImp(const char *const input, const int32_t nelements, cha
if (selector == 0 || selector == 1) { if (selector == 0 || selector == 1) {
int32_t batch = elems >> 2; int32_t batch = elems >> 2;
int32_t remainder = elems & 0x3; int32_t remainder = elems & 0x3;
int32_t gBatch = (nelements - count) >> 2;
int32_t gRemainder = nelements - count;
int32_t gBatch = gRemainder >> 2;
int32_t minBatch = TMIN(batch, gBatch); int32_t minBatch = TMIN(batch, gBatch);
int32_t minRemain = TMIN(remainder, gRemainder);
for(int32_t i = 0; i < minBatch; ++i) { for(int32_t i = 0; i < minBatch; ++i) {
p[_pos++] = prev_value; p[_pos++] = prev_value;
p[_pos++] = prev_value; p[_pos++] = prev_value;
@ -299,20 +296,21 @@ int32_t tsDecompressINTImp(const char *const input, const int32_t nelements, cha
p[_pos++] = prev_value; p[_pos++] = prev_value;
} }
count += (minBatch << 2);
int32_t gRemainder = nelements - count;
int32_t minRemain = TMIN(remainder, gRemainder);
for (int32_t i = 0; i < minRemain; i++) { for (int32_t i = 0; i < minRemain; i++) {
p[_pos++] = prev_value; p[_pos++] = prev_value;
} }
count += ((minBatch << 2)+ minRemain); count += minRemain;
} else { } else {
int32_t batch = elems >> 2; int32_t batch = elems >> 2;
int32_t remain = elems & 0x03; int32_t remain = elems & 0x03;
int32_t globalBatch = (nelements - count) >> 2;
int32_t globalRemain = (nelements - count);
int32_t globalBatch = globalRemain >> 2;
int32_t minBatch = TMIN(batch, globalBatch); int32_t minBatch = TMIN(batch, globalBatch);
int32_t minRemain = TMIN(remain, globalRemain);
#if 1 #if 1
// manual unrolling, to erase the hotspot // manual unrolling, to erase the hotspot
for (int32_t i = 0; i < minBatch; ++i) { for (int32_t i = 0; i < minBatch; ++i) {
@ -342,6 +340,10 @@ int32_t tsDecompressINTImp(const char *const input, const int32_t nelements, cha
} }
// handle the remain // handle the remain
count += (minBatch << 2);
int32_t globalRemain = (nelements - count);
int32_t minRemain = TMIN(remain, globalRemain);
for (int32_t i = 0; i < minRemain; i++) { for (int32_t i = 0; i < minRemain; i++) {
zigzag_value = ((w >> v) & mask); zigzag_value = ((w >> v) & mask);
prev_value += ZIGZAG_DECODE(int64_t, zigzag_value); prev_value += ZIGZAG_DECODE(int64_t, zigzag_value);
@ -350,7 +352,7 @@ int32_t tsDecompressINTImp(const char *const input, const int32_t nelements, cha
v += bit; v += bit;
} }
count += ((minBatch << 2)+ minRemain); count += minRemain;
#else #else
for (int32_t i = 0; i < elems && count < nelements; i++, count++) { for (int32_t i = 0; i < elems && count < nelements; i++, count++) {
zigzag_value = ((w >> (4 + v)) & mask); zigzag_value = ((w >> (4 + v)) & mask);