fix implicit conversion in TensorChipping

This commit is contained in:
Charles Schlosser 2024-09-24 16:58:49 +00:00 committed by Rasmus Munk Larsen
parent b6b8b54e5e
commit 3ffb4e50df

View File

@ -177,7 +177,7 @@ struct TensorEvaluator<const TensorChippingOp<DimId, ArgType>, Device> {
// Check if chipping is effectively inner or outer: products of dimensions // Check if chipping is effectively inner or outer: products of dimensions
// before or after the chipped dimension is `1`. // before or after the chipped dimension is `1`.
Index after_chipped_dim_product = 1; Index after_chipped_dim_product = 1;
for (int i = m_dim.actualDim() + 1; i < NumInputDims; ++i) { for (int i = static_cast<int>(m_dim.actualDim()) + 1; i < NumInputDims; ++i) {
after_chipped_dim_product *= input_dims[i]; after_chipped_dim_product *= input_dims[i];
} }