Skip to content

Commit 68a98dc

Browse files
Merge pull request #14 from sundw2014/master
fix a bug of checking dimensions in NEGEMMTranspose1xWKernel.cpp
2 parents 12fcfd5 + e5a267b commit 68a98dc

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/core/NEON/kernels/NEGEMMTranspose1xWKernel.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -45,9 +45,9 @@ void NEGEMMTranspose1xWKernel::configure(const ITensor *input, ITensor *output)
4545
ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(input, 1, DataType::U8, DataType::F16, DataType::F32);
4646
ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(output, 1, DataType::U8, DataType::F16, DataType::F32);
4747
ARM_COMPUTE_ERROR_ON_MISMATCHING_DATA_TYPES(input, output);
48-
ARM_COMPUTE_ERROR_ON((output->info()->dimension(1) != std::ceil(input->info()->dimension(0)) / 8.0f) && (input->info()->data_type() == DataType::F16));
49-
ARM_COMPUTE_ERROR_ON((output->info()->dimension(1) != std::ceil(input->info()->dimension(0)) / 4.0f) && (input->info()->data_type() == DataType::F32));
50-
ARM_COMPUTE_ERROR_ON((output->info()->dimension(1) != std::ceil(input->info()->dimension(0)) / 4.0f) && (input->info()->data_type() == DataType::U32));
48+
ARM_COMPUTE_ERROR_ON((output->info()->dimension(1) != std::ceil(input->info()->dimension(0) / 8.0f)) && (input->info()->data_type() == DataType::F16));
49+
ARM_COMPUTE_ERROR_ON((output->info()->dimension(1) != std::ceil(input->info()->dimension(0) / 4.0f)) && (input->info()->data_type() == DataType::F32));
50+
ARM_COMPUTE_ERROR_ON((output->info()->dimension(1) != std::ceil(input->info()->dimension(0) / 4.0f)) && (input->info()->data_type() == DataType::U32));
5151

5252
unsigned int num_elems_processed_per_iteration(0);
5353
switch(input->info()->data_type())

0 commit comments

Comments
 (0)