From 4143babfdda9e8161908796474c3410f4f096094 Mon Sep 17 00:00:00 2001 From: ustclight-sls Date: Tue, 3 Sep 2024 19:56:14 +0800 Subject: [PATCH] reduce warn level using pt2.2_fa --- .../functions/functions_ext/flash-attention/CMakeLists.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/impl/torch/functions/functions_ext/flash-attention/CMakeLists.txt b/impl/torch/functions/functions_ext/flash-attention/CMakeLists.txt index 83108553f..1ad766cf0 100644 --- 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}")