Allow building for sm90a (pytorch#125523)

# Summary
Fixes: pytorch#125413
Pull Request resolved: pytorch#125523
Approved by: https://github.com/Skylion007

Author: drisspg

cmake/Modules_CUDA_fix/upstream/FindCUDA/select_compute_arch.cmake

@@ -242,8 +242,8 @@ function(CUDA_SELECT_NVCC_ARCH_FLAGS out_variable)
   # remove dots and convert to lists
   string(REGEX REPLACE "\\." "" cuda_arch_bin "${cuda_arch_bin}")
   string(REGEX REPLACE "\\." "" cuda_arch_ptx "${cuda_arch_ptx}")
-  string(REGEX MATCHALL "[0-9()]+" cuda_arch_bin "${cuda_arch_bin}")
-  string(REGEX MATCHALL "[0-9]+"   cuda_arch_ptx "${cuda_arch_ptx}")
+  string(REGEX MATCHALL "[0-9()]+a?" cuda_arch_bin "${cuda_arch_bin}")
+  string(REGEX MATCHALL "[0-9]+a?"   cuda_arch_ptx "${cuda_arch_ptx}")
 
   if(cuda_arch_bin)
     list(REMOVE_DUPLICATES cuda_arch_bin)