diff --git a/CMakeLists.txt b/CMakeLists.txt index 870e67f0a..00c8f2d97 100755 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -209,6 +209,7 @@ set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/bin) set(COMMON_HEADER_DIRS ${PROJECT_SOURCE_DIR} ${CUDA_PATH}/include + ${CUDNN_INCLUDE_PATH} ${CUTLASS_HEADER_DIR} ${CUTLASS_EXTENSIONS_DIR} ${PROJECT_SOURCE_DIR}/3rdparty/trt_fp8_fmha/src