Skip to content

Commit 2a270aa

Browse files
remove verbosity
1 parent 1b29ea7 commit 2a270aa

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

onnxscript/rewriter/ort_fusions/attention.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -224,6 +224,7 @@ def rewrite(
224224
_domain="com.microsoft",
225225
_outputs=2,
226226
)
227+
# TODO: Switch back order of outputs
227228
return present, attention
228229
else:
229230
return op.Attention(
@@ -277,6 +278,6 @@ def fuse_attention(model: ir.Model, *, debug: bool = False) -> int:
277278
count = attention_rules.apply_to_model(model)
278279
if debug and count == 0:
279280
tracer = pattern.MatchingTracer()
280-
attention_rules.apply_to_model(model, verbose=3, tracer=tracer)
281+
attention_rules.apply_to_model(model, tracer=tracer)
281282
tracer.report()
282283
return count

0 commit comments

Comments
 (0)