diff --git a/tests/analytical_visualization/app.py b/tests/analytical_visualization/app.py index 0285b02..645c11d 100644 --- a/tests/analytical_visualization/app.py +++ b/tests/analytical_visualization/app.py @@ -374,7 +374,7 @@ with st.sidebar.expander("Hardware", expanded=False): bw_intra = st.selectbox( "intra-cube GB/s", [128.0, 256.0, 512.0, 1024.0, 2048.0, 4096.0], - index=2, key="bw_intra", + index=0, key="bw_intra", ) bw_inter = st.selectbox( "inter-cube GB/s", diff --git a/tests/analytical_visualization/auto_hardware.py b/tests/analytical_visualization/auto_hardware.py index 219e36c..4f15b96 100644 --- a/tests/analytical_visualization/auto_hardware.py +++ b/tests/analytical_visualization/auto_hardware.py @@ -38,7 +38,7 @@ _HW_KNOB_DEFAULTS = { "pe_hbm_gb": 6.0, "bw_hbm_gbs": 256.0, "peak_tflops_f16": 8.0, - "bw_intra_gbs": 512.0, + "bw_intra_gbs": 128.0, "bw_inter_gbs": 128.0, "bw_intersip_gbs": 50.0, } @@ -47,7 +47,7 @@ _HW_VALUES_BALANCED = { "pe_hbm_gb": [6.0, 12.0], "bw_hbm_gbs": [256.0, 512.0], "peak_tflops_f16": [8.0, 16.0], - "bw_intra_gbs": [512.0, 1024.0], + "bw_intra_gbs": [128.0, 256.0], "bw_inter_gbs": [128.0, 256.0], "bw_intersip_gbs": [50.0, 100.0], } @@ -56,7 +56,7 @@ _HW_VALUES_COARSE = { "pe_hbm_gb": [6.0, 12.0, 24.0], "bw_hbm_gbs": [256.0, 512.0, 1024.0], "peak_tflops_f16": [8.0, 16.0, 32.0], - "bw_intra_gbs": [512.0, 1024.0, 2048.0], + "bw_intra_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], } diff --git a/tests/analytical_visualization/model_config.py b/tests/analytical_visualization/model_config.py index 3bb7df5..8f66d37 100644 --- a/tests/analytical_visualization/model_config.py +++ b/tests/analytical_visualization/model_config.py @@ -215,7 +215,7 @@ class MachineParams: # HBM per-PE bandwidth (GB/s). bw_hbm_gbs: float = 256.0 # Intra-cube PE↔PE link BW (GB/s). - bw_intra_gbs: float = 512.0 + bw_intra_gbs: float = 128.0 # Inter-cube (UCIe D2D) BW (GB/s). bw_inter_gbs: float = 128.0 # Inter-SIP (C2C / RDMA) BW (GB/s).