ywkang
eb792e6212
- Delete xbar.py and noc.py (TwoDMeshNocComponent) — unused since router mesh - Remove xbar_v1/noc_2d_mesh_v1 from components.yaml - Fix pe_to_xbar → pe_to_router in routing exclusion set - Fix xbar_to_hbm_bw_gbs → hbm_to_router_bw_gbs in report.py - Update all docstrings/comments referencing xbar/bridge → router mesh - Cube-view connectors: rule-based _connector_points helper - PE↔router: single diagonal line (not chevron) - UCIe N/S: 45°→horizontal→45° - UCIe E/W: 45°→vertical→45° - HBM ports: 45°→horizontal→45° Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
51 lines
2.5 KiB
YAML
51 lines
2.5 KiB
YAML
# Component implementation registry.
|
|
# Maps impl names (used in topology.yaml) to Python class paths.
|
|
# Format: impl_name: module.path:ClassName
|
|
#
|
|
# ── Adding custom components ──────────────────────────────────────────
|
|
#
|
|
# 1. Create your implementation in:
|
|
# src/kernbench/components/custom/<your_component>.py
|
|
#
|
|
# Your class must inherit from ComponentBase (or PeEngineBase for PE engines).
|
|
#
|
|
# 2. Register it below under "Custom" with a unique impl name:
|
|
# my_pe_cpu_v2: kernbench.components.custom.my_pe_cpu:MyPeCpuComponent
|
|
#
|
|
# 3. Reference it in topology.yaml:
|
|
# pe_cpu: { kind: pe_cpu, impl: my_pe_cpu_v2, attrs: { ... } }
|
|
#
|
|
# 4. Add unit tests in:
|
|
# tests/custom/test_<your_component>.py
|
|
#
|
|
# External packages also work — use the full module path:
|
|
# fast_gemm_v1: my_team.accel.fast_gemm:FastGemmComponent
|
|
# ──────────────────────────────────────────────────────────────────────
|
|
|
|
components:
|
|
# Infrastructure
|
|
forwarding_v1: kernbench.components.builtin.forwarding:TransitComponent
|
|
switch_v1: kernbench.components.builtin.forwarding:TransitComponent
|
|
noc_v1: kernbench.components.builtin.forwarding:TransitComponent
|
|
ucie_v1: kernbench.components.builtin.forwarding:TransitComponent
|
|
# IO / Host interface
|
|
pcie_ep_v1: kernbench.components.builtin.pcie_ep:PcieEpComponent
|
|
io_cpu_v1: kernbench.components.builtin.io_cpu:IoCpuComponent
|
|
|
|
# Cube-level
|
|
m_cpu_v1: kernbench.components.builtin.m_cpu:MCpuComponent
|
|
hbm_ctrl_v1: kernbench.components.builtin.hbm_ctrl:HbmCtrlComponent
|
|
sram_v1: kernbench.components.builtin.sram:SramComponent
|
|
|
|
# PE-level
|
|
pe_cpu_v1: kernbench.components.builtin.pe_cpu:PeCpuComponent
|
|
pe_scheduler_v1: kernbench.components.builtin.pe_scheduler:PeSchedulerComponent
|
|
pe_dma_v1: kernbench.components.builtin.pe_dma:PeDmaComponent
|
|
pe_gemm_v1: kernbench.components.builtin.pe_gemm:PeGemmComponent
|
|
pe_math_v1: kernbench.components.builtin.pe_math:PeMathComponent
|
|
pe_mmu_v1: kernbench.components.builtin.pe_mmu:PeMmuComponent
|
|
pe_tcm_v1: kernbench.components.builtin.pe_tcm:PeTcmComponent
|
|
|
|
# Custom — add your implementations here
|
|
pe_scheduler_v2: kernbench.components.custom.pe_accel.scheduler:SchedulerV2Component
|