Skip to content

Commit 31b3a85

Browse files
committed
Bug fix to disable lower recomposed sdpa
1 parent 63e8025 commit 31b3a85

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

backends/xnnpack/partition/xnnpack_partitioner.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -869,7 +869,9 @@ def __init__(
869869
self.quant = quant
870870

871871
# TODO(T174256335) - remove this once we have a better way to handle >2d Mask
872-
self._lower_recomposed_sdpa: bool = _lower_recomposed_sdpa or True
872+
self._lower_recomposed_sdpa: bool = (
873+
_lower_recomposed_sdpa if _lower_recomposed_sdpa is not None else True
874+
)
873875

874876
self.delegation_spec = DelegationSpec(XnnpackBackend.__name__, [])
875877
self.partition_tags: Dict[str, DelegationSpec] = {}

0 commit comments

Comments
 (0)