Fix if returned index exceeds 32-bit in TensileLite client (#1538)

ROCm · Jan 14, 2025 · ce603f2 · ce603f2
1 parent 579164a
commit ce603f2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tensilelite/Tensile/Source/lib/include/Tensile/TensorDescriptor.hpp b/tensilelite/Tensile/Source/lib/include/Tensile/TensorDescriptor.hpp
@@ -371,7 +371,7 @@ namespace TensileLite
                 if(indices[i] >= m_sizes[i])
                     throw std::runtime_error("Index out of bounds.");
 
-            return std::inner_product(indices.begin(), indices.end(), m_strides.begin(), 0);
+            return std::inner_product(indices.begin(), indices.end(), m_strides.begin(), (size_t)0);
         }
 
         template <typename T>
@@ -385,7 +385,7 @@ namespace TensileLite
                 if(*i.first >= *i.second)
                     throw std::runtime_error("Index out of bounds.");
 
-            return std::inner_product(indices.begin(), indices.end(), m_strides.begin(), 0);
+            return std::inner_product(indices.begin(), indices.end(), m_strides.begin(), (size_t)0);
         }
 
         template <class... Ts,