(tvm) branch refactor-s1 updated: Fix MS trace

[tqchen]

This is an automated email from the ASF dual-hosted git repository.

tqchen pushed a commit to branch refactor-s1
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/refactor-s1 by this push:
     new 88a2f9a72f Fix MS trace
88a2f9a72f is described below

commit 88a2f9a72fe43624bc82740822e01e5f61bc130f
Author: tqchen <tianqi.tc...@gmail.com>
AuthorDate: Wed Apr 16 12:13:05 2025 -0400

    Fix MS trace
---
 .../postproc/rewrite_parallel_vectorize_unroll.cc        | 16 ++++++++--------
 1 file changed, 8 insertions(+), 8 deletions(-)

diff --git a/src/meta_schedule/postproc/rewrite_parallel_vectorize_unroll.cc 
b/src/meta_schedule/postproc/rewrite_parallel_vectorize_unroll.cc
index 9779603f90..d97c39d420 100644
--- a/src/meta_schedule/postproc/rewrite_parallel_vectorize_unroll.cc
+++ b/src/meta_schedule/postproc/rewrite_parallel_vectorize_unroll.cc
@@ -106,23 +106,23 @@ bool ParseAnnotation(const Block& block, 
ParsedAnnotation* parsed) {
   for (const auto& ann : block->annotations) {
     if (ann.first == attr::meta_schedule_parallel) {
       found = true;
-      if (const auto* imm = ann.second.as<tir::IntImmNode>()) {
-        parsed->max_parallel_extent = imm->value;
+      if ( auto opt_int_imm = ann.second.as<IntImm>()) {
+        parsed->max_parallel_extent = (*opt_int_imm)->value;
       }
     } else if (ann.first == attr::meta_schedule_vectorize) {
       found = true;
-      if (const auto* imm = ann.second.as<tir::IntImmNode>()) {
-        parsed->max_vectorize_extent = imm->value;
+      if (auto opt_int_imm = ann.second.as<IntImm>()) {
+        parsed->max_vectorize_extent = (*opt_int_imm)->value;
       }
     } else if (ann.first == attr::meta_schedule_unroll_explicit) {
       found = true;
-      if (const auto* imm = ann.second.as<tir::IntImmNode>()) {
-        parsed->unroll_explicit = imm->value;
+      if (auto opt_int_imm = ann.second.as<IntImm>()) {
+        parsed->unroll_explicit = (*opt_int_imm)->value;
       }
     } else if (ann.first == attr::meta_schedule_unroll_implicit) {
       found = true;
-      if (const auto* imm = ann.second.as<tir::IntImmNode>()) {
-        parsed->unroll_implicit = imm->value;
+      if (auto opt_int_imm = ann.second.as<IntImm>()) {
+        parsed->unroll_implicit = (*opt_int_imm)->value;
       }
     }
   }