diff --git a/onnxruntime/core/providers/cuda/math/binary_elementwise_ops.h b/onnxruntime/core/providers/cuda/math/binary_elementwise_ops.h index fcab87c193..c07324a8ac 100644 --- a/onnxruntime/core/providers/cuda/math/binary_elementwise_ops.h +++ b/onnxruntime/core/providers/cuda/math/binary_elementwise_ops.h @@ -57,7 +57,7 @@ struct BinaryElementwisePreparation { [&C](int64_t dim) { if (dim != 1) C = dim; return (dim != 1); })) { int32_t dim_C = gsl::narrow_cast(std::find(rhs_dims.begin(), rhs_dims.end(), C) - rhs_dims.begin() + output_shape.NumDimensions() - rhs_shape.NumDimensions()); int64_t N = output_shape.SizeToDimension(dim_C); - int64_t H = (dim_C < out_rank - 1 ? output_shape.SizeFromDimension(dim_C + 1) : 1); + int64_t H = (dim_C < out_rank - 1 ? output_shape.SizeFromDimension(static_cast(dim_C) + 1) : 1); std::vector new_output_dims; if (N == 1) {