Merge pull request #627 from OpenXiangShan/tage-indexNew2-align

cpu-o3: update tage index/bank/tag use startPC 1 bit shift instead of 5 bit shift

Author: jensen-yan

docs/Gem5_Docs/frontend/TAGE_BANK_IMPLEMENTATION_PLAN.md

@@ -22,12 +22,7 @@
 ## 设计决策
 
 ### 1. Bank分区方案
-```
-blockSize = 32B (2^5)
-pc[4:0]   - 忽略（32B块内偏移）
-pc[6:5]   - bank_id (4个bank, NumBanks=4)
-pc[N:7]   - 用于计算table index（与folded history XOR）
-```
+以 XiangShan 语义对齐：先去掉 `instShiftAmt`（目前为 1）个指令粒度位，再取随后的 `ceilLog2(numBanks)` 位作为 bank id，剩余更高位与 folded history 组合得到 index / tag。这样可以直接利用 `startPC` 的最低有效位，默认只屏蔽半字节对齐，并允许在未来按需扩展 bank 颗粒度。
 
 ### 2. 周期模型（已确认实际调用顺序）✅
 - 一个"周期"内：**先调用 `update(块B)` 进行更新，后调用 `putPCHistory(块A)` 进行预测**
@@ -72,213 +67,17 @@ void putPCHistory() {
 
 ### 3. 数据结构
 - **保持三维数组**：`tageTable[table][index][way]`
-- **不改为四维**：避免大规模重构，在计算index时分离bank信息
-- bank信息通过计算函数隐式处理
+- **不改为四维**：在 index/tag 计算时剥离 bank 比特即可
+- bank 信息通过 `bankBaseShift`/`bankIdWidth` 描述，内联函数统一取用
 
 ---
 
 ## 修改清单
 
-### 文件1: `btb_tage.hh`
-
-#### A. 添加Bank相关成员变量（已参数化）✅
-```cpp
-class BTBTAGE {
-private:
-    // Bank configuration (参数化，从BranchPredictor.py传入)
-    const unsigned numBanks;         // Number of banks (e.g., 4)
-    const unsigned bankIdWidth;      // log2(numBanks), computed in constructor
-    const unsigned bankIdShift;      // floorLog2(blockSize), e.g., 5 for 32B blocks
-    const unsigned indexShift;       // bankIdShift + bankIdWidth when enabled; fallback uses bankIdShift
-
-    // Track last prediction bank for conflict detection
-    unsigned lastPredBankId;         // Bank ID of last prediction
-    bool predBankValid;              // Whether lastPredBankId is valid
-```
-
-#### B. 添加Bank相关统计
-```cpp
-struct TageStats {
-    // ... existing stats ...
-
-    Scalar updateBankConflict;       // Bank冲突次数
-    Scalar updateDroppedDueToConflict; // 因冲突丢弃的更新次数
-}
-```
-
-#### C. 添加Bank计算函数声明
-```cpp
-private:
-    // Get bank ID from aligned PC
-    // Extract pc[bankIdShift+bankIdWidth-1 : bankIdShift]
-    // For 32B blocks with 4 banks: pc[6:5]
-    unsigned getBankId(Addr alignedPC) const;
-```
-
----
-
-### 文件2: `btb_tage.cc`
-
-#### A. 实现Bank计算函数（参数化版本）✅
-```cpp
-unsigned
-BTBTAGE::getBankId(Addr alignedPC) const
-{
-    // Extract bank ID bits from aligned PC
-    // bankIdShift is the starting bit position (5 for 32B blocks)
-    // bankIdWidth is the number of bits (2 for 4 banks)
-    // Example: pc[6:5] for 32B blocks with 4 banks
-    return (alignedPC >> bankIdShift) & ((1 << bankIdWidth) - 1);
-}
-```
-
-#### B. 修改 `getTageIndex()` 函数（参数化版本）✅
-**原实现**：
-```cpp
-Addr getTageIndex(Addr pc, int t, uint64_t foldedHist) {
-    Addr mask = (1ULL << tableIndexBits[t]) - 1;
-    Addr pcBits = (pc >> floorLog2(blockSize)) & mask;  // pc >> 5
-    Addr foldedBits = foldedHist & mask;
-    return pcBits ^ foldedBits;
-}
-```
-
-**修改为（依据 enableBankConflict 动态选择 shift）**：
-```cpp
-Addr
-BTBTAGE::getTageIndex(Addr pc, int t, uint64_t foldedHist)
-{
-    // Create mask for tableIndexBits[t] to limit result size
-    Addr mask = (1ULL << tableIndexBits[t]) - 1;
-
-    // Index calculation skips bank bits to avoid bank aliasing
-    // For 32B blocks (5 bits) with 4 banks (2 bits):
-    //   - pc[4:0]: block offset (ignored)
-    //   - pc[6:5]: bank ID (skipped)
-    //   - pc[N:7]: used for index calculation
-    // Each bank effectively manages 1/4 of the PC space with the same table size
-    const unsigned pcShift = enableBankConflict ? indexShift : bankIdShift;
-    Addr pcBits = (pc >> pcShift) & mask;  // Skip blockSize + bank bits only when enabled
-    Addr foldedBits = foldedHist & mask;
-
-    return pcBits ^ foldedBits;
-}
-```
-
-**重要说明**：
-- index计算跳过了bank位，避免bank混叠
-- 每个bank管理不同的PC空间范围
-- 4个bank共享tableSizes，总容量不变
-
-#### C. 修改 `putPCHistory()` - 记录预测的bank
-```cpp
-void
-BTBTAGE::putPCHistory(Addr stream_start, const bitset &history,
-                      std::vector<FullBTBPrediction> &stagePreds) {
-    Addr alignedPC = stream_start & ~(blockSize - 1);
-
-    // Record prediction bank for conflict detection
-    lastPredBankId = getBankId(alignedPC);
-    predBankValid = true;
-
-    DPRINTF(TAGE, "putPCHistory startAddr: %#lx, alignedPC: %#lx, bank: %u\n",
-            stream_start, alignedPC, lastPredBankId);
-
-    // ... rest of the function remains same ...
-}
-```
-
-#### D. 修改 `update()` - 检测bank冲突
-```cpp
-void
-BTBTAGE::update(const FetchStream &stream) {
-    Addr startAddr = stream.getRealStartPC();
-    Addr alignedPC = startAddr & ~(blockSize - 1);
-    unsigned updateBank = getBankId(alignedPC);
-
-    DPRINTF(TAGE, "update startAddr: %#lx, alignedPC: %#lx, bank: %u\n",
-            startAddr, alignedPC, updateBank);
-
-    // Check bank conflict
-    if (predBankValid && updateBank == lastPredBankId) {
-        tageStats.updateBankConflict++;
-        tageStats.updateDroppedDueToConflict++;
-        DPRINTF(TAGE, "Bank conflict detected: bank %u, dropping update\n", updateBank);
-        return;  // Drop this update
-    }
-
-    // ... rest of the function remains same ...
-}
-```
-
-#### E. 初始化Bank状态（构造函数）
-```cpp
-BTBTAGE::BTBTAGE(...)
-    : ...,
-      lastPredBankId(0),
-      predBankValid(false),
-      ...
-{
-    // ... existing initialization ...
-}
-```
-
-#### F. 统计初始化（TageStats构造函数）
-```cpp
-BTBTAGE::TageStats::TageStats(...)
-    : ...,
-      ADD_STAT(updateBankConflict, statistics::units::Count::get(),
-               "Number of bank conflicts detected"),
-      ADD_STAT(updateDroppedDueToConflict, statistics::units::Count::get(),
-               "Number of updates dropped due to bank conflict"),
-      ...
-{
-    // ... existing stats ...
-}
-```
-
-#### G. 强制 updateOnRead = true, 通过添加warn 的额方式
-在构造函数中：
-```cpp
-#ifndef UNIT_TEST
-BTBTAGE::BTBTAGE(const Params& p):
-    ...,
-    updateOnRead(p.updateOnRead)
-    // Add warning if parameter was set differently
-    if (!p.updateOnRead) {
-        warn("BTBTAGE: updateOnRead forced to true for bank simulation");
-    }
-```
-
----
-
-## 实施步骤（全部完成）✅
-
-### 阶段1: 基础Bank计算 ✅
-1. ✅ 添加参数化的 `getBankId()` 函数
-2. ✅ 在 `putPCHistory()` 和 `update()` 中调用并打印bank信息
-3. ✅ 验证bank计算正确性（debug输出显示bank ID正确）
-
-### 阶段2: 修改index计算 ✅
-1. ✅ 修改 `getTageIndex()`：在启用 bank 模拟时使用 `indexShift`，关闭时退回 `pc >> floorLog2(blockSize)`
-2. ✅ 运行测试，确认功能正确性（编译通过，运行成功）
-
-### 阶段3: 添加bank冲突检测 ✅
-1. ✅ 添加 `lastPredBankId` 和 `predBankValid` 状态
-2. ✅ 在 `putPCHistory()` 中记录预测bank
-3. ✅ 在 `update()` 中检测冲突并丢弃更新（方案2：结束时清空）
-4. ✅ 添加统计counter（updateBankConflict, updateDroppedDueToConflict）
-
-### 阶段4: updateOnRead警告 ✅
-1. ✅ 添加warning信息（当updateOnRead=false时）
-
-### 阶段5: 测试与验证 ✅
-1. ✅ 编译成功（gem5.debug，365MB）
-2. ✅ 运行dummy测试，观察到：
-   - Bank ID计算正确（0x800000c0→bank 2, 0x80000100→bank 0）
-   - 检测到bank冲突（tage.updateBankConflict: 1）
-   - 丢弃更新正常（tage.updateDroppedDueToConflict: 1）
-3. ✅ Debug trace显示预测和更新的bank信息
+### 调整记录（概要）
+- `btb_tage.hh` 中新增 `blockWidth` 与 `bankBaseShift`，前者仍等于 `floorLog2(blockSize)`，后者默认为 `instShiftAmt`，用于剥离指令对齐位；`indexShift` 计算依赖上述两个常量。
+- `getBankId()`/`getTageIndex()`/`getTageTag()` 等 helper 统一使用这些新偏移，确保预测与更新都以同样的 `startPC` 语义工作；bank 位只在启用冲突模拟时跳过，其余情况下维持 legacy 行为。
+- `putPCHistory()`/`update()` 记录 bank id 时直接基于 `startPC`，减少额外对齐开销，统计项维持不变。
 
 ---
 
@@ -342,114 +141,3 @@ private:
     }
 }
 ```
-
----
-
-## 验证标准（已验证）✅
-
-### 功能正确性 ✅
-- ✅ **bank计算正确**：debug输出显示bank ID符合预期
-  ```
-  0x800000c0 (pc[6:5]=11b) -> bank 2 ✓
-  0x80000100 (pc[6:5]=00b) -> bank 0 ✓
-  0x80000140 (pc[6:5]=10b) -> bank 2 ✓
-  0x80000180 (pc[6:5]=00b) -> bank 0 ✓
-  ```
-- ✅ **index计算正确**：启用bank模拟时跳过 bank 位，关闭后退回旧逻辑
-- ✅ **冲突检测正确**：统计显示检测到bank冲突并丢弃更新
-
-### 性能指标 ✅
-- ✅ **bank冲突统计正常**：
-  ```
-  tage.updateBankConflict: 1
-  tage.updateDroppedDueToConflict: 1
-  microtage.updateBankConflict: 0
-  ```
-- ✅ **统计项正常工作**：所有新增统计项均正确记录
-
-### 实际测试结果 ✅
-```bash
-# 测试程序：dummy-riscv64-xs.bin
-# 编译：gem5.debug (365MB, 2025-11-19 18:42)
-# 运行：成功完成，无崩溃
-
-# Debug输出示例：
-putPCHistory startAddr: 0x800000c0, alignedPC: 0x800000c0, bank: 2
-putPCHistory startAddr: 0x80000100, alignedPC: 0x80000100, bank: 0
-putPCHistory startAddr: 0x80000140, alignedPC: 0x80000140, bank: 2
-
-# 统计结果：
-system.cpu.branchPred.tage.updateBankConflict: 1
-system.cpu.branchPred.tage.updateDroppedDueToConflict: 1
-```
-
----
-
-## 参数配置（BranchPredictor.py）✅
-
-**已实现的参数**：
-```python
-class BTBTAGE(TimedBaseBTBPredictor):
-    # ... existing params ...
-
-    # Bank configuration
-    numBanks = Param.Unsigned(4, "Number of banks for bank conflict simulation")
-    numDelay = 2
-```
-
-**使用方式**：
-- 默认配置：`numBanks = 4`，启用bank冲突模拟
-- 修改bank数量：在配置文件中设置不同的numBanks值
-- ⚠️ 当前实现总是启用bank冲突检测（无独立开关）
-
-**未来扩展**（可选）：
-```python
-enable_bank_simulation = Param.Bool(True, "Enable bank conflict simulation")
-```
-如果添加此参数并设为False，可以：
-- `getBankId()` 总是返回0
-- 不检测bank冲突
-- 行为退化为原来的版本
-
----
-
-## 代码统计（实际）✅
-- **新增代码**：约130行（包括注释）
-- **修改代码**：约40行
-- **修改文件**：3个（btb_tage.hh, btb_tage.cc, BranchPredictor.py）
-- **编译时间**：约2-3分钟（-j64，服务器配置良好）
-
-## 实际完成时间 ✅
-- **阶段1-4**：实际编码时间约1小时
-- **阶段5**：编译+测试约10分钟
-- **总计**：约1.5小时（2025-11-19完成）
-
----
-
-## 后续建议
-
-### 1. 运行更多测试
-```bash
-# 运行完整的SPEC测试观察冲突率
-python3 debug/run_cpt.py --debug-dir debug/bank_final
-grep 'tage.*BankConflict' debug/bank_final/*/stats.txt
-
-# 统计平均冲突率
-grep 'updateBankConflict' debug/bank_final/*/stats.txt | awk '{sum+=$2} END {print sum}'
-```
-
-### 2. 性能对比
-```bash
-# 对比有/无bank机制的性能差异
-grep 'cpu.ipc' debug/*/stats.txt
-grep 'tage.*Mispred' debug/*/stats.txt
-```
-
-### 3. 可选扩展功能
-- 添加每个bank的访问分布统计
-- 实现开窗机制（如果冲突率过高）
-- 支持动态禁用bank模拟（numBanks=1时自动禁用）
-
----
-
-**✅ 实现已完成并通过测试！Bank机制正常工作。**

src/cpu/pred/BranchPredictor.py

@@ -1057,7 +1057,7 @@ class BTBTAGE(TimedBaseBTBPredictor):
     useAltOnNaSize = Param.Unsigned(128, "Size of the useAltOnNa table")
     useAltOnNaWidth = Param.Unsigned(7, "Width of the useAltOnNa table")
     numBanks = Param.Unsigned(4, "Number of banks for bank conflict simulation")
-    enableBankConflict = Param.Bool(False, "Enable bank conflict simulation")
+    enableBankConflict = Param.Bool(True, "Enable bank conflict simulation")
     numDelay = 2
 
 class MicroTAGE(BTBTAGE):