reduce warn level using pt2.2_fa (#1354)

DeepLink-org · Sep 3, 2024 · 02f03c6 · 02f03c6
1 parent 7b59b56
commit 02f03c6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/impl/torch/functions/functions_ext/flash-attention/CMakeLists.txt b/impl/torch/functions/functions_ext/flash-attention/CMakeLists.txt
@@ -19,7 +19,7 @@ if(${Torch_VERSION_MAJOR} EQUAL 2 AND ${Torch_VERSION_MINOR} EQUAL 0)
 elseif(${Torch_VERSION_MAJOR} EQUAL 2 AND ${Torch_VERSION_MINOR} EQUAL 1)
   set(FLASH_ATTN_LIB_PATH "${FLASH_ATTN_LIB_PATH_2_1}")
 else()
-  message(FATAL_ERROR "No valid torch version for setting FLASH_ATTN_LIB_PATH")
+  message(STATUS "No valid torch version for setting FLASH_ATTN_LIB_PATH")
 endif()
 
 message(STATUS "FLASH_ATTN_LIB_PATH: ${FLASH_ATTN_LIB_PATH}")