IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Correct assertion in PlainTensor member functions (#21199)

Browse Source 
- PlainTensor::size
- PlainTensor::stride
This commit is contained in:
Pawel Raasz 2023-11-22 09:41:53 +01:00 committed by GitHub
parent d85e6c34a7
commit 97381e0b63
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/plugins/intel_cpu/src/utils/plain_tensor.hpp
View File

@ -131,11 +131,11 @@ struct PlainTensor {
size_t size(int i) const { size_t size(int i) const {
if (i < 0) if (i < 0)
i += m_rank; i += m_rank;
assert(i < m_rank); assert(static_cast<typename std::make_unsigned<decltype(i)>::type>(i) < m_rank);
return m_dims[i]; return m_dims[i];
} }
size_t stride(int i) const { size_t stride(int i) const {
assert(i < m_rank); assert(i >= 0 && static_cast<typename std::make_unsigned<decltype(i)>::type>(i) < m_rank);
return m_strides[i]; return m_strides[i];
} }
PlainTensor(MemoryPtr mem) { PlainTensor(MemoryPtr mem) {