# -----------------------------------------------------------------------------------------------------------
# Copyright (c) 2025 Huawei Technologies Co., Ltd.
# This program is free software, you can redistribute it and/or modify it under the terms and conditions of
# CANN Open Software License Agreement Version 2.0 (the "License").
# Please refer to the License for details. You may not use this file except in compliance with the License.
# THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED,
# INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.
# See LICENSE in the root of the software repository for the full text of the License.
# -----------------------------------------------------------------------------------------------------------

# Ascend mode
if(DEFINED ENV{ASCEND_HOME_PATH})
  set(ASCEND_DIR $ENV{ASCEND_HOME_PATH})
else()
  if ("$ENV{USER}" STREQUAL "root")
    if(EXISTS /usr/local/Ascend/ascend-toolkit/latest)
      set(ASCEND_DIR /usr/local/Ascend/ascend-toolkit/latest)
    else()
      set(ASCEND_DIR /usr/local/Ascend/latest)
    endif()
  else()
    if(EXISTS $ENV{HOME}/Ascend/ascend-toolkit/latest)
      set(ASCEND_DIR $ENV{HOME}/Ascend/ascend-toolkit/latest)
    else()
      set(ASCEND_DIR $ENV{HOME}/Ascend/latest)
    endif()
  endif()
endif()
message(STATUS "Search libs under install path ${ASCEND_DIR}")

set(CMAKE_PREFIX_PATH ${ASCEND_DIR}/)

set(CMAKE_MODULE_PATH
  ${CMAKE_CURRENT_SOURCE_DIR}/cmake/modules
  ${CMAKE_MODULE_PATH}
  ${ASCEND_DIR}/include/ge/cmake
)
message(STATUS "CMAKE_MODULE_PATH            :${CMAKE_MODULE_PATH}")

set(OPS_TRANSFORMER_CXX_FLAGS)
string(APPEND OPS_TRANSFORMER_CXX_FLAGS " ${COMPILE_OP_MODE}")
string(APPEND OPS_TRANSFORMER_CXX_FLAGS " -Wall")
string(APPEND OPS_TRANSFORMER_CXX_FLAGS " -Wextra")
string(APPEND OPS_TRANSFORMER_CXX_FLAGS " -Wshadow")
string(APPEND OPS_TRANSFORMER_CXX_FLAGS " -Wformat=2")
string(APPEND OPS_TRANSFORMER_CXX_FLAGS " -fno-common")
string(APPEND OPS_TRANSFORMER_CXX_FLAGS " -fPIC")
if(NOT "${CMAKE_CXX_COMPILER_ID}" STREQUAL "Clang")
#  string(APPEND OPS_TRANSFORMER_CXX_FLAGS " -Werror")   # TODO: add -Werror when fix all compile warnings
  string(APPEND OPS_TRANSFORMER_CXX_FLAGS " -Wformat-signedness")
  string(APPEND OPS_TRANSFORMER_CXX_FLAGS " -Wno-missing-include-dirs")
  string(APPEND OPS_TRANSFORMER_CXX_FLAGS " -Wno-write-strings")
endif()
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${OPS_TRANSFORMER_CXX_FLAGS}")
message(STATUS "compile option:${CMAKE_CXX_FLAGS}")

find_package(dlog MODULE REQUIRED)
find_package(securec MODULE)
find_package(OPBASE MODULE REQUIRED)
find_package(platform MODULE REQUIRED)
find_package(metadef MODULE REQUIRED)
find_package(mmpa MODULE)
find_package(runtime MODULE REQUIRED)
find_package(nnopbase MODULE REQUIRED)
find_package(tilingapi MODULE REQUIRED)
find_package(aicpu MODULE REQUIRED)

if(BUILD_WITH_INSTALLED_DEPENDENCY_CANN_PKG)
 	message(STATUS "Finding gen es pkg")
 	find_package(GenerateEsPackage MODULE QUIET)
 	if(NOT GenerateEsPackage_FOUND)
 	  message(FATAL_ERROR "GenerateEsPackage not found")
 	endif()
endif()