@@ -1 +1 @@
-8949def81eb744b9219e2d794c80265928c40a5b
+fde43b6d617752433310b4668c6b5c55e503521a
@@ -254,6 +254,8 @@ def fake_output_layer(hidden, weight=None, runtime_gather_output=None):  # noqa:
         mtp_loss_scaling_factor=1.0,
         calculate_per_token_loss=True,
         use_mup=False,
+        # process_mtp_loss reads the CP layout; mirror TransformerConfig's default.
+        attention_cp_layout="zigzag",
     )
     output = process_mtp_loss(
         hidden_states=hidden_states,
Original file line number	Diff line number	Diff line change
`@@ -1 +1 @@`
`1`		`-8949def81eb744b9219e2d794c80265928c40a5b`
	`1`	`+fde43b6d617752433310b4668c6b5c55e503521a`
Original file line number	Diff line number	Diff line change
`@@ -254,6 +254,8 @@ def fake_output_layer(hidden, weight=None, runtime_gather_output=None): # noqa:`
`254`	`254`	`mtp_loss_scaling_factor=1.0,`
`255`	`255`	`calculate_per_token_loss=True,`
`256`	`256`	`use_mup=False,`
	`257`	`+ # process_mtp_loss reads the CP layout; mirror TransformerConfig's default.`
	`258`	`+ attention_cp_layout="zigzag",`
`257`	`259`	`)`
`258`	`260`	`output = process_mtp_loss(`
`259`	`261`	`hidden_states=hidden_states,`