IPCQ-DMA co-design HW design doc + fix IPCQ slot BW model

Add hardware design document (docs/ipcq-dma-codesign-hw.md) covering
PE_IPCQ high-level architecture, simulator verification, proposed HW
implementation, and alternatives analysis. Include D2 block diagrams
for baseline and proposed PE architectures.

Fix IPCQ slot-memory bandwidth parameters to match topology.yaml:
SRAM 128→512 GB/s (intrinsic BW, NoC-bottlenecked at 128),
HBM 32→256 GB/s (was per-channel, now per-PE aggregate).

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
2026-04-28 13:31:02 -07:00
parent 54fcb7e4bc
commit 533e699299
9 changed files with 1121 additions and 6 deletions
+157
View File
@@ -0,0 +1,157 @@
direction: right
pe: PE {
style.fill: "#f8f9fa"
style.stroke: "#495057"
style.border-radius: 8
cpu: PE_CPU (control) {
style.fill: "#bbdefb"
style.stroke: "#1565c0"
style.border-radius: 4
}
sched: PE_SCHED (dispatch) {
style.fill: "#bbdefb"
style.stroke: "#1565c0"
style.border-radius: 4
}
ipcq_added: IPCQ (added) {
style.fill: "#e1f5fe"
style.stroke: "#0277bd"
style.stroke-dash: 5
style.stroke-width: 2
style.border-radius: 6
ipcq: PE_IPCQ (control plane) {
style.fill: "#bbdefb"
style.stroke: "#1565c0"
style.border-radius: 4
}
}
dma: PE_DMA (single FIFO inbox) {
style.fill: "#fff3e0"
style.stroke: "#e65100"
style.border-radius: 6
}
fs: PE_FETCH_STORE {
style.fill: "#c8e6c9"
style.stroke: "#2e7d32"
style.border-radius: 4
}
tcm: TCM (16MB SRAM) {
style.fill: "#fce4ec"
style.stroke: "#c62828"
style.border-radius: 6
ipcq_slot: IPCQ Slot Region {
style.stroke-dash: 5
style.fill: "#ffcdd2"
style.stroke: "#c62828"
style.border-radius: 3
}
}
gemm: GEMM engine {
style.fill: "#c8e6c9"
style.stroke: "#2e7d32"
style.border-radius: 4
}
math: MATH engine {
style.fill: "#c8e6c9"
style.stroke: "#2e7d32"
style.border-radius: 4
}
fport: Fabric Port {
style.fill: "#ffe0b2"
style.stroke: "#e65100"
style.border-radius: 4
}
# Control → dispatch
cpu -> sched: cmd dispatch
cpu -> ipcq_added.ipcq: IpcqRequest
# Compute pipeline
sched -> dma: TileToken\n(compute port)
dma -> fs: TileToken
dma <-> tcm: DMA_READ/WRITE\n(HBM ↔ TCM)
fs <-> tcm: fetch/store\n(TCM ↔ reg)
fs -> gemm: TileToken
fs -> math: TileToken
gemm -> fs: TileToken
math -> fs: TileToken
# IPCQ data path — outbound
ipcq_added.ipcq -> dma: IpcqDmaToken\n(IPCQ port) {style.stroke: "#1565c0"}
# IPCQ data path — inbound (MetaArrival: DMA → IPCQ)
dma -> ipcq_added.ipcq: IpcqMetaArrival {style.stroke: "#1565c0"}
# Credit return (dashed)
ipcq_added.ipcq -> dma: IpcqCreditMetadata\n(NoC latency charged) {
style.stroke: "#7b1fa2"
style.stroke-dash: 5
}
# DMA ↔ Fabric
dma <-> fport
}
# ── NoC Router + attached resources ──
noc: NoC Router {
style.fill: "#f3e5f5"
style.stroke: "#6a1b9a"
style.border-radius: 6
}
hbm: Local HBM {
style.fill: "#e8eaf6"
style.stroke: "#283593"
style.border-radius: 6
ipcq_slot_hbm: IPCQ Slot Region {
style.stroke-dash: 5
style.fill: "#c5cae9"
style.stroke: "#283593"
style.border-radius: 3
}
}
sram: Cube SRAM {
style.fill: "#e0f7fa"
style.stroke: "#00695c"
style.border-radius: 6
ipcq_slot_sram: IPCQ Slot Region {
style.stroke-dash: 5
style.fill: "#b2dfdb"
style.stroke: "#00695c"
style.border-radius: 3
}
}
other_pe: Other PEs {
style.fill: "#ede7f6"
style.stroke: "#6a1b9a"
style.border-radius: 6
}
other_cube: Other Cubes / SIPs {
style.fill: "#ede7f6"
style.stroke: "#6a1b9a"
style.border-radius: 6
}
pe.fport <-> noc
noc <-> hbm
noc <-> sram
noc <-> other_pe
noc <-> other_cube
Binary file not shown.

