[ROCm][CI] Update AITER AR+RMS e2e fusion counts for final-norm coverage (#51273)
Signed-off-by: Divakar Verma <divakar.verma@amd.com>
Divakar Verma committed
Aug 6, 2026 at 14:05 UTC
4f851bef6c4ef3a691aa19f798b220819d19dde4
1 file changed
+2
-2
tests/compile/fusions_e2e/models.py
+2
-2
@@ -75,7 +75,7 @@ llama3_8b = ModelFusionInfo(
75
model_name="meta-llama/Llama-3.1-8B-Instruct",
76
matches=lambda n_layers: Matches(
77
ar_rms_fusion=n_layers * 2 + 1,
78
- aiter_ar_rms_fusion=n_layers * 2,
78
+ aiter_ar_rms_fusion=n_layers * 2 + 1,
79
sequence_parallel=n_layers * 2 + 1,
80
async_tp=n_layers * 4,
81
),
@@ -137,7 +137,7 @@ qwen3_a3b = ModelFusionInfo(
137
matches=lambda n_layers: Matches(
138
norm_rope_fusion=n_layers,
139
ar_rms_fusion=n_layers * 2 + 1,
140
- aiter_ar_rms_fusion=n_layers * 2,
140
+ aiter_ar_rms_fusion=n_layers * 2 + 1,
141
sequence_parallel=n_layers * 2 + 1,
142
async_tp=n_layers * 2,
143
),