Skip to content

Commit 705f963

Browse files
committed
removeusage of TracingEqn
1 parent 15755c9 commit 705f963

File tree

1 file changed

+1
-10
lines changed

1 file changed

+1
-10
lines changed

pennylane/capture/dynamic_shapes.py

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -232,16 +232,7 @@ def custom_staging_rule(
232232
jax.core.no_effects,
233233
source_info,
234234
)
235-
tracing_eqn = TracingEqn(
236-
list(tracers),
237-
returned_vars,
238-
primitive,
239-
params,
240-
eqn.effects,
241-
source_info,
242-
eqn.ctx,
243-
)
244-
jaxpr_trace.frame.add_eqn(tracing_eqn)
235+
jaxpr_trace.frame.add_eqn(eqn)
245236
return out_tracers
246237

247238
pe.custom_staging_rules[primitive] = custom_staging_rule

0 commit comments

Comments
 (0)