diff --git a/src/midend/SysYIRGenerator.cpp b/src/midend/SysYIRGenerator.cpp index b84461d..97b25b5 100644 --- a/src/midend/SysYIRGenerator.cpp +++ b/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(currentValue)) { + if (constInt->getInt() == 0 && currentRepeatNum >= 16) { // 阈值可调整（如16、32等） + // 计算 memset 的起始地址（基于当前线性偏移量） + std::vector memsetStartIndices; + int tempLinearIndex = linearIndexOffset; + + // 将线性索引转换为多维索引 + for (int dimIdx = dimSizes.size() - 1; dimIdx >= 0; --dimIdx) { + memsetStartIndices.insert(memsetStartIndices.begin(), + ConstantInteger::get(static_cast(tempLinearIndex % dimSizes[dimIdx]))); + tempLinearIndex /= dimSizes[dimIdx]; + } + + // 构造 GEP 计算 memset 的起始地址 + std::vector 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 currentIndices; int tempLinearIndex = linearIndexOffset + i; // 使用偏移量和当前重复次数内的索引 @@ -761,39 +798,73 @@ std::any SysYIRGenerator::visitVarDecl(SysYParser::VarDeclContext *ctx) { ConstantInteger::get(0)); } else { - + int linearIndexOffset = 0; // 用于追踪当前处理的线性索引的偏移量 for (int k = 0; k < counterValues.size(); ++k) { - // 当前 Value 的值和重复次数 - Value* currentValue = counterValues[k]; - unsigned currentRepeatNum = counterNumbers[k]; + // 当前 Value 的值和重复次数 + Value *currentValue = counterValues[k]; + unsigned currentRepeatNum = counterNumbers[k]; + // 检查是否是0，并且重复次数足够大（例如 >16），才用 memset + if (ConstantInteger *constInt = dynamic_cast(currentValue)) { + if (constInt->getInt() == 0 && currentRepeatNum >= 16) { // 阈值可调整（如16、32等） + // 计算 memset 的起始地址（基于当前线性偏移量） + std::vector memsetStartIndices; + int tempLinearIndex = linearIndexOffset; - for (unsigned i = 0; i < currentRepeatNum; ++i) { - std::vector currentIndices; - int tempLinearIndex = linearIndexOffset + i; // 使用偏移量和当前重复次数内的索引 + // 将线性索引转换为多维索引 + for (int dimIdx = dimSizes.size() - 1; dimIdx >= 0; --dimIdx) { + memsetStartIndices.insert(memsetStartIndices.begin(), + ConstantInteger::get(static_cast(tempLinearIndex % dimSizes[dimIdx]))); + tempLinearIndex /= dimSizes[dimIdx]; + } - // 将线性索引转换为多维索引 - for (int dimIdx = dimSizes.size() - 1; dimIdx >= 0; --dimIdx) { - currentIndices.insert(currentIndices.begin(), - ConstantInteger::get(static_cast(tempLinearIndex % dimSizes[dimIdx]))); - tempLinearIndex /= dimSizes[dimIdx]; - } - - // 对于局部数组，alloca 本身就是 GEP 的基指针。 - // GEP 的第一个索引必须是 0，用于“步过”整个数组。 - std::vector gepIndicesForInit; - gepIndicesForInit.push_back(ConstantInteger::get(0)); - gepIndicesForInit.insert(gepIndicesForInit.end(), currentIndices.begin(), currentIndices.end()); - - // 计算元素的地址 - Value* elementAddress = getGEPAddressInst(alloca, gepIndicesForInit); - // 生成 store 指令 - builder.createStoreInst(currentValue, elementAddress); + // 构造 GEP 计算 memset 的起始地址 + std::vector 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; // 直接进入下一次循环 } - // 更新线性索引偏移量，以便下一次迭代从正确的位置开始 - linearIndexOffset += currentRepeatNum; - } + } + for (unsigned i = 0; i < currentRepeatNum; ++i) { + std::vector currentIndices; + int tempLinearIndex = linearIndexOffset + i; // 使用偏移量和当前重复次数内的索引 + // 将线性索引转换为多维索引 + for (int dimIdx = dimSizes.size() - 1; dimIdx >= 0; --dimIdx) { + currentIndices.insert(currentIndices.begin(), + ConstantInteger::get(static_cast(tempLinearIndex % dimSizes[dimIdx]))); + tempLinearIndex /= dimSizes[dimIdx]; + } + + // 对于局部数组，alloca 本身就是 GEP 的基指针。 + // GEP 的第一个索引必须是 0，用于“步过”整个数组。 + std::vector gepIndicesForInit; + gepIndicesForInit.push_back(ConstantInteger::get(0)); + gepIndicesForInit.insert(gepIndicesForInit.end(), currentIndices.begin(), currentIndices.end()); + + // 计算元素的地址 + 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(visitExp(exp)); + Value* dimVal = computeExp(exp); // 确保维度是常量整数，否则 buildArrayType 会断言失败 assert(dynamic_cast(dimVal) && "Array dimension in parameter must be a constant integer!"); currentParamDims.push_back(dimVal);