executorch/backends/xnnpack/partition/graphs (#7384)

gmagogsfm · facebook-github-bot · commit b7def6935f59 · 2024-12-19T10:34:45.000-08:00
Summary: Pull Request resolved: #7384 Reviewed By: avikchaudhuri, ydwu4 Differential Revision: D67381578
diff --git a/backends/xnnpack/partition/graphs/sdpa.py b/backends/xnnpack/partition/graphs/sdpa.py
@@ -76,6 +76,7 @@ def forward(
                         v,
                         mask,
                     ),
+                    strict=True,
                 ),
                 compile_config=get_xnnpack_edge_compile_config(),
             )

Original file line number	Diff line number	Diff line change
`@@ -76,6 +76,7 @@ def forward(`
`76`	`76`	`v,`
`77`	`77`	`mask,`
`78`	`78`	`),`
	`79`	`+ strict=True,`
`79`	`80`	`),`
`80`	`81`	`compile_config=get_xnnpack_edge_compile_config(),`
`81`	`82`	`)`