Merge branch 'midend' into backend-IRC

src/midend/SysYIRGenerator.cpp

@@ -653,7 +653,44 @@ std::any SysYIRGenerator::visitConstDecl(SysYParser::ConstDeclContext *ctx) {
           Value *currentValue = counterValues[k];
           unsigned currentRepeatNum = counterNumbers[k];
 
+          // 检查是否是0，并且重复次数足够大（例如 >16），才用 memset
+          if (ConstantInteger *constInt = dynamic_cast<ConstantInteger *>(currentValue)) {
+            if (constInt->getInt() == 0 && currentRepeatNum >= 16) { // 阈值可调整（如16、32等）
+              // 计算 memset 的起始地址（基于当前线性偏移量）
+              std::vector<Value *> memsetStartIndices;
+              int tempLinearIndex = linearIndexOffset;
+
+              // 将线性索引转换为多维索引
+              for (int dimIdx = dimSizes.size() - 1; dimIdx >= 0; --dimIdx) {
+                memsetStartIndices.insert(memsetStartIndices.begin(),
+                                          ConstantInteger::get(static_cast<int>(tempLinearIndex % dimSizes[dimIdx])));
+                tempLinearIndex /= dimSizes[dimIdx];
+              }
+
+              // 构造 GEP 计算 memset 的起始地址
+              std::vector<Value *> gepIndicesForMemset;
+              gepIndicesForMemset.push_back(ConstantInteger::get(0)); // 跳过 alloca 类型
+              gepIndicesForMemset.insert(gepIndicesForMemset.end(), memsetStartIndices.begin(),
+                                         memsetStartIndices.end());
+
+              Value *memsetPtr = builder.createGetElementPtrInst(alloca, gepIndicesForMemset);
+
+              // 计算 memset 的字节数 = 元素个数 × 元素大小
+              Type *elementType = type;;
+              uint64_t elementSize = elementType->getSize();
+              Value *size = ConstantInteger::get(currentRepeatNum * elementSize);
+
+              // 生成 memset 指令（假设你的 IRBuilder 有 createMemset 方法）
+              builder.createMemsetInst(memsetPtr, ConstantInteger::get(0), size, ConstantInteger::get(0));
+
+              // 跳过这些已处理的0
+              linearIndexOffset += currentRepeatNum;
+              continue; // 直接进入下一次循环
+            }
+          }
+
           for (unsigned i = 0; i < currentRepeatNum; ++i) {
+            // 对于非零值，生成对应的 store 指令
             std::vector<Value *> currentIndices;
             int tempLinearIndex = linearIndexOffset + i; // 使用偏移量和当前重复次数内的索引
 
@@ -765,9 +802,44 @@ std::any SysYIRGenerator::visitVarDecl(SysYParser::VarDeclContext *ctx) {
           int linearIndexOffset = 0; // 用于追踪当前处理的线性索引的偏移量
           for (int k = 0; k < counterValues.size(); ++k) {
             // 当前 Value 的值和重复次数
-              Value* currentValue = counterValues[k];
+            Value *currentValue = counterValues[k];
             unsigned currentRepeatNum = counterNumbers[k];
+            // 检查是否是0，并且重复次数足够大（例如 >16），才用 memset
+            if (ConstantInteger *constInt = dynamic_cast<ConstantInteger *>(currentValue)) {
+              if (constInt->getInt() == 0 && currentRepeatNum >= 16) { // 阈值可调整（如16、32等）
+                // 计算 memset 的起始地址（基于当前线性偏移量）
+                std::vector<Value *> memsetStartIndices;
+                int tempLinearIndex = linearIndexOffset;
 
+                // 将线性索引转换为多维索引
+                for (int dimIdx = dimSizes.size() - 1; dimIdx >= 0; --dimIdx) {
+                  memsetStartIndices.insert(memsetStartIndices.begin(),
+                                            ConstantInteger::get(static_cast<int>(tempLinearIndex % dimSizes[dimIdx])));
+                  tempLinearIndex /= dimSizes[dimIdx];
+                }
+
+                // 构造 GEP 计算 memset 的起始地址
+                std::vector<Value *> gepIndicesForMemset;
+                gepIndicesForMemset.push_back(ConstantInteger::get(0)); // 跳过 alloca 类型
+                gepIndicesForMemset.insert(gepIndicesForMemset.end(), memsetStartIndices.begin(),
+                                           memsetStartIndices.end());
+
+                Value *memsetPtr = builder.createGetElementPtrInst(alloca, gepIndicesForMemset);
+
+                // 计算 memset 的字节数 = 元素个数 × 元素大小
+                Type *elementType = type;
+                ;
+                uint64_t elementSize = elementType->getSize();
+                Value *size = ConstantInteger::get(currentRepeatNum * elementSize);
+
+                // 生成 memset 指令（假设你的 IRBuilder 有 createMemset 方法）
+                builder.createMemsetInst(memsetPtr, ConstantInteger::get(0), size, ConstantInteger::get(0));
+
+                // 跳过这些已处理的0
+                linearIndexOffset += currentRepeatNum;
+                continue; // 直接进入下一次循环
+              }
+            }
             for (unsigned i = 0; i < currentRepeatNum; ++i) {
               std::vector<Value *> currentIndices;
               int tempLinearIndex = linearIndexOffset + i; // 使用偏移量和当前重复次数内的索引
@@ -781,19 +853,18 @@ std::any SysYIRGenerator::visitVarDecl(SysYParser::VarDeclContext *ctx) {
 
               // 对于局部数组，alloca 本身就是 GEP 的基指针。
               // GEP 的第一个索引必须是 0，用于“步过”整个数组。
-                  std::vector<Value*> gepIndicesForInit;
+              std::vector<Value *> gepIndicesForInit;
               gepIndicesForInit.push_back(ConstantInteger::get(0));
               gepIndicesForInit.insert(gepIndicesForInit.end(), currentIndices.begin(), currentIndices.end());
 
               // 计算元素的地址
-                  Value* elementAddress = getGEPAddressInst(alloca, gepIndicesForInit);
+              Value *elementAddress = getGEPAddressInst(alloca, gepIndicesForInit);
               // 生成 store 指令
               builder.createStoreInst(currentValue, elementAddress);
             }
             // 更新线性索引偏移量，以便下一次迭代从正确的位置开始
             linearIndexOffset += currentRepeatNum;
           }
-
         }
       }
     }
@@ -895,7 +966,7 @@ std::any SysYIRGenerator::visitFuncDef(SysYParser::FuncDefContext *ctx){
         currentParamDims.push_back(ConstantInteger::get(-1)); // 标记第一个维度为未知
         for (const auto &exp : param->exp()) {
           // 访问表达式以获取维度大小，这些维度必须是常量
-          Value* dimVal = std::any_cast<Value *>(visitExp(exp));
+          Value* dimVal = computeExp(exp);
           // 确保维度是常量整数，否则 buildArrayType 会断言失败
           assert(dynamic_cast<ConstantInteger*>(dimVal) && "Array dimension in parameter must be a constant integer!");
           currentParamDims.push_back(dimVal);