Skip to content

Commit

Permalink
Do not create fuse_seqopt optionally
Browse files Browse the repository at this point in the history
  • Loading branch information
Ricardo Vieira committed Oct 7, 2022
1 parent a619a4e commit 9a4c2c3
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions aesara/tensor/rewriting/elemwise.py
Original file line number Diff line number Diff line change
Expand Up @@ -896,16 +896,16 @@ def print_profile(cls, stream, prof, level=0):
print(blanc, " time_toposort", prof[7], file=stream)


fuse_seqopt = SequenceDB()
if config.tensor__local_elemwise_fusion:
# Must be after gpu(48.5) and before AddDestroyHandler(49.5)
fuse_seqopt = SequenceDB()
fuse_seqopt.register(
"composite_elemwise_fusion",
FusionOptimizer(local_elemwise_fusion),
"fast_run",
"fusion",
position=1,
)
# Position before AddDestroyHandler(49.5)
compile.optdb.register( # type: ignore
"elemwise_fusion",
fuse_seqopt,
Expand Down

0 comments on commit 9a4c2c3

Please sign in to comment.