InjectCopyIntrin Pass with letstmtnode

hchen · April 12, 2024, 9:51am

When using InjectCopyIntrin， I find that the loop containing letstmtnode cannot be processed，example

Because this situation is not considered in the code

apache/tvm/blob/88a1c6560cb5fe3a757b9b9053bb71421728aedd/src/tir/transforms/inject_copy_intrin.cc#L64


   return StmtMutator::VisitStmt_(op);
 }


private:
 bool MatchCopyPattern(Stmt stmt, Stmt* out, std::string* error_info) {
   using namespace arith;
   Stmt body = stmt;


   // strip the loops
   std::vector<const ForNode*> loops;
   while (const ForNode* op = body.as<ForNode>()) {
     if (!is_zero(op->min)) {
       *error_info = "the 'min' value of body 'Fonode' is 0.";
       return false;
     }
     loops.push_back(op);
     body = op->body;
   }
   auto store = body.as<BufferStoreNode>();
   if (store == nullptr) {
     *error_info = "the body is not a 'BufferStoreNode'";

can anyone give me some advice how to deal with letstmtnode？and how to convert cse_var_1 to (ax2*256) in bufferstorenode?