Fix induction variable type

triton-lang · Jun 7, 2024 · 1438856 · 1438856
1 parent 7a22fc7
commit 1438856
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/third_party/cpu/backend/driver.py b/third_party/cpu/backend/driver.py
@@ -281,7 +281,7 @@ def format_of(ty):
     if (getBoolEnv("TRITON_CPU_OMP_DEBUG"))
       printf("Single core launcher\\n");
 
-    for (uint32_t i = 0; i < N; ++i) {{
+    for (size_t i = 0; i < N; ++i) {{
       const auto [x, y, z] = all_grids[i];
       (*kernel_ptr)({kernel_fn_args_list + ', ' if len(kernel_fn_args) > 0 else ''} x, y, z);
     }}
@@ -299,7 +299,7 @@ def format_of(ty):
 
   // For now, use the default chunk size, total iterations / max_threads.
 #pragma omp parallel for schedule(static) num_threads(max_threads.value())
-  for (uint32_t i = 0; i < N; ++i) {{
+  for (size_t i = 0; i < N; ++i) {{
     const auto [x, y, z] = all_grids[i];
     (*kernel_ptr)({kernel_fn_args_list + ', ' if len(kernel_fn_args) > 0 else ''} x, y, z);
   }}