Removed duplicate include for ROCM_VERSION_DEV. Corrected indentation and code cleanup.

Author: akashveramd

CMakeLists.txt

@@ -895,9 +895,9 @@ if(USE_ROCM)
   endif()
 endif()
 
-# CK shared lib linkage
+# link CK library
 if(USE_ROCM)
-    if(UNIX AND (USE_CK_FLASH_ATTENTION))
+  if(UNIX AND USE_CK_FLASH_ATTENTION)
     include(cmake/External/ck.cmake)
   endif()
 endif()

aten/src/ATen/CMakeLists.txt

@@ -173,13 +173,6 @@ file(GLOB flash_attention_hip_hip "native/transformers/hip/flash_attn/*.hip")
 # if USE_FLASH_ATTENTION is set, ensure CK instances get generated
 if(USE_FLASH_ATTENTION)
   if(USE_CK_FLASH_ATTENTION)
-    if(DEFINED ENV{PYTORCH_ROCM_ARCH})
-      list(LENGTH PYTORCH_ROCM_ARCH NUM_ARCHS)
-      if(NUM_ARCHS GREATER 1)
-        message(WARNING "Building CK for multiple archs can increase build time considerably!
-        Consider setting PYTORCH_ROCM_ARCH env var value as the gfx arch you need to build for")
-      endif()
-    endif()
     message(STATUS "USE_CK_FLASH_ATTENTION is set; building PyTorch with CK Flash Attention enabled")
     message(STATUS "Generating CK kernel instances...")
     # disable buidling CK files

cmake/External/ck.cmake

@@ -18,9 +18,6 @@ else()
     message(FATAL_ERROR "Failed to get submodule status for composable_kernel.")
 endif()
 
-# get ROCm version from LoadHIP.cmake
-include(${CMAKE_SOURCE_DIR}/cmake/public/LoadHIP.cmake)
-
 # full path for CK library on compute-artifactory.amd.com
 set(url "https://compute-artifactory.amd.com/artifactory/rocm-generic-local")
 set(ck_lib_full_path "${url}/torch_ck_gen_lib/ck_${ck_commit_hash}/rocm_${ROCM_VERSION_DEV}/libck_kernels.so")