Correct assertion in PlainTensor member functions (openvinotoolkit#21199

) - PlainTensor::size - PlainTensor::stride
akuporos · Nov 22, 2023 · 97381e0 · 97381e0
1 parent d85e6c3
commit 97381e0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/plugins/intel_cpu/src/utils/plain_tensor.hpp b/src/plugins/intel_cpu/src/utils/plain_tensor.hpp
@@ -131,11 +131,11 @@ struct PlainTensor {
     size_t size(int i) const {
         if (i < 0)
             i += m_rank;
-        assert(i < m_rank);
+        assert(static_cast<typename std::make_unsigned<decltype(i)>::type>(i) < m_rank);
         return m_dims[i];
     }
     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];
     }
     PlainTensor(MemoryPtr mem) {