Skip to content

Commit b7df20d

Browse files
guangy10facebook-github-bot
authored andcommitted
Bug fix to disable lower recomposed sdpa (#4211)
Summary: Pull Request resolved: #4211 Reviewed By: digantdesai, mcr229 Differential Revision: D59640840 Pulled By: guangy10 fbshipit-source-id: 984dec35485b4396ace28310eb61376b7bbebd80
1 parent a924500 commit b7df20d

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)