Skip to content

Commit 16c0c2a

Browse files
committed
fix format
1 parent be00662 commit 16c0c2a

9 files changed

Lines changed: 526 additions & 21 deletions

File tree

compiler/luci-interpreter/src/BuddyMemoryManager.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -44,12 +44,13 @@ void BuddyMemoryManager::allocate_memory(luci_interpreter::Tensor &tensor)
4444
{
4545
const size_t element_size = getDataTypeSize(tensor.element_type());
4646
const int64_t num_elements = tensor.shape().large_num_elements();
47-
47+
4848
// Check for integer overflow in size calculation
49-
if (num_elements < 0 || num_elements > SIZE_MAX / element_size) {
49+
if (num_elements < 0 || num_elements > SIZE_MAX / element_size)
50+
{
5051
throw std::runtime_error("Integer overflow in size calculation");
5152
}
52-
53+
5354
const int64_t total_size = num_elements * element_size;
5455
auto size = static_cast<size_t>(total_size);
5556
auto footprint = size + sizeof(Block);

compiler/luci-interpreter/src/kernels/ExpandDims.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -81,12 +81,13 @@ void ExpandDims::execute() const
8181

8282
const size_t element_size = getDataTypeSize(input()->element_type());
8383
const int64_t num_elements = input()->shape().large_num_elements();
84-
84+
8585
// Check for integer overflow in size calculation
86-
if (num_elements < 0 || num_elements > SIZE_MAX / element_size) {
86+
if (num_elements < 0 || num_elements > SIZE_MAX / element_size)
87+
{
8788
throw std::runtime_error("Integer overflow in size calculation");
8889
}
89-
90+
9091
const int64_t total_size = num_elements * element_size;
9192
std::memcpy(output_data, input_data, static_cast<size_t>(total_size));
9293
}

compiler/luci-interpreter/src/kernels/If.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -85,12 +85,13 @@ void If::execute() const
8585

8686
const int64_t num_elements = output(i)->shape().large_num_elements();
8787
const std::size_t element_size = getDataTypeSize(output(i)->element_type());
88-
88+
8989
// Check for integer overflow in size calculation
90-
if (num_elements < 0 || num_elements > SIZE_MAX / element_size) {
90+
if (num_elements < 0 || num_elements > SIZE_MAX / element_size)
91+
{
9192
throw std::runtime_error("Integer overflow in size calculation");
9293
}
93-
94+
9495
const int64_t total_size = num_elements * element_size;
9596
std::memcpy(output(i)->data<void>(), graph_outputs[i]->data<void>(),
9697
static_cast<size_t>(total_size));

compiler/luci-interpreter/src/kernels/Reshape.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -102,12 +102,13 @@ void Reshape::execute() const
102102

103103
const size_t element_size = getDataTypeSize(input()->element_type());
104104
const int64_t num_elements = input()->shape().large_num_elements();
105-
105+
106106
// Check for integer overflow in size calculation
107-
if (num_elements < 0 || num_elements > SIZE_MAX / element_size) {
107+
if (num_elements < 0 || num_elements > SIZE_MAX / element_size)
108+
{
108109
throw std::runtime_error("Integer overflow in size calculation");
109110
}
110-
111+
111112
const int64_t total_size = num_elements * element_size;
112113
std::memcpy(output_data, input_data, static_cast<size_t>(total_size));
113114
}

compiler/luci-interpreter/src/kernels/TransposeConv.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -298,12 +298,13 @@ void TransposeConv::evalQuantizedS16() const
298298

299299
const int64_t num_elements = scratch_tensor->shape().large_num_elements();
300300
const size_t element_size = sizeof(int64_t);
301-
301+
302302
// Check for integer overflow in size calculation
303-
if (num_elements < 0 || num_elements > SIZE_MAX / element_size) {
303+
if (num_elements < 0 || num_elements > SIZE_MAX / element_size)
304+
{
304305
throw std::runtime_error("Integer overflow in size calculation");
305306
}
306-
307+
307308
const int64_t total_size = num_elements * element_size;
308309
std::memset(scratch_data, 0, static_cast<size_t>(total_size));
309310

compiler/luci-interpreter/src/kernels/While.cpp

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -37,12 +37,13 @@ void copy(const std::vector<const Tensor *> &src, const std::vector<Tensor *> &d
3737

3838
const int64_t num_elements = src[i]->shape().large_num_elements();
3939
const std::size_t element_size = getDataTypeSize(src[i]->element_type());
40-
40+
4141
// Check for integer overflow in size calculation
42-
if (num_elements < 0 || num_elements > SIZE_MAX / element_size) {
42+
if (num_elements < 0 || num_elements > SIZE_MAX / element_size)
43+
{
4344
throw std::runtime_error("Integer overflow in size calculation");
4445
}
45-
46+
4647
const int64_t total_size = num_elements * element_size;
4748
std::memcpy(dst[i]->data<void>(), src[i]->data<void>(), static_cast<size_t>(total_size));
4849
}

compiler/luci-interpreter/src/loader/GraphLoader.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,8 @@ template <DataType DT> const void *getNodeDataImpl(const luci::CircleConst *node
4242
const int64_t num_elements = node->size<DT>(); // Assuming size<DT>() uses large_num_elements()
4343

4444
// Check for integer overflow in size calculation
45-
if (num_elements < 0 || num_elements > SIZE_MAX / element_size) {
45+
if (num_elements < 0 || num_elements > SIZE_MAX / element_size)
46+
{
4647
throw std::runtime_error("Integer overflow in size calculation");
4748
}
4849

0 commit comments

Comments
 (0)