Correct assertion in PlainTensor member functions (#21199)

- PlainTensor::size
- PlainTensor::stride
This commit is contained in:
Pawel Raasz authored and GitHub committed 2023-11-22 09:41:53 +01:00
1 parent d85e6c34a7
commit 97381e0b63
1 file changed
+2 -2
@@ -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) {