diff --git a/src/ir.cc b/src/ir.cc index 97b51b5ee..b40723a84 100644 --- a/src/ir.cc +++ b/src/ir.cc @@ -112,8 +112,6 @@ ForFrame PipelinedFor(PrimExpr start, const PrimExpr &stop, int num_stages, anno.Set("tl_pipeline_order", order); if (!stages.empty()) anno.Set("tl_pipeline_stage", stages); - if (!sync.empty()) - anno.Set("tl_pipeline_sync", sync); if (!groups.empty()) anno.Set("tl_pipeline_group", groups); Optional step = diff --git a/tilelang/analysis/nested_loop_checker.py b/tilelang/analysis/nested_loop_checker.py index 51da7f4c8..96f24ed17 100644 --- a/tilelang/analysis/nested_loop_checker.py +++ b/tilelang/analysis/nested_loop_checker.py @@ -11,7 +11,7 @@ def is_pipelined_for(op: For) -> bool: """Check if a for loop is pipelined.""" - anno_keys = ["num_stages", "tl_pipeline_order", "tl_pipeline_stage", "tl_pipeline_sync", "tl_pipeline_group"] + anno_keys = ["num_stages", "tl_pipeline_order", "tl_pipeline_stage", "tl_pipeline_group"] return any(key in op.annotations for key in anno_keys)