diff --git a/python/triton/compiler/compiler.py b/python/triton/compiler/compiler.py
index 0f7d09a8d469..cd33b5674d7f 100644
--- a/python/triton/compiler/compiler.py
+++ b/python/triton/compiler/compiler.py
@@ -164,7 +164,7 @@ def optimize_ttgir(mod, num_stages, num_warps, num_ctas, target, cluster_info, e
     pm.add_tritongpu_remove_layout_conversions_pass()
     pm.add_tritongpu_decompose_conversions_pass()
     pm.add_tritongpu_ws_fixup_missing_attrs_pass()
-    if is_hip() and num_stages != 0:
+    if is_hip():
         pm.add_tritonamdgpu_reorder_instructions_pass()
     else:
         pm.add_tritongpu_reorder_instructions_pass()