We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 4be6ae1 commit ba12b06Copy full SHA for ba12b06
1 file changed
onnxscript/rewriter/ort_fusions/attention.py
@@ -176,8 +176,10 @@ def no_match(val: ir.Value, dims: Sequence[str]) -> bool:
176
return check_result.fail("Hidden size is not an integer.", projected)
177
if not (
178
_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)
+ and _ir_utils.get_singleton_value(end1)
+ == _ir_utils.get_singleton_value(start2)
181
+ and _ir_utils.get_singleton_value(end2)
182
+ == _ir_utils.get_singleton_value(start3)
183
and _ir_utils.is_singleton_value(end3, lambda x: x >= hidden_size)
184
):
185
return check_result.fail(
0 commit comments