fix: add linear_before_reset=1 to GRU op calls in aten_gru

Copilot · justinchuby · Copilot · commit 170a407aa07b · 2026-03-14T21:35:32.000Z
Co-authored-by: justinchuby &lt;11205048+justinchuby@users.noreply.github.com&gt;
diff --git a/onnxscript/function_libs/torch_lib/ops/core.py b/onnxscript/function_libs/torch_lib/ops/core.py
@@ -4357,6 +4357,7 @@ def aten_gru(
                 initial_h=layer_h,
                 direction=direction,
                 hidden_size=hidden_size_attr,
+                linear_before_reset=1,
             )
         else:
             Y, Y_h = op.GRU(
@@ -4366,6 +4367,7 @@ def aten_gru(
                 initial_h=layer_h,
                 direction=direction,
                 hidden_size=hidden_size_attr,
+                linear_before_reset=1,
             )
 
         # Y shape: [seq_length, num_directions, batch_size, hidden_size]

Original file line number	Diff line number	Diff line change
`@@ -4357,6 +4357,7 @@ def aten_gru(`
`4357`	`4357`	`initial_h=layer_h,`
`4358`	`4358`	`direction=direction,`
`4359`	`4359`	`hidden_size=hidden_size_attr,`
	`4360`	`+ linear_before_reset=1,`
`4360`	`4361`	`)`
`4361`	`4362`	`else:`
`4362`	`4363`	`Y, Y_h = op.GRU(`
`@@ -4366,6 +4367,7 @@ def aten_gru(`
`4366`	`4367`	`initial_h=layer_h,`
`4367`	`4368`	`direction=direction,`
`4368`	`4369`	`hidden_size=hidden_size_attr,`
	`4370`	`+ linear_before_reset=1,`
`4369`	`4371`	`)`
`4370`	`4372`
`4371`	`4373`	`# Y shape: [seq_length, num_directions, batch_size, hidden_size]`