ywkang/kernbench2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add SchedulerV2 (pe_accel), DPPolicy overrides, and new benchmarks

Browse Source 
- Add cycle-accurate PE accelerator scheduler (SchedulerV2) with tiled
  GEMM/Math pipelines (DMA_IN → GEMM → MATH → DMA_WB)
- Add DPPolicy num_pes/num_cubes/num_sips overrides for single-PE testing
- Support tuple target_pe for targeting specific PE subsets
- Add gemm_single_pe and gpt3_qkv benchmarks
- Switch default topology to pe_scheduler_v2

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Yangwook
2026-03-26 23:18:49 -07:00
parent 63669f82cb
commit 114510d4b9
22 changed files with 1822 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
components.yaml
+1 -1
View File
@@ -50,4 +50,4 @@ components:
pe_tcm_v1: kernbench.components.builtin.pe_tcm:PeTcmComponent
# Custom — add your implementations here
# pe_cpu_v2: kernbench.components.custom.my_pe_cpu:MyPeCpuComponent
pe_scheduler_v2: kernbench.components.custom.pe_accel.scheduler:SchedulerV2Component