[xla:cpu] Fix mac os compilation error part 2

PiperOrigin-RevId: 719022795
openxla · Jan 23, 2025 · e9a1a97 · e9a1a97
1 parent 53c999d
commit e9a1a97
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xla/backends/cpu/runtime/kernel.cc b/xla/backends/cpu/runtime/kernel.cc
@@ -98,7 +98,7 @@ Kernel::ParallelTask<thread_dim_x_only>::ParallelTask(
 
 template <bool thread_dim_x_only>
 absl::Status Kernel::ParallelTask<thread_dim_x_only>::operator()(
-    uint64_t task_index) const {
+    size_t task_index) const {
   DCHECK_LT(task_index, num_tasks_) << "Task index out of range";  // Crash OK
 
   XLA_CPU_KernelThread kernel_thread = Delinearize(task_index);