- Fix to work around an issue in auto selection logic which requires the strides of tensors to be present (#168)

Anerudhan · web-flow · commit 80a8e4af4d89 · 2025-09-17T17:15:03.000-07:00
diff --git a/include/cudnn_frontend/graph_interface.h b/include/cudnn_frontend/graph_interface.h
@@ -279,9 +279,11 @@ class Graph : public ICudnn, public INode {
         }
 
         if (attributes.implementation == AttentionImplementation_t::AUTO) {
+            // Temporary WAR to not choose UNIFIED for now.
+            attributes.implementation = AttentionImplementation_t::COMPOSITE;
             // Sets attributes.implementation to a supporting implementation,
             // or leaves as AUTO if none found
-            attributes._auto_select_implementation(context);
+            // attributes._auto_select_implementation(context);
         }
 
         switch (attributes.implementation) {

Original file line number	Diff line number	Diff line change
`@@ -279,9 +279,11 @@ class Graph : public ICudnn, public INode {`
`279`	`279`	`}`
`280`	`280`
`281`	`281`	`if (attributes.implementation == AttentionImplementation_t::AUTO) {`
	`282`	`+ // Temporary WAR to not choose UNIFIED for now.`
	`283`	`+ attributes.implementation = AttentionImplementation_t::COMPOSITE;`
`282`	`284`	`// Sets attributes.implementation to a supporting implementation,`
`283`	`285`	`// or leaves as AUTO if none found`
`284`		`- attributes._auto_select_implementation(context);`
	`286`	`+ // attributes._auto_select_implementation(context);`
`285`	`287`	`}`
`286`	`288`
`287`	`289`	`switch (attributes.implementation) {`