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

Revert "analytical-viz: default intra-cube (PE↔PE) BW to 128 GB/s"

Browse Source 
This reverts commit 84abf847c4.
This commit is contained in:
Mukesh Garg
2026-07-28 21:50:17 -07:00
parent 84abf847c4
commit afe35ee0b5
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/analytical_visualization/app.py
+1 -1
View File
@@ -374,7 +374,7 @@ with st.sidebar.expander("Hardware", expanded=False):
bw_intra = st.selectbox( bw_intra = st.selectbox(
"intra-cube GB/s", "intra-cube GB/s",
[128.0, 256.0, 512.0, 1024.0, 2048.0, 4096.0], [128.0, 256.0, 512.0, 1024.0, 2048.0, 4096.0],
index=0, key="bw_intra", index=2, key="bw_intra",
) )
bw_inter = st.selectbox( bw_inter = st.selectbox(
"inter-cube GB/s", "inter-cube GB/s",
tests/analytical_visualization/auto_hardware.py
+3 -3
View File
@@ -38,7 +38,7 @@ _HW_KNOB_DEFAULTS = {
"pe_hbm_gb": 6.0, "pe_hbm_gb": 6.0,
"bw_hbm_gbs": 256.0, "bw_hbm_gbs": 256.0,
"peak_tflops_f16": 8.0, "peak_tflops_f16": 8.0,
"bw_intra_gbs": 128.0, "bw_intra_gbs": 512.0,
"bw_inter_gbs": 128.0, "bw_inter_gbs": 128.0,
"bw_intersip_gbs": 50.0, "bw_intersip_gbs": 50.0,
} }
@@ -47,7 +47,7 @@ _HW_VALUES_BALANCED = {
"pe_hbm_gb": [6.0, 12.0], "pe_hbm_gb": [6.0, 12.0],
"bw_hbm_gbs": [256.0, 512.0], "bw_hbm_gbs": [256.0, 512.0],
"peak_tflops_f16": [8.0, 16.0], "peak_tflops_f16": [8.0, 16.0],
"bw_intra_gbs": [128.0, 256.0], "bw_intra_gbs": [512.0, 1024.0],
"bw_inter_gbs": [128.0, 256.0], "bw_inter_gbs": [128.0, 256.0],
"bw_intersip_gbs": [50.0, 100.0], "bw_intersip_gbs": [50.0, 100.0],
} }
@@ -56,7 +56,7 @@ _HW_VALUES_COARSE = {
"pe_hbm_gb": [6.0, 12.0, 24.0], "pe_hbm_gb": [6.0, 12.0, 24.0],
"bw_hbm_gbs": [256.0, 512.0, 1024.0], "bw_hbm_gbs": [256.0, 512.0, 1024.0],
"peak_tflops_f16": [8.0, 16.0, 32.0], "peak_tflops_f16": [8.0, 16.0, 32.0],
"bw_intra_gbs": [128.0, 256.0, 512.0], "bw_intra_gbs": [512.0, 1024.0, 2048.0],
"bw_inter_gbs": [128.0, 256.0, 512.0], "bw_inter_gbs": [128.0, 256.0, 512.0],
"bw_intersip_gbs": [50.0, 100.0, 200.0], "bw_intersip_gbs": [50.0, 100.0, 200.0],
} }
tests/analytical_visualization/model_config.py
+1 -1
View File
@@ -215,7 +215,7 @@ class MachineParams:
# HBM per-PE bandwidth (GB/s). # HBM per-PE bandwidth (GB/s).
bw_hbm_gbs: float = 256.0 bw_hbm_gbs: float = 256.0
# Intra-cube PE↔PE link BW (GB/s). # Intra-cube PE↔PE link BW (GB/s).
bw_intra_gbs: float = 128.0 bw_intra_gbs: float = 512.0
# Inter-cube (UCIe D2D) BW (GB/s). # Inter-cube (UCIe D2D) BW (GB/s).
bw_inter_gbs: float = 128.0 bw_inter_gbs: float = 128.0
# Inter-SIP (C2C / RDMA) BW (GB/s). # Inter-SIP (C2C / RDMA) BW (GB/s).