adding Enabling Blackwell support (#1254)

Signed-off-by: LosCrossos <165311345+loscrossos@users.noreply.github.com>

Author: loscrossos

setup.py

@@ -176,7 +176,9 @@ def get_flash_attention2_nvcc_archs_flags(cuda_version: int):
         return []
     # Figure out default archs to target
     DEFAULT_ARCHS_LIST = ""
-    if cuda_version >= 1108:
+    if cuda_version >= 1208:
+        DEFAULT_ARCHS_LIST = "8.0;8.6;9.0;10.0;12.0"
+    elif cuda_version >= 1108:
         DEFAULT_ARCHS_LIST = "8.0;8.6;9.0"
     elif cuda_version > 1100:
         DEFAULT_ARCHS_LIST = "8.0;8.6"