newell
/
nac3
forked from M-Labs/nac3
1
0
Fork 0
Code Pull Requests Activity

core: Update __builtin_assume expressions 

No dimension size should be 0.
This commit is contained in:
David Mak 2024-03-07 13:01:06 +08:00
parent ebf86cd134
commit a94927a11d
1 changed files with 8 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
nac3core/src/codegen/irrt/irrt.c
View File

@ -205,8 +205,8 @@ uint32_t __nac3_ndarray_calc_size(
uint32_t num_elems = 1; uint32_t num_elems = 1;
for (uint32_t i = 0; i < list_len; ++i) { for (uint32_t i = 0; i < list_len; ++i) {
uint64_t val = list_data[i]; uint64_t val = list_data[i];
__builtin_assume(val >= 0); __builtin_assume(val > 0);
num_elems *= list_data[i]; num_elems *= val;
} }
return num_elems; return num_elems;
} }
@ -218,8 +218,8 @@ uint64_t __nac3_ndarray_calc_size64(
uint64_t num_elems = 1; uint64_t num_elems = 1;
for (uint64_t i = 0; i < list_len; ++i) { for (uint64_t i = 0; i < list_len; ++i) {
uint64_t val = list_data[i]; uint64_t val = list_data[i];
__builtin_assume(val >= 0); __builtin_assume(val > 0);
num_elems *= list_data[i]; num_elems *= val;
} }
return num_elems; return num_elems;
} }
@ -233,6 +233,7 @@ void __nac3_ndarray_calc_nd_indices(
uint32_t stride = 1; uint32_t stride = 1;
for (uint32_t dim = 0; dim < num_dims; dim++) { for (uint32_t dim = 0; dim < num_dims; dim++) {
uint32_t i = num_dims - dim - 1; uint32_t i = num_dims - dim - 1;
__builtin_assume(dims[i] > 0);
idxs[i] = (index / stride) % dims[i]; idxs[i] = (index / stride) % dims[i];
stride *= dims[i]; stride *= dims[i];
} }
@ -247,6 +248,7 @@ void __nac3_ndarray_calc_nd_indices64(
uint64_t stride = 1; uint64_t stride = 1;
for (uint64_t dim = 0; dim < num_dims; dim++) { for (uint64_t dim = 0; dim < num_dims; dim++) {
uint64_t i = num_dims - dim - 1; uint64_t i = num_dims - dim - 1;
__builtin_assume(dims[i] > 0);
idxs[i] = (index / stride) % dims[i]; idxs[i] = (index / stride) % dims[i];
stride *= dims[i]; stride *= dims[i];
} }
@ -266,6 +268,7 @@ uint32_t __nac3_ndarray_flatten_index(
idx += (stride * indices[ri]); idx += (stride * indices[ri]);
} }
__builtin_assume(dims[i] > 0);
stride *= dims[ri]; stride *= dims[ri];
} }
return idx; return idx;
@ -285,6 +288,7 @@ uint64_t __nac3_ndarray_flatten_index64(
idx += (stride * indices[ri]); idx += (stride * indices[ri]);
} }
__builtin_assume(dims[i] > 0);
stride *= dims[ri]; stride *= dims[ri];
} }
return idx; return idx;