Skip to content

Commit ba12b06

Browse files
lint fixes
1 parent 4be6ae1 commit ba12b06

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

onnxscript/rewriter/ort_fusions/attention.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -176,8 +176,10 @@ def no_match(val: ir.Value, dims: Sequence[str]) -> bool:
176176
return check_result.fail("Hidden size is not an integer.", projected)
177177
if not (
178178
_ir_utils.is_singleton_value(start1, 0)
179-
and _ir_utils.get_singleton_value(end1) == _ir_utils.get_singleton_value(start2)
180-
and _ir_utils.get_singleton_value(end2) == _ir_utils.get_singleton_value(start3)
179+
and _ir_utils.get_singleton_value(end1)
180+
== _ir_utils.get_singleton_value(start2)
181+
and _ir_utils.get_singleton_value(end2)
182+
== _ir_utils.get_singleton_value(start3)
181183
and _ir_utils.is_singleton_value(end3, lambda x: x >= hidden_size)
182184
):
183185
return check_result.fail(

0 commit comments

Comments
 (0)