diff '--color=auto' -ur pytorch-v2.10.0.orig/caffe2/CMakeLists.txt pytorch-v2.10.0/caffe2/CMakeLists.txt
--- pytorch-v2.10.0.orig/caffe2/CMakeLists.txt	2026-03-19 19:44:30.899933030 +0100
+++ pytorch-v2.10.0/caffe2/CMakeLists.txt	2026-03-19 19:47:12.177312807 +0100
@@ -942,7 +942,7 @@
   set(CUDA_LINK_LIBRARIES_KEYWORD PRIVATE)
   list(APPEND Caffe2_HIP_SRCS ${GENERATED_CXX_TORCH_CUDA})
   hip_add_library(torch_hip ${Caffe2_HIP_SRCS})
-  if(USE_FLASH_ATTENTION)
+  if(USE_FLASH_ATTENTION OR USE_MEM_EFF_ATTENTION)
     target_link_libraries(torch_hip PRIVATE __caffe2_aotriton)
   endif()
   set(CUDA_LINK_LIBRARIES_KEYWORD)
diff '--color=auto' -ur pytorch-v2.10.0.orig/cmake/External/aotriton.cmake pytorch-v2.10.0/cmake/External/aotriton.cmake
--- pytorch-v2.10.0.orig/cmake/External/aotriton.cmake	2026-03-19 19:44:30.604614804 +0100
+++ pytorch-v2.10.0/cmake/External/aotriton.cmake	2026-03-19 19:46:32.080467007 +0100
@@ -234,10 +234,6 @@
 
   # Note it is INSTALL"ED"
   if(DEFINED ENV{AOTRITON_INSTALLED_PREFIX})
-    install(DIRECTORY
-            $ENV{AOTRITON_INSTALLED_PREFIX}/${CMAKE_INSTALL_LIBDIR}
-            $ENV{AOTRITON_INSTALLED_PREFIX}/include
-            DESTINATION ${__AOTRITON_INSTALL_DIR})
     set(__AOTRITON_INSTALL_DIR "$ENV{AOTRITON_INSTALLED_PREFIX}")
     message(STATUS "Using Preinstalled AOTriton at ${__AOTRITON_INSTALL_DIR}")
   elseif(DEFINED ENV{AOTRITON_INSTALL_FROM_SOURCE})