Disable specific schedule in MetaSchedule?

reikdas · July 17, 2025, 4:04pm

Specifically, is there a way to force MetaSchedule to ignore the parallelize primitive from its search space?

ConvolutedDog · July 17, 2025, 4:49pm

There is no option to turn it off for now, but you can manually comment out the relevant schedule rule and recompile TVM:

github.com/apache/tvm

src/meta_schedule/schedule_rule/parallel_vectorize_unroll.cc

main


      
          // Parallelization
          if (max_jobs_per_core != -1) {
            sch->Annotate(root_rv, tir::attr::meta_schedule_parallel,
                          Integer(this->max_parallel_extent_));
          }

Also if you know the code of meta_schedule, you can add an option to it (extending from the front-end Python to the back-end C++).