Make last_dim ptrdiff_t

adityagoel4512 · Jan 11, 2024 · f2f46b5 · f2f46b5
1 parent 0acf472
commit f2f46b5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/onnxruntime/core/providers/cpu/text/string_split.cc b/onnxruntime/core/providers/cpu/text/string_split.cc
@@ -71,12 +71,12 @@ Status StringSplit::Compute(OpKernelContext* context) const {
 
   InlinedVector<InlinedVector<std::string_view>> input_slices;
   input_slices.reserve(input_data.size());
-  int64_t last_dim = 0;
+  ptrdiff_t last_dim = 0;
 
   for (const auto& s : input_data) {
     auto& substrs = input_slices.emplace_back();
     ComputeSubstrings(s, delimiter_, maxsplit_, substrs);
-    auto substr_count = static_cast<int64_t>(substrs.size());
+    auto substr_count = static_cast<ptrdiff_t>(substrs.size());
     last_dim = std::max(last_dim, substr_count);
     *num_tokens_iter = substr_count;
     ++num_tokens_iter;