After

Width:  |  Height:  |  Size: 1014 KiB

File diff suppressed because one or more lines are too long

After

Width:  |  Height:  |  Size: 44 KiB

+166
View File
@@ -0,0 +1,166 @@
direction: right
pe: PE {
style.fill: "#f8f9fa"
style.stroke: "#495057"
style.border-radius: 8
cpu: PE_CPU (control) {
style.fill: "#bbdefb"
style.stroke: "#1565c0"
style.border-radius: 4
}
sched: PE_SCHED (dispatch) {
style.fill: "#bbdefb"
style.stroke: "#1565c0"
style.border-radius: 4
}
ipcq: IPCQ Controller (NEW) {
style.fill: "#e1f5fe"
style.stroke: "#0277bd"
style.border-radius: 6
style.stroke-width: 2
ptrmgmt: Pointer Mgmt {
style.fill: "#b3e5fc"
style.stroke: "#0277bd"
style.border-radius: 4
qprf: QPair Reg File
bp: Backpressure
sag: Slot Addr Gen
}
sideband: Sideband {
style.fill: "#b3e5fc"
style.stroke: "#0277bd"
style.border-radius: 4
metax: Meta Extractor
crinj: Credit Injector
crrcv: Credit Receiver
}
}
dma: PE_DMA (MOD) {
style.fill: "#fff3e0"
style.stroke: "#e65100"
style.border-radius: 6
compute_port: compute port {
style.fill: "#ffe0b2"
style.stroke: "#e65100"
style.border-radius: 4
}
ipcq_port: IPCQ port {
style.fill: "#ffe0b2"
style.stroke: "#e65100"
style.border-radius: 4
}
wrr: WRR Arbiter (NEW) {
style.fill: "#ffcc80"
style.stroke: "#e65100"
style.border-radius: 4
style.stroke-width: 2
}
compute_port -> wrr
ipcq_port -> wrr
}
fs: PE_FETCH_STORE {
style.fill: "#c8e6c9"
style.stroke: "#2e7d32"
style.border-radius: 4
}
tcm: TCM (16MB SRAM) {
style.fill: "#fce4ec"
style.stroke: "#c62828"
style.border-radius: 6
work: Kernel Working Memory {
style.fill: "#f8bbd0"
style.stroke: "#c62828"
style.border-radius: 4
}
slot: IPCQ Slot Region (rsv) {
style.fill: "#f48fb1"
style.stroke: "#c62828"
style.border-radius: 4
style.stroke-width: 2
}
}
gemm: GEMM engine {
style.fill: "#c8e6c9"
style.stroke: "#2e7d32"
style.border-radius: 4
}
math: MATH engine {
style.fill: "#c8e6c9"
style.stroke: "#2e7d32"
style.border-radius: 4
}
fport: Fabric Port {
style.fill: "#ffe0b2"
style.stroke: "#e65100"
style.border-radius: 4
}
# Control
cpu -> sched: cmd dispatch
cpu -> ipcq: MMIO
# Compute pipeline
sched -> dma.compute_port: TileToken
dma -> fs: TileToken
dma <-> tcm.work: DMA_READ/WRITE\n(HBM ↔ TCM)
fs <-> tcm.work: fetch/store\n(TCM ↔ reg)
fs -> gemm: TileToken
fs -> math: TileToken
gemm -> fs: TileToken
math -> fs: TileToken
# IPCQ data path
ipcq -> dma.ipcq_port: IpcqDmaToken {style.stroke: "#0277bd"}
dma -> ipcq.sideband.metax: IpcqMetaArrival {style.stroke: "#0277bd"}
# IPCQ slot R/W
dma <-> tcm.slot: slot read/write {
style.stroke: "#0277bd"
style.stroke-dash: 3
}
# Credit via fabric port
ipcq.sideband.crinj -> fport: credit out (16B) {
style.stroke: "#7b1fa2"
style.stroke-dash: 5
}
fport -> ipcq.sideband.crrcv: credit in (16B) {
style.stroke: "#7b1fa2"
style.stroke-dash: 5
}
# DMA ↔ Fabric
dma.wrr <-> fport
}
noc: NoC Router {
style.fill: "#f3e5f5"
style.stroke: "#6a1b9a"
style.border-radius: 6
}
ext: Other PEs / Cubes / SIPs {
style.fill: "#ede7f6"
style.stroke: "#6a1b9a"
style.border-radius: 6
}
pe.fport <-> noc
noc <-> ext
Binary file not shown.

After

Width:  |  Height:  |  Size: 836 KiB

File diff suppressed because one or more lines are too long

After

Width:  |  Height:  |  Size: 48 KiB

+548
View File
@@ -0,0 +1,548 @@
# IPCQ-DMA Co-design Hardware Design Document
**Status**: Draft — Review Requested
**Date**: 2026-04-28
**Authors**: YW Kang
**Reviewers**: (HW team TBD)
**Related**: ADR-0023 (IPCQ PE Collective), ADR-0025 (Direction Addressing)
---
## 1. Background & Motivation
IPCQ(Inter-PE Communication Queue)는 PE 간 collective communication을 위한
하드웨어 큐 메커니즘이다. 핵심 설계 원리는 **DMA가 데이터 전송 시 별도의
제어 메시지 없이, piggyback된 메타 정보를 바탕으로 IPCQ의 head/tail pointer를
자동 업데이트**하는 IPCQ-DMA co-design이다.
이 문서는:
1. 현재 PE 아키텍처에서 IPCQ가 하드웨어 수준에서 어떻게 동작하는지 기술하고,
2. 이 하드웨어를 시뮬레이터에서 어떻게 모델링하고 있는지 검증하며,
3. 실제 하드웨어 구현을 위한 설계를 제안하고,
4. 대안들을 검토하여 최적 접근을 확정한다.
---
## 2. High-level Behavior of PE_IPCQ
![PE Baseline Architecture](diagrams/pe_baseline.png)
> source: [`diagrams/pe_baseline.d2`](diagrams/pe_baseline.d2) — `d2 --layout=elk --scale 1.5` 로 렌더링.
### IPCQ 하드웨어 동작
**HW Configuration**:
* IPCQ는 PE 간에 ring buffer 기반의 단방향 큐를 설정하여 데이터를 전달한다.
* 각 PE는 방향별(N/S/E/W 등)로 독립적인 queue pair 를 유지한다.
* IPCQ는 각 queue pair 마다 sender's head/tail pointer, receiver's head/tail pointer 를 유지한다.
* **IPCQ Slot Region**: IPCQ의 수신 버퍼로, 다이어그램의 점선 박스로 표시된 것처럼 TCM, Cube SRAM, Local HBM 중 하나를 buffer_kind로 지정하여 사용할 수 있다.
각 tier별 성능 특성 (시뮬레이션 모델 값, `ipcq_types.py`):
| Buffer Kind | Intrinsic BW | Effective BW (NoC bottleneck) | 용도 |
|-------------|-------------|-------------------------------|------|
| TCM | 512 GB/s | 512 GB/s (직결, NoC 미경유) | 최저 latency, PE 내부 전용 |
| Cube SRAM | 512 GB/s | 128 GB/s (`sram_to_router_bw`) | Cube 내 공유, NoC BW에 제한 |
| Local HBM | 256 GB/s | 256 GB/s (`hbm_to_router_bw`) | 대용량, NoC BW에 제한 |
**Send 경로 (fire-and-forget)**:
1. PE_CPU가 `tl.send(dir, src_addr)` 발행 → PE_IPCQ에 IpcqRequest 전달
2. PE_IPCQ가 backpressure 확인: `(my_head - peer_tail_cache) < peer.n_slots`
3. Peer의 rx slot 주소 계산: `peer_rx_base + (my_head % n_slots) × slot_size`
4. IpcqDmaToken(data + piggyback metadata: sender_seq)을 PE_DMA에 전달
5. PE_IPCQ가 `my_head++`, PE_CPU에 즉시 반환 (DMA 완료를 기다리지 않음)
6. PE_DMA가 src data를 snapshot 후 NoC를 통해 peer PE_DMA로 전송
**Receive 경로 (blocking)**:
1. Peer PE_DMA가 data를 slot에 write하고, **같은 사이클에** metadata(sender_seq, dst_addr)를 추출
2. PE_IPCQ가 dst_addr range matching으로 방향을 식별, `peer_head_cache` 업데이트
3. `tl.recv(dir)` 대기 중인 PE_CPU에 wakeup signal 전달
4. PE_CPU가 slot에서 데이터 읽기, PE_IPCQ가 `my_tail++`
5. **Credit return**: PE_IPCQ가 16B credit packet(`consumer_seq`)을 NoC를 통해 sender에게 전송
6. Sender PE_IPCQ가 `peer_tail_cache` 업데이트, backpressure 해제
**핵심 설계 원리**:
- **Data + head pointer piggyback**: 별도의 head 동기화 메시지 없이, DMA data flit에 sender_seq를 실어보냄
- **Atomic write + metadata**: 수신측 DMA가 slot write와 metadata 전달을 같은 사이클에 수행 (I6 invariant)
- **Address-based direction matching**: 같은 peer에 여러 방향이 연결되어도 dst_addr range로 구분 (ADR-0025)
- **Credit-based flow control**: Receiver가 slot 소비 후 16B credit으로 sender에게 알림
---
## 3. Simulator Implementation Verification
위의 하드웨어 동작을 시뮬레이터에서 어떻게 모델링하는지 검증한다.
### 3.1 의도와 구현의 매핑
| 설계 의도 | 시뮬레이터 구현 | 위치 |
|-----------|----------------|------|
| DMA가 데이터 전송 시 head pointer를 piggyback | `IpcqDmaToken.sender_seq` 필드가 data flit과 함께 전달 | `ipcq_types.py:185` |
| 수신측 DMA가 data write + metadata 전달을 atomic 처리 | `_handle_ipcq_inbound`에서 `store.write``IpcqMetaArrival` 사이에 yield 없음 (I6) | `pe_dma.py:232-275` |
| Send는 fire-and-forget | `_handle_ipcq_outbound`에서 `sub_done`을 기다리지 않음 | `pe_dma.py:182` |
| Recv는 데이터 도착까지 block | `peer_head_cache > my_tail` 조건으로 대기 | `pe_ipcq.py:263` |
| Credit return은 별도 fast-path | SimPy Store를 통한 direct put (latency는 NoC 경로 기반으로 charge) | `pe_ipcq.py:443-469` |
| In-flight data semantics (snapshot) | Send 시점에 data snapshot 보존, 이후 src 수정과 무관 | `pe_dma.py:142-155` |
| PE_DMA 단일 inbox | 모든 in_port를 `_fan_in`으로 단일 FIFO에 merge (`base.py:51-53`) | compute port와 IPCQ port 사이에 arbiter 없음 |
### 3.2 Credit Return Path 모델링 상세
Credit return은 실제 NoC 경로를 `router.find_path()`로 찾고,
`compute_path_latency_ns()`로 hop latency + BW drain을 계산하여 charge한다.
```python
# pe_ipcq.py:471-492
def _credit_latency_ns(self, direction: str) -> float:
path = self.ctx.router.find_path(self._pe_prefix, peer_pe_dma)
return self.ctx.compute_path_latency_ns(path, self._credit_size_bytes)
```
단, latency를 `env.timeout()`으로 지불한 후 `peer_credit_store`(SimPy Store)에
직접 put하는 방식이다. 실제 `Transaction`을 만들어 NoC를 hop-by-hop 통과시키지는
않으므로, **다른 트래픽과의 bandwidth contention은 모델링되지 않는다.**
| | Latency | BW Contention |
|---|---|---|
| Data path (IpcqDmaToken) | NoC Transaction으로 정확 모델링 | 실제 fabric 통과 |
| Credit path (16B) | NoC 경로 latency 정확 반영 | fabric Transaction 미주입 (단순화) |
Credit은 16B로 data transfer(수십~수백 KB) 대비 무시 가능한 크기이므로,
이 단순화로 인한 실질적 오차는 거의 없다.
### 3.3 검증 결론
시뮬레이터 구현은 IPCQ-DMA co-design 의도를 **정확하게 모델링**하고 있다.
---
## 4. Proposed Hardware Design
### 4.1 Block Diagram (변경 후)
변경점을 강조 표시: **(NEW)** = 신규, **(MOD)** = 수정.
![PE Proposed Architecture](diagrams/pe_proposed.png)
> Source: [`diagrams/pe_proposed.d2`](diagrams/pe_proposed.d2) — `d2 --layout=elk` 로 렌더링.
**Baseline → Proposed 핵심 변경**:
- 단일 FIFO inbox → **compute port / IPCQ port 분리 + WRR Arbiter** (NEW)
- PE_IPCQ (SimPy component) → **IPCQ Controller** (HW register + combinational logic)
- TCM 내 **IPCQ Slot Region 예약 영역** 명시
- Credit Injector / Receiver가 Fabric Port를 통해 NoC에 직접 연결
### 4.2 Module Details
#### 4.2.1 IPCQ Controller (신규 모듈)
PE_CPU와 DMA Engine 사이에 위치하는 하드웨어 제어 블록.
시뮬레이터의 `PeIpcqComponent`에 대응한다.
##### QPair Register File
방향별 queue pair 상태를 flip-flop으로 유지한다.
```
Per-direction registers (each 64-bit):
my_head — sender write position (monotonic)
my_tail — receiver read position (monotonic)
peer_head_cache — last known peer head (updated by Meta Extractor)
peer_tail_cache — last known peer tail (updated by Credit Receive)
rx_base_pa — this PE's rx buffer base physical address
peer_rx_base_pa — peer's rx buffer base physical address
n_slots — ring depth (power-of-2 제약, 아래 참조)
slot_size — bytes per slot
peer_credit_tgt — peer PE의 credit receive 주소
Directions: 최대 8 (N/S/E/W/parent/child_left/child_right + spare)
Total: 8 dirs × 9 regs × 8B = 576B flip-flops
```
PE_CPU가 MMIO(CSR)로 읽기/쓰기 가능. Init 시점에 소프트웨어가 채워넣는다.
##### Slot Address Generator (combinational)
```
Input: pointer (my_head or my_tail), n_slots, slot_size, base_pa
Output: slot_addr = base_pa + (pointer % n_slots) * slot_size
Implementation:
n_slots power-of-2 제약 → pointer & (n_slots - 1) (AND mask, 1 gate delay)
slot_size power-of-2 → barrel shift (1 cycle)
64-bit add → ripple/kogge-stone adder (1 cycle)
Latency: 1-2 cycles combinational
```
##### Backpressure Comparator (combinational)
```
full = (my_head - peer_tail_cache) >= n_slots
Implementation: 64-bit subtract + unsigned compare
Output: stall signal → PE_CPU (IPCQ send blocked) or DMA issue hold
Latency: 1 cycle
```
##### Meta Extractor (inbound datapath sideband)
DMA Engine의 inbound vc_comm path에 wired. Arriving IPCQ flit의 header에서
metadata를 추출하여 queue pair 상태를 업데이트한다.
```
Trigger: DMA inbound write completion (same cycle)
Extract: {sender_seq, dst_addr} from flit header
Direction matching (ADR-0025 D2):
for each dir:
match = (base_pa[dir] <= dst_addr) && (dst_addr < base_pa[dir] + n_slots[dir] * slot_size[dir])
8× parallel range comparators + priority encoder
Update: peer_head_cache[matched_dir] = max(peer_head_cache, sender_seq + 1)
Output: recv_wake signal for matched direction → PE_CPU interrupt/flag
Implementation: 8× (2 comparators + AND) + priority encoder
Latency: 1 cycle (pipelined with DMA write — I6 atomicity 자연 보장)
```
##### Credit Injector (outbound)
```
Trigger: recv completion (my_tail 증가 후)
Action: pack 16B credit packet → DMA vc_comm (또는 dedicated credit VC)
Packet: {consumer_seq = my_tail, dst_rx_base_pa = my_rx_base_pa}
Latency: 1 cycle to generate, then NoC traversal
```
##### Credit Receiver (inbound sideband)
```
Trigger: 16B credit packet arrival (from NoC)
Extract: {consumer_seq, dst_rx_base_pa}
Direction matching (ADR-0025 D3):
for each dir:
match = (peer_rx_base_pa[dir] == credit.dst_rx_base_pa)
Update: peer_tail_cache[matched_dir] = max(peer_tail_cache, consumer_seq)
Output: send_wake signal → deassert backpressure stall
Latency: 1 cycle
```
#### 4.2.2 DMA Engine 수정사항
##### vc_comm IPCQ-aware mode
기존 vc_comm 채널에 IPCQ flit 처리 모드를 추가한다.
**Outbound**:
1. IPCQ Controller로부터 command 수신: {src_addr, dst_addr, nbytes, sender_seq}
2. TCM에서 src_addr read → DMA read buffer에 snapshot (기존 DMA behavior)
3. Flit pack: data + piggyback metadata (sender_seq, dst_addr)
4. NoC fabric port에 inject
5. Fire-and-forget (completion을 기다리지 않음)
**Inbound**:
1. NoC로부터 IPCQ flit 수신
2. Terminal BW drain charge (drain_ns = nbytes / bottleneck_bw)
3. Slot write latency charge (backing memory tier)
4. **ATOMIC** (same pipeline stage, no stall insertion):
- TCM write: data → slot address
- Meta Extractor trigger: sender_seq + dst_addr → IPCQ Controller
5. Done
**I6 atomicity 하드웨어 보장**: TCM write completion과 Meta Extractor trigger가
동일 pipeline stage에서 발생하므로 별도 synchronization이 불필요하다.
시뮬레이터의 "no yield between write and IpcqMetaArrival"이 자연스럽게 보장된다.
##### Data Snapshot Semantics
DMA read buffer에 latch된 데이터는 src memory의 이후 수정에 영향받지 않는다.
이는 DMA의 standard read-then-write behavior이므로 추가 HW가 불필요하다.
##### Credit Virtual Channel (선택적)
옵션 A: vc_comm에 credit을 multiplexing (16B header-only flit으로 구분)
옵션 B: 3rd dedicated credit VC 추가 (strict priority > data)
옵션 B가 deadlock prevention에 유리하나, 16B credit의 BW 영향이 무시 가능하므로
옵션 A로도 충분하다.
#### 4.2.3 Fabric Flit Format 확장
```
일반 data flit (예: 512-bit):
┌──────────────────────────────────────────┐
│ [511:480] routing header (32b) │
│ [479:0] payload (480b = 60B) │
└──────────────────────────────────────────┘
IPCQ data flit (첫 flit에만 metadata 포함):
┌──────────────────────────────────────────┐
│ [511:480] routing header (32b) │
│ [511] ipcq_flag (1b) │ ← IPCQ vs normal DMA 식별
│ [510:509] vc_id (2b) │
│ [508:480] route + hop count │
│ [479:416] ipcq_metadata (64b) │ ← piggyback
│ [479:448] sender_seq (32b) │
│ [447:416] dst_addr[31:0] (32b) │ ← direction matching용
│ [415:0] payload (416b = 52B) │
└──────────────────────────────────────────┘
후속 flits: full 60B payload (metadata 없음)
Credit-only flit (128-bit, header-only):
┌──────────────────────────────────────────┐
│ [127:96] routing header (32b) │
│ [127] credit_flag (1b) │
│ [95:64] consumer_seq (32b) │
│ [63:0] dst_rx_base_pa (64b) │
└──────────────────────────────────────────┘
```
첫 flit의 payload가 60B → 52B로 감소 (13% overhead).
Multi-flit transfer에서는 후속 flit이 full payload이므로 대형 전송에서 overhead < 1%.
#### 4.2.4 TCM IPCQ Slot Region
```
TCM Memory Map (16MB):
┌─────────────────────────────┐ 0x000000
│ Kernel Working Memory │
│ (compute tensors) │
│ ~14MB │
├─────────────────────────────┤ 0xE00000
│ IPCQ RX Buffers │
│ Dir N: slots × slot_size │
│ Dir S: slots × slot_size │
│ Dir E: slots × slot_size │
│ Dir W: slots × slot_size │
│ ~1MB │
├─────────────────────────────┤ 0xF00000
│ IPCQ Metadata / Scratch │
│ ~1MB │
└─────────────────────────────┘ 0xFFFFFF
```
IPCQ region을 TCM의 상위 bank에 배치하여 compute access와의
bank conflict를 최소화한다 (Section 6.1 참조).
---
## 5. End-to-End Dataflow
### 5.1 Sequence Diagram
```mermaid
sequenceDiagram
participant CPU_A as PE_A: PE_CPU
participant IPCQ_A as PE_A: IPCQ Ctrl
participant DMA_A as PE_A: DMA
participant NOC as NoC Fabric
participant DMA_B as PE_B: DMA
participant IPCQ_B as PE_B: IPCQ Ctrl
participant TCM_B as PE_B: TCM
participant CPU_B as PE_B: PE_CPU
Note over CPU_A: tl.send(dir="E", src=0x1000)
CPU_A->>IPCQ_A: MMIO: send request
Note over IPCQ_A: Backpressure check:<br/>(head - peer_tail_cache) < n_slots → PASS<br/>Slot addr gen:<br/>dst = peer_rx_base + (head%n) × slot_size
IPCQ_A->>DMA_A: IpcqDmaToken {src, dst, sender_seq=head}
Note over IPCQ_A: my_head++
IPCQ_A-->>CPU_A: send returns (fire-and-forget)
Note over DMA_A: TCM read → snapshot in read buffer<br/>Flit pack: data + {sender_seq, dst_addr}
DMA_A->>NOC: IPCQ data flit(s)
Note over NOC: hop latency + BW drain
NOC->>DMA_B: IPCQ data flit(s)
Note over DMA_B: Terminal BW drain<br/>Slot write latency
rect rgb(255, 240, 220)
Note over DMA_B,IPCQ_B: ATOMIC (I6): same cycle, no stall
DMA_B->>TCM_B: write data → slot address
DMA_B->>IPCQ_B: Meta Extractor: {sender_seq, dst_addr}
end
Note over IPCQ_B: Range match dst_addr → direction "W"<br/>peer_head_cache["W"] = sender_seq + 1
IPCQ_B-->>CPU_B: recv_wake signal
Note over CPU_B: tl.recv(dir="W") wakes up
CPU_B->>IPCQ_B: recv request
Note over IPCQ_B: peer_head_cache > my_tail → YES<br/>slot_addr = rx_base + (tail%n) × slot_size
IPCQ_B-->>CPU_B: return slot_addr
CPU_B->>TCM_B: read data from slot
Note over IPCQ_B: my_tail++
IPCQ_B->>NOC: Credit (16B): {consumer_seq, dst_rx_base_pa}
Note over NOC: credit traversal (NoC latency)
NOC->>IPCQ_A: Credit arrival
Note over IPCQ_A: Match dst_rx_base_pa → direction "E"<br/>peer_tail_cache["E"] = consumer_seq<br/>Backpressure deassert (if stalled)
```
---
## 6. 2nm Implementation Analysis
### 6.1 Area Estimate
| Module | Gate Count | Area (2nm est.) | Notes |
|--------|-----------|-----------------|-------|
| QPair Register File | ~4.6K FF | 0.002 mm² | 576B flip-flops |
| Slot Addr Gen + Backpressure | ~5K gates | 0.001 mm² | Combinational |
| Meta Extractor + Credit Logic | ~3K gates | 0.001 mm² | 8× parallel comparators |
| **Total IPCQ Controller** | **~12.6K** | **~0.004 mm²** | **PE 전체 대비 < 0.1%** |
| DMA vc_comm 확장 | ~2K gates | 0.002 mm² | Flit pack/unpack |
| **Total 변경분** | **~14.6K** | **~0.006 mm²** | |
### 6.2 Timing
| Path | Delay (2nm est.) | Target Clock | Margin |
|------|-------------------|-------------|--------|
| Backpressure (sub + cmp) | ~0.3 ns | 1 GHz (1 ns) | 3× |
| Slot Addr Gen (mask + shift + add) | ~0.5 ns | 1 GHz | 2× |
| Meta Extractor (8× range match) | ~0.4 ns | 1 GHz | 2.5× |
| Credit Receiver (8× equality) | ~0.3 ns | 1 GHz | 3× |
모든 critical path가 1 cycle 이내. Timing closure 문제 없음.
### 6.3 Power
- Active: ~1 mW (register read/write + comparators, send/recv 동작 시)
- Idle: leakage only
- PE 전체 전력 대비 무시 가능
### 6.4 Constraints
| 항목 | 제약 | 근거 |
|------|------|------|
| `n_slots` | **반드시 power-of-2** | mod → AND mask (1 gate). 임의 값은 divider 필요 (~10 cycles) |
| `slot_size` | **power-of-2 권장** | mul → barrel shift. 임의 값은 multiplier 필요 |
| TCM IPCQ region | **전용 bank 배치** | Compute access와 bank conflict 방지 |
---
## 7. Risk Assessment
### 7.1 TCM Bank Conflict
- **Risk**: IPCQ slot write와 compute read가 동일 bank 접근 시 stall
- **Mitigation**: IPCQ region을 TCM 상위 address의 전용 bank에 배치
- **Cost**: TCM banking flexibility 소폭 감소
- **Severity**: Medium (성능 영향), Low (correctness 문제 아님)
### 7.2 Credit Return Latency under Congestion
- **Risk**: NoC 혼잡 시 credit return 지연 → sender backpressure stall
- **Mitigation**:
- Credit을 별도 VC로 분리 + strict priority (16B로 BW impact 미미)
- 또는 n_slots를 넉넉히(8+) 설정하여 credit 지연을 buffer로 흡수
- **Severity**: Low (credit 16B는 congestion에 거의 기여하지 않음)
### 7.3 Inter-Direction Ordering
- **Risk**: 같은 PE에서 여러 방향으로 동시 send 시 순서
- **Mitigation**: Per-direction monotonic seq으로 충분. Inter-direction ordering은
kernel(소프트웨어) 책임 — 현재 시뮬레이터 모델과 동일
- **Severity**: Low (아키텍처 설계에 의해 해소)
---
## 8. Alternatives Considered
### 8.1 Doorbell + Polling (전통적 방식)
```
Send: DMA write data → DMA write doorbell register at peer → peer polls doorbell
Recv: Polling loop on doorbell, or interrupt-driven
```
| 장점 | 단점 |
|------|------|
| 단순한 HW (IPCQ controller 불필요) | 2번의 DMA transaction (data + doorbell) |
| 기존 DMA 재사용 | Data/doorbell 사이 ordering 보장 필요 (fence) |
| | Polling은 전력 낭비, interrupt는 latency overhead |
**평가**: Piggyback 대비 latency 2-3× 증가. **불채택.**
### 8.2 Hardware Message Queue (NVIDIA NVLink 스타일)
```
Send: CPU → HMQ에 descriptor push → HW가 peer HMQ로 자동 전달
Recv: HMQ에서 descriptor pop → data pointer 확인
```
| 장점 | 단점 |
|------|------|
| CPU는 descriptor만 작성 | 별도 HMQ engine 필요 (~0.05 mm²) |
| Descriptor/data 분리 → 유연 | DMA와 별개 datapath → area/power 중복 |
| | Large tensor에는 결국 DMA 필요 |
**평가**: CCL의 large tensor 패턴에서 DMA 필수이므로 HMQ + DMA 이중 구조는
면적 낭비. **불채택.**
### 8.3 RDMA-style Completion Queue (CQ)
```
Send: DMA write → peer에 CQE 자동 생성
Recv: CQ poll/interrupt → data 위치 확인
```
| 장점 | 단점 |
|------|------|
| InfiniBand/RoCE 성숙 모델 | CQ 관리 logic + CQE memory overhead |
| Multi-tenant/isolation 용이 | CQE/data ordering 보장 추가 필요 |
| | PE-to-PE CCL에는 over-engineered |
**평가**: RDMA CQ는 host-facing NIC의 multi-tenant 격리에 적합.
PE 간 단일 owner 환경에서는 불필요한 복잡성. **불채택.**
### 8.4 Credit-in-Data Piggyback (v2 최적화 후보)
현재 설계에서 credit return은 별도 16B packet이다.
Bidirectional 통신 패턴에서는 **reverse 방향 data flit에 credit을 합칠 수 있다.**
```
PE_A →E→ PE_B: data + sender_seq=3
PE_B →W→ PE_A: data + sender_seq=5 + credit_ack=4 ← credit이 data에 합쳐짐
```
| 장점 | 단점 |
|------|------|
| Credit 전용 packet 제거 → NoC BW 절약 | Unidirectional 패턴에서는 fallback 필요 |
| Bidirectional allreduce에서 credit latency → 0 | Flit header에 8B 추가 (overhead 미미) |
| | Logic 복잡도 소폭 증가 |
**평가**: 현재 설계의 우수한 최적화.
Bidirectional allreduce에서 credit packet을 완전 제거 가능.
Standalone credit fallback도 유지. **v2로 채택 권고.**
---
## 9. Recommendations
1. **현재 IPCQ-DMA co-design을 기본 하드웨어 설계로 채택**
— 단순하고, 면적 효율적이며, 2nm에서 timing/power 문제 없음
2. **n_slots를 반드시 power-of-2로 제약**
— mod 연산을 AND mask로 대체, critical path 단축
3. **TCM banking에서 IPCQ region 전용 bank 할당**
— compute와의 bank conflict 방지
4. **v2에서 Credit-in-Data Piggyback (Section 8.4) 추가 검토**
— bidirectional 패턴에서 credit overhead 제거
---
## 10. Open Questions
- [ ] IPCQ slot region size를 TCM의 몇 %까지 허용할 것인가? (현재 가정: ~1MB / 16MB = 6.25%)
- [ ] Credit VC를 별도로 둘 것인가, vc_comm에 multiplexing할 것인가?
- [ ] Inter-SIP link에서의 flit format 호환성 검증 필요
- [ ] n_slots 최대값 제한? (8 directions × 8 slots × 64KB = 4MB → TCM의 25%)
+2 -2
View File
@@ -37,8 +37,8 @@ class IpcqInvalidDirection(ValueError):
# and slot read (recv consume). Mirrors topology.yaml component values. # and slot read (recv consume). Mirrors topology.yaml component values.
_BUFFER_KIND_BW: dict[str, tuple[float, float]] = { _BUFFER_KIND_BW: dict[str, tuple[float, float]] = {
"tcm": (512.0, 0.0), "tcm": (512.0, 0.0),
"sram": (128.0, 2.0), "sram": (512.0, 2.0),
"hbm": (32.0, 6.0), "hbm": (256.0, 6.0),
} }
+4 -4
View File
@@ -47,8 +47,8 @@ from tests.test_allreduce_multidevice import (
# pe_ipcq.py). Mirrors topology.yaml component values. # pe_ipcq.py). Mirrors topology.yaml component values.
_EXPECTED_BW = { _EXPECTED_BW = {
"tcm": (512.0, 0.0), "tcm": (512.0, 0.0),
"sram": (128.0, 2.0), "sram": (512.0, 2.0),
"hbm": (32.0, 6.0), "hbm": (256.0, 6.0),
} }
@@ -160,10 +160,10 @@ def test_slot_io_scales_linearly_with_nbytes(tmp_path):
lat_8k = _run_torus_allreduce(tmp_path, buffer_kind="hbm", n_elem=4096) lat_8k = _run_torus_allreduce(tmp_path, buffer_kind="hbm", n_elem=4096)
# Expected delta from doubling: at least one slot-IO event per cube # Expected delta from doubling: at least one slot-IO event per cube
# in the critical path (very conservative). Per-access add = 4096/32 ≈ 128 # in the critical path (very conservative). Per-access add = 4096/256 = 16
# ns on HBM going from 4k → 8k. Multiple slot accesses on the critical # ns on HBM going from 4k → 8k. Multiple slot accesses on the critical
# path should make the observed delta meaningfully larger. # path should make the observed delta meaningfully larger.
expected_min_delta = 0.5 * (4096 / 32.0) # ≈ 64 ns expected_min_delta = 0.5 * (4096 / 256.0) # ≈ 8 ns
assert lat_8k - lat_4k > expected_min_delta, ( assert lat_8k - lat_4k > expected_min_delta, (
f"doubling nbytes on hbm should add ≥ {expected_min_delta:.1f} ns " f"doubling nbytes on hbm should add ≥ {expected_min_delta:.1f} ns "
f"of slot-IO latency, got delta={lat_8k - lat_4k:.1f} ns " f"of slot-IO latency, got delta={lat_8k - lat_4k:.1f} ns "