Allow d > 128 for cudnn version > 9.11.0 for B100 (#145)

Author: Anerudhan

include/cudnn_frontend/node/scaled_dot_product_flash_attention.h

@@ -1,4 +1,4 @@
-#pragma once
+    #pragma once
 
 #include <cstdlib>
 
@@ -940,6 +940,11 @@ class SDPABackwardNode : public NodeCRTP<SDPABackwardNode> {
             RETURN_CUDNN_FRONTEND_ERROR_IF((d_qk > 256) || (d_qk % 8 != 0) || (d_v > 256) || (d_v % 8 != 0),
                                         error_code_t::GRAPH_NOT_SUPPORTED,
                                         "Num hidden_dim shoud be less than or equal to 256 and hidden_dim should be multiple of 8");
+        } else if (prop.major == 10 && detail::get_backend_version() >= 91100) {
+            // validate basic dimension requirements
+            RETURN_CUDNN_FRONTEND_ERROR_IF((d_qk % 8 != 0) || (d_v % 8 != 0),
+                                        error_code_t::GRAPH_NOT_SUPPORTED,
+                                        "Num hidden_dim shoud be should be multiple of 8");
         } else {
             // validate basic dimension requirements
             RETURN_CUDNN_FRONTEND_ERROR_IF((d_qk > 128) || (d_qk % 8 != 0) || (d_v > 128) || (d_v % 8 != 0